[LeetCode] 26、删除排序数组中的重复项

原创已于 2025-11-25 19:59:31 修改 · 146 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

于 2019-12-06 21:39:04 首次发布

算法（UVa + LeetCode + OJ + ……）专栏收录该内容

442 篇文章

订阅专栏

本文介绍了一种在原地删除排序数组中重复元素的方法，确保每个元素仅出现一次，并返回新长度。通过两个指针i和j，i为慢指针用于记录新数组的位置，j为快指针用于遍历数组。当遇到不同元素时，将其值复制到i位置并递增i。

题目描述

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

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

给定 nums = [0,0,1,1,1,2,2,3,3,4]
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。

解题思路

数组完成排序后，我们可以放置两个指针 $i$ 和 $j$ ，其中 $i$ 是慢指针，而 $j$ 是快指针。只要 $n u m s [i] = n u m s [j]$ ，我们就增加 $j $ 以跳过重复项。
当我们遇到 $\neq nums[i]$ 时，跳过重复项的运行已经结束，因此我们必须把它（ $n u m s [j]$ ）的值复制到 $n u m s [i + 1]$ 。然后递增 $i$ ，接着我们将再次重复相同的过程，直到 $j $ 到达数组的末尾为止。

参考代码

自己A掉的代码：

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int length = nums.size();
        if(length == 0 || length == 1)
            return length;
        
        int newLength = 0;
        for(int i = 0; i < length; i++){
            // 遇到重复数字
            if(i != length-1 && nums[i] == nums[i+1]){
                int duplicationNum = nums[i];
                for(; i < length; i++){
                    if(nums[i] != duplicationNum)
                        break;
                }
                
                nums[newLength] = nums[--i];
            }else{
                nums[newLength] = nums[i];
            }
            
            newLength++;
        }
        
        return newLength;
    }
    
};

优雅写法：（推荐）

public int removeDuplicates(int[] nums) {
    if (nums.length == 0) return 0;
    int i = 0;
    for (int j = 1; j < nums.length; j++) {
        if (nums[j] != nums[i]) {
            nums[++i] = nums[j];
        }
    }
    return i + 1;
}