1. 程式人生 > >LeetCode--陣列類演算法--刪除排序陣列中的重複項 II

LeetCode--陣列類演算法--刪除排序陣列中的重複項 II

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

不要使用額外的陣列空間,你必須在原地修改輸入陣列並在使用 O(1) 額外空間的條件下完成。

示例 1:

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

函式應返回新長度 length = 5, 並且原陣列的前五個元素被修改為 1, 1, 2, 2, 3 
你不需要考慮陣列中超出新長度後面的元素。

示例 2:

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

函式應返回新長度 length = 7, 並且原陣列的前七個元素被修改為 0, 0
, 1, 1, 2, 3, 3 。
你不需要考慮陣列中超出新長度後面的元素

 

思路:和前一個題很相像,不過比較的是陣列前一個元素而已。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int begin=1;
        int end=2;
        int n=nums.size();
        if(n<3) return n;
        while(end<nums.size())
        {
            if(nums[begin-1]!=nums[end])
            {
                nums[++begin]=nums[end];
            }
            end++;
        }
        return begin+1;
    }
};