【leetcode】80. Remove Duplicates from Sorted List II

题目说明

给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/

解法1

时间复杂度:O(n)
空间复杂度:O(1)
思路:使用双指针,第一个指针j表示调整后的数组最后一个位置,第二个指针i遍历原数组。
当nums[i]与nums[j]不等时,直接添加即可,将i位置添加到新数组末尾,并记录下匹配次数(此时为1)。
当nums[i]与nums[j]相等时,就需要判断匹配次数了,判断是否满足匹配次数小于2,若满足则将位置添加到新数组末尾,匹配次数加1;不满足,则直接跳过。
若要修改最大出现的次数,修改max值即可。

int removeDuplicates(vector<int>& nums) 
{
    if (nums.empty())
        return 0;
    int j = 0;
    int count = 1;
    int max = 2;
    for (int i = 1; i < nums.size(); i ++)
    {
        if (nums[i] == nums[j] && count < max)
        {
            nums[++j] = nums[i];
            count ++;
        }
        else if (nums[i] != nums[j])
        {
            nums[++j] = nums[i];
            count = 1;
        }
    }
    return j + 1;
}

猜你喜欢

转载自www.cnblogs.com/JesseTsou/p/10322463.html