80. Remove Duplicates from Sorted Array II

最新推荐文章于 2024-10-30 17:45:35 发布

快乐划水程序猿

最新推荐文章于 2024-10-30 17:45:35 发布

阅读量134

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题

本文链接：https://blog.youkuaiyun.com/Alexia23/article/details/90904709

刷题专栏收录该内容

174 篇文章

订阅专栏

本文介绍了一种解决LeetCode上移除排序数组中的重复项II问题的方法，通过统计数组中每个数字出现的次数，并确保任何数字最多只出现两次。文章详细解释了代码实现过程，包括特殊情况处理，如数组元素数量小于等于2时的特判，以及如何正确地将不重复的数字推入结果数组，并在最后删除多余的元素。

题目：https://leetcode.com/problems/remove-duplicates-from-sorted-array-ii/
思路：统计每个数字出现的次数，出现多于两次，那就push两个进去
坑：
1. vector元素只有2个或者更少，需要特判
2. 注意边界，初始值和最终（是否将所有的不重复的数字都push进去了）
3. 注意要删除最后不要的元素 pop_back

代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int res = 0;
        if (nums.size() <= 2)
            return nums.size();
        int one_number = 1;
        for (int i = 1; i < nums.size(); i++) {
            if (nums[i] != nums[i-1]) {
                nums[res] = nums[i-1];
                res++;
                if (one_number >= 2) {
                    nums[res] = nums[i-1];
                    res++;
                }
                one_number = 1;
            } else {
                one_number++;
            }
        }
        nums[res] = nums[nums.size()-1];
        res++;
        if (one_number >= 2) {
            nums[res] = nums[nums.size()-1];
            res++;
        } 
        for (int i = res; i < nums.size(); i++)
            nums.pop_back();
        return res;
    }
};