1. 程式人生 > 遊戲 >《貪婪之秋2》公佈 2024年登陸PC和主機

《貪婪之秋2》公佈 2024年登陸PC和主機

80. 刪除有序陣列中的重複項 II

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

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

說明:

為什麼返回數值是整數,但輸出的答案是陣列呢?

請注意,輸入陣列是以「引用」方式傳遞的,這意味著在函式裡修改輸入陣列對於呼叫者是可見的。

你可以想象內部操作如下:

// nums 是以“引用”方式傳遞的。也就是說,不對實參做任何拷貝
int len = removeDuplicates(nums);

// 在函式裡修改輸入陣列對於呼叫者是可見的。
// 根據你的函式返回的長度, 它會打印出陣列中 該長度範圍內 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

示例 1:

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

示例 2:

輸入: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 。 不需要考慮陣列中超出新長度後面的元素。

提示:

  • 1 <= nums.length <= 3 * 104
  • -104 <= nums[i] <= 104
  • nums 已按升序排列

思路:

​ 可以直接擴充套件到每個元素出現k次。

​ 在有序陣列中,每個元素可以出現k次,那麼前k個可以直接取(無論如何不會超過k個),然後對於後面的任意數字,能夠保留的前提是:與當前寫入的位置前面的第 k 個元素進行比較,不相同則保留

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int len=0;
        int n=nums.size();
        for(int i=0;i<n;i++){
            if(len<2||nums[len-2]!=nums[i])//len<2的意思就是前k個直接取
                nums[len++]=nums[i];
        }
        return len;
    }
};