LeetCode刷题之删除排序数组中的重复项

291 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

题意描述

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

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

示例 1:

输入:nums = [1,1,2] 输出:2, nums = [1,2]

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

示例 2:

输入:nums = [0,0,1,1,1,2,2,3,3,4] 输出:5, nums = [0,1,2,3,4]

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

思路1:双指针

分析:

  • 数组是有序的,不存在[0, 1, 1, 2, 3, 1, 1] 这种情况
  • 操作待处理的数组nums,将不重复的元素往左边移动,最终得到从nums[0]开始到nums[newLength-1] 结束的共newLength个不重复的数组个数.
class Solution {
    public int removeDuplicates(int[] nums) {
        if(nums == null || nums.length == 0) {
            return 0;
        }
        int i = 0, j = 1;
        while(j<=nums.length-1) {
            if(nums[i] != nums[j]) {
               if(i + 1 < j) {
                    nums[i+1] = nums[j]; 
               }
               i = i+1;
            }
            j = j+1;
        } 
        return i+1;
    }
}

思路2:递归

public static int removeDuplicates(int[] nums) {
    if (nums.length == 0) {
        return 0;
    }
    int j = 0;
    int i = 0;
    int next = nextDiff(nums, nums[i], i);
    while (next > 0) {
        j++;
        nums[j] = nums[next];
        next = nextDiff(nums, nums[next], next);
    }
    return j+1;
}

private static int nextDiff(int[] newArray, int val, int start) {
    if (start == newArray.length - 1) {
        return 0;
    }
    start++;
    if (newArray[start] == val) {
        return nextDiff(newArray, val, start);
    } else {
        return start;
    }
}