题目描述

给你一个 非严格递增排列 的数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。

考虑 nums 的唯一元素的数量为 k ，你需要做以下事情确保你的题解可以被通过：

  • 更改数组 nums ，使 nums 的前 k 个元素包含唯一元素，并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。
  • 返回 k 。

示例 1：

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

算法思路

这个题能想到双指针，但是我一直没考虑清楚i, j两个指针的实际意义是什么，脑子傻了，因为我当时没搞清楚怎么确定这个元素在之前是否出现过？，所以自己写的方法依然是臭的不行，大体思路就是如果nums[i] <= nums[i-1]，也就是出现重复元素，就从当前位置找第一个比nums[i-1]大的元素nums[j]，得到的代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        // 如果当前位置小于等于前一个位置，就需要找到下一个大于当前位置的数
        int i = 1;
        for (i = 1; i < nums.size(); i++) {
            if (nums[i] <= nums[i-1]) {
                int j = i;
                while (j < nums.size() && nums[j] <= nums[i-1]) {
                    j++;
                    continue;
                }
                if (j != nums.size()) {
                    nums[i] = nums[j];
                }
                else {
                    break;
                }
            }
        }
        return i;
    }
};

​ 提交发现时间复杂度$O(N^2)$:烂的不行：

image-20250520120747189

所以去看了官方题解，真觉得对于双指针的理解必须到位，哎

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        // 快慢指针，快指针指向“不同”的元素，慢指针指向需要被替换的元素
        // 快指针从下表1~n-1，如果nums[fast] != nums[fast-1]，则说明这个fast指向的元素是新元素
        // 此时将nums[slow++] = nums[fast]
        int slow = 1;
        for (int fast = 1; fast < nums.size(); fast++) {
            if (nums[fast] != nums[fast-1]) {
                nums[slow++] = nums[fast];
            }
        }
        return slow;
    }
};

大致的思路可以去看官方题解。