【LeetCode】80. 刪除有序數組中的重複項 II

80. 刪除有序數組中的重複項 II

知識點:數組;排序;雙指針

題目描述

給你一個有序數組 nums ,請你 原地 刪除重複出現的元素,使每個元素 最多出現兩次 ,返回刪除後數組的新長度。

不要使用額外的數組空間,你必須在 原地 修改輸入數組 並在使用 O(1) 額外空間的條件下完成。

示例
輸入:nums = [1,1,1,2,2,3]
輸出:5, nums = [1,1,2,2,3]
解釋:函數應返回新長度 length = 5, 並且原數組的前五個元素被修改為 1, 1, 2, 2, 3 。 不需要考慮數組中超出新長度後面的元素。

輸入:nums = [0,0,1,1,1,1,2,3,3]
輸出:7, nums = [0,0,1,1,2,3,3]
解釋:函數應返回新長度 length = 7, 並且原數組的前五個元素被修改為 0, 0, 1, 1, 2, 3, 3 。 不需要考慮數組中超出新長度後面的元素。

解法一:雙指針

這道題目和26題唯一的區別就是這道題目要求是可以有兩個重複的,仍然用雙指針,一個從頭到尾遍歷,一個是指示當前需要的數組的最後一個;因為可以有兩個重複的,所以比較當前最後一個的前一個,如果不一樣,那就可以添加(即使和最後一個一樣也沒事),如果一樣了,那就證明index-1、index、還有i都是一樣的,那就不添加,直接後移;

class Solution {
    public int removeDuplicates(int[] nums) {
        int index = 1;  //仍然指示結果數組的最後一個;
        for(int i = 2; i < nums.length; i++){
            if(nums[i] != nums[index-1]){
                index++;  //如果和index-1相等,那證明index-1、index、i都相等,因為是排序的,所以不能加入,直接後移;
                nums[index] = nums[i];
            }
        }
        return index+1;
    }
}

相關題目

26. 刪除有序數組中的重複項