26. 删除排序数组中的重复项 (有序数组去重)

题目

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

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

示例 1:

给定数组 nums = [1,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。

你不需要考虑数组中超出新长度后面的元素。

来源:力扣(LeetCode)


解答

C语言

按照双指针的思路,iji, j分别从0和1开始依次往后移动,当nums[i]nums[j]不相等时,i++,将后者的值赋值给i的新位置,否则只按照循环j++,i保持不变

最终返回值需要i+1,因为i是下标。

//24 ms	7.7 MB
int removeDuplicates(int* nums, int numsSize){
    if (numsSize == 0) {
        return 0;
    }

    int i, j;

    for (i = 0, j = 1; j < numsSize; j++) {
        if (nums[i] != nums[j]) {
            nums[++i] = nums[j];
        }
    }

    return i+1;
}

golang

同理

// 4 ms	4.6 MB
func removeDuplicates(nums []int) int {
    if len(nums) == 0 {
        return 0
    }

    var i int

    for j := 1; j < len(nums); j++ {
        if nums[i] != nums[j] {
            i++
            nums[i] = nums[j]
        }
    }

    return i+1
}
posted @ 2020-10-12 09:41  Dar1inge  阅读(199)  评论(0)    收藏  举报