1. 程式人生 > >LeetCode 26. 刪除排序陣列中的重複項 C++程式碼實現

LeetCode 26. 刪除排序陣列中的重複項 C++程式碼實現

題目描述:

給定一個排序陣列,你需要在原地刪除重複出現的元素,使得每個元素只出現一次,返回移除後陣列的新長度。

不要使用額外的陣列空間,你必須在原地修改輸入陣列並在使用 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。

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

解題思路:使用直接插入的思想,將vector陣列中元素重新插入,初始時將第一個元素當成有序表,依次判斷與前一個元素是否重複,不重複則插入。

程式碼實現:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(!nums.size()) return NULL;
        int k=0;
        for(int i=1;i<nums.size();i++){//直接插入思想
            if(nums[i]!=nums[k]){
               nums[++k]=nums[i]; 
            }
        }
        nums.resize(k+1);
        return k+1;
    }
};