LeetCode 26. 删除排序數組中的重複項

題目描述

給定一個排序數組，你需要在原地删除重複出現的元素，使得每個元素隻出現一次，傳回移除後數組的新長度。

不要使用額外的數組空間，你必須在原地修改輸入數組并在使用** O(1) 額外**空間的條件下完成。

示例 1:

給定數組 nums = [1,1,2], 

函數應該傳回新的長度 2, 并且原數組 nums 的前兩個元素被修改為 1, 2。 
           

你不需要考慮數組中超出新長度後面的元素。

示例 2:

給定 nums = [0,0,1,1,1,2,2,3,3,4],

函數應該傳回新的長度 5, 并且原數組 nums 的前五個元素被修改為 0, 1, 2, 3, 4。
           

你不需要考慮數組中超出新長度後面的元素。

解題思路

解法一：标記下标，然後周遊數組元素依次進行比較，出現不等元素就指派給标記的下一個位置

解法二：利用STL裡的unique函數（由于給的是有序數組，是以可以直接用）

• unique()是C++标準庫函數裡面的函數，其功能是去除相鄰的重複元素（隻保留一個），是以使用前需要對數組進行排序
• distance主要是用來求兩個疊代器之間的元素個數。

具體代碼

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.empty()) return 0;
        int pos = 0;
        for(int i=0;i<nums.size()-1;i++){
            if(nums[i]!=nums[i+1]){
                nums[++pos] = nums[i+1];
            }
        }
        return pos+1;
    }
};

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        return distance(nums.begin(),unique(nums.begin(),nums.end()));
    }
};