題目

給定一個排序陣列，你需要在 原地 刪除重複出現的元素，使得每個元素只出現一次，返回移除後陣列的新長度。

不要使用額外的陣列空間，你必須在 原地 修改輸入陣列 並在使用 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語言

按照雙指標的思路，i和j，i, 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
}