點選 這裡 可以檢視更多演算法面試相關內容~

題目描述

給你一個數組 nums 和一個值 val，你需要 原地 移除所有數值等於 val 的元素，並返回移除後陣列的新長度。

不要使用額外的陣列空間，你必須僅使用 O(1) 額外空間並 原地 修改輸入陣列。

元素的順序可以改變。你不需要考慮陣列中超出新長度後面的元素。

示例 1:

給定 nums = [3,2,2,3], val = 3,

函式應該返回新的長度 2, 並且 nums 中的前兩個元素均為 2。

你不需要考慮陣列中超出新長度後面的元素。

示例 2:

給定 nums = [0,1,2,2,3,0,4,2], val = 2,

函式應該返回新的長度 5, 並且 nums 中的前五個元素為 0, 1, 3, 0, 4。

注意這五個元素可為任意順序。

你不需要考慮陣列中超出新長度後面的元素。

說明:

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

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

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

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

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

雙指標解法

這題的思路和我們上一題「26. 刪除排序陣列中的重複項（簡單）」是類似的。

還是使用指標 i 進行陣列遍歷，但另外一個指標 j 指向的是待插入的位置。

當遍歷過程中 i 所指向的值不等於 val 時，將 i 的值放到 j 的位置，然後讓 j 後移（指向下一個待插入的位置）：

class Solution {
    public int removeElement(int[] nums, int val) {
        int n = nums.length;
        int j = 0;
        for (int i = 0; i < n; i++) {
            if (nums[i] != val) {
                nums[j++] = nums[i];
            }
        }
        return j;
    }
}

• 時間複雜度： O ( n ) O(n) O(n)

• 空間複雜度： O ( 1 ) O(1) O(1)

最後

這是我們「刷穿 LeetCode」系列文章的第 No.27 篇，系列開始於 2021/01/01，截止於起始日 LeetCode 上共有 1916 道題目，部分是有鎖題，我們將先將所有不帶鎖的題目刷完。

在這個系列文章裡面，除了講解解題思路以外，還會盡可能給出最為簡潔的程式碼。如果涉及通解還會相應的程式碼模板。

由於 LeetCode 的題目隨著周賽 & 雙週賽不斷增加，為了方便我們統計進度，我們將按照系列起始時的總題數作為分母，完成的題目作為分子，進行進度計算。當前進度為 27/1916 。

為了方便各位同學能夠電腦上進行除錯和提交程式碼，我建立了相關的倉庫：Github 地址 & Gitee 地址。

在倉庫地址裡，你可以看到系列文章的題解連結、系列文章的相應程式碼、LeetCode 原題連結和一些其他的優選題解。

#演算法與資料結構
#LeetCode題解
#演算法面試