80. 删除有序数组中的重复项 II

【中等题】

题目:

给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。

不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

示例:

示例 1:

输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。
示例 2:

输入:nums = [0,0,1,1,1,1,2,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
解释:函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。

思路:

【双指针】维护一个区间,left作为左边界,right作为有边界。如果区间大小大于2,就移动数组。如果right元素与left不同的时候,就让 left变成right继续往右走。

代码:

class Solution {
public:
    int removeDuplicates(vector& nums) {
        int left = 0, right = 0;
        while(left < nums.size() && right < nums.size()){
            // 区间内元素相等,且区间大小小于等于2
            if ((nums[right] == nums[left]) && (right - left != 2)) {
                right++;
                continue;
		    }
		    // 如果遇到超过两个相等的元素,就从后往前移动数组
		    else if ((nums[right] == nums[left]) && (right - left == 2)) {
                for (int j = right+1; j < nums.size(); j++) {
                    nums[j - 1] = nums[j];
                }
                // 删除尾元素,vector数组长度减1
                nums.pop_back();
		    }
            // 更新左边界
            else if (nums[right] != nums[left]) {
                left = right;
            }
        }
        return nums.size();
    }
};

运行结果 

80. 删除有序数组中的重复项 II_第1张图片

把while判断条件改成:

 while(right < nums.size())

运行结果明显变好:

80. 删除有序数组中的重复项 II_第2张图片

你可能感兴趣的:(LeetCode,算法)