1. 題目詳情

給你一個陣列 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
注意這五個元素可為任意順序
你不需要考慮陣列中超出新長度後面的元素

2. 題解

從題目瞭解到要求在陣列中匹配預期值,將其移除,但不能申請陣列以外的空間,可以將解題思路轉變為,新增座標元素 k 從 0 開始自增,匹配到與預期值不一樣的,把當前元素賦值給陣列的 k 位置,等遍歷完後返回 K 值就是新陣列長度

class Solution {
    public int removeElement(int[] nums, int val) {
        int k = 0;
        for (int data : nums) {
            if (data != val) {
                nums[k++] = data;
            }
        }
        return k;
    }
}