給定一個(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í)參作任何拷貝
這道題的描述很長(zhǎng),實(shí)際上刨除描述來(lái)看,就是把數(shù)組中所有的給定val都去掉,然后得到一個(gè)刪減后的數(shù)組(如果有空位需要用后面的補(bǔ)上)
使用一個(gè)index變量,記錄新數(shù)組的尾指針位置。
遍歷舊數(shù)組,和val相等則continue,不等則新數(shù)組增加一項(xiàng)
這道題所有操作都在一個(gè)給定數(shù)組num上執(zhí)行。
舊數(shù)組是nums[0-i]
新數(shù)組是nums[0-index]
隨著 i 的增加,index也相應(yīng)增加
具體代碼
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int index = 0; //新數(shù)組尾指針為0
for(int i = 0; i < nums.size(); i++){ //遍歷舊數(shù)組
if(nums[i] == val){ //如果是需要?jiǎng)h去的值,直接跳過(guò)
continue;
}
nums[index++] = nums[i]; //不需要?jiǎng)h去的值,放到新數(shù)組
}
return index; //返回新數(shù)組尾指針位置
}
};