【Leetcode】之Remove Duplicates from Sorted Array II

一.问题描述

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array nums = [1,1,1,2,2,3],

Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

二.我的解题思路

这道题首先需要理解题意，即需要以复制覆盖的方式来remove duplicate。我的思路是首先建立一个vector,然后遍历一次数组，记录下相邻元素不同的位置并且将这些位置存入vector中。比如对于[1,1,1,2,2,3]，那么遍历完一次数组之后，vector中就存放着[-1,2,4,5]其中-1是人为加的，而2，4,5则代表着nums[2]!=nums[2+1]诸如此类。vector中的这些位置下标将数组nums分成了不同的段。

接下来就是遍历vector并进行移位覆盖，注意移动的位数是有叠加的。测试通过的程序如下：

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int len = nums.size();
        int res=0;
        if(len<=2) return len;
        int i=0;
        vector<int> record;record.push_back(-1);
        while(i+1<len){
            if(nums[i]!=nums[i+1])
                record.push_back(i);
            i++;
        }
        record.push_back(len-1);
        int rec_len = record.size(); int step=0;
        for(int i=1;i<rec_len;i++){
            int curr_step = (record[i]-record[i-1]<=2)?0:(record[i]-record[i-1]-2);
            step+=curr_step;
            if(curr_step==0) res+=record[i]-record[i-1]; else res+=2;
            int st_idx=record[i]+1;
            if(st_idx>=len) break;
            int cnt=0;
            while(st_idx<=record[i+1] && cnt<2){
                nums[st_idx-step]=nums[st_idx];
                cnt++;
                st_idx++;
                
            }
            
                
            
             
        }
        return res;
      
    }
};