算法描述 :
給定一個(gè)數(shù)組nums和一個(gè)值val，你需要原地移除所有數(shù)值等于val的元素，返回移除后數(shù)組的新長(zhǎng)度。
不要使用額外的數(shù)組空間，你必須在原地修改輸入數(shù)組并在使用 O(1) 額外空間的條件下完成。
元素的順序可以改變。你不需要考慮數(shù)組中超出新長(zhǎng)度后面的元素。
示例 1:
給定 nums = [3,2,2,3], val = 3,
函數(shù)應(yīng)該返回新的長(zhǎng)度2, 并且nums中的前兩個(gè)元素均為2。
你不需要考慮數(shù)組中超出新長(zhǎng)度后面的元素。
示例 2:
給定 nums = [0,1,2,2,3,0,4,2], val=2,
函數(shù)應(yīng)該返回新的長(zhǎng)度`5`, 并且nums 中的前五個(gè)元素為`0`, `1`, `3`, `0`, `4`。
注意這五個(gè)元素可為任意順序。
你不需要考慮數(shù)組中超出新長(zhǎng)度后面的元素。
說(shuō)明:
為什么返回?cái)?shù)值是整數(shù)，但輸出的答案是數(shù)組呢?
請(qǐng)注意，輸入數(shù)組是以“引用”方式傳遞的，這意味著在函數(shù)里修改輸入數(shù)組對(duì)于調(diào)用者是可見(jiàn)的。
你可以想象內(nèi)部操作如下:
nums是以“引用”方式傳遞的。也就是說(shuō)，不對(duì)實(shí)參作任何拷貝
int len = removeElement(nums, val);
// 在函數(shù)里修改輸入數(shù)組對(duì)于調(diào)用者是可見(jiàn)的。
// 根據(jù)你的函數(shù)返回的長(zhǎng)度, 它會(huì)打印出數(shù)組中**該長(zhǎng)度范圍內(nèi)**的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
算法實(shí)現(xiàn) :
Java實(shí)現(xiàn) :
class Solution {
public int removeElement(int[] nums, int val) {
int index = 0; // 記錄下一次要存不為val的元素
// 從前往后掃描, 將不等于val的元素依次前移,后索引后移
for (int i = 0; i < nums.length; i++) {
if (nums[i] != val) {
nums[index] = nums[i];
index++;
}
}
return index;
}
}
