力扣:27. 移除元素
題目:
27. 移除元素
難度簡(jiǎn)單1701收藏分享切換為英文接收動(dòng)態(tài)反饋
給你一個(gè)數(shù)組?nums
?和一個(gè)值?val
,你需要?原地?移除所有數(shù)值等于?val
?的元素,并返回移除后數(shù)組的新長(zhǎng)度。
不要使用額外的數(shù)組空間,你必須僅使用?O(1)
?額外空間并?原地?修改輸入數(shù)組。
元素的順序可以改變。你不需要考慮數(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]); }
?
示例 1:
輸入:nums = [3,2,2,3], val = 3輸出:2, nums = [2,2]解釋?zhuān)汉瘮?shù)應(yīng)該返回新的長(zhǎng)度 2, 并且 nums 中的前兩個(gè)元素均為 2。你不需要考慮數(shù)組中超出新長(zhǎng)度后面的元素。例如,函數(shù)返回的新長(zhǎng)度為 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也會(huì)被視作正確答案。
示例 2:
輸入:nums = [0,1,2,2,3,0,4,2], val = 2輸出:5, nums = [0,1,4,0,3]解釋?zhuān)汉瘮?shù)應(yīng)該返回新的長(zhǎng)度 5
, 并且 nums 中的前五個(gè)元素為 0
, 1
, 3
, 0
, 4。注意這五個(gè)元素可為任意順序。你不需要考慮數(shù)組中超出新長(zhǎng)度后面的元素。
?
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
第一種對(duì)法:
class?Solution?{
public:
????int?removeElement(vector<int>&?nums,?int?val)?{
????????int?cnt=0;
????????for(int?i=0;i<nums.size()-cnt;i++){
????????????if(nums[i]==val){
????????????????cnt++;
????????????????for(int?j=i;j<nums.size()-1;j++){
????????????????????nums[j]=nums[j+1];
????????????????}
????????????????i--;
????????????}
????????}
????????return?nums.size()-cnt;
????}
};
暴力法,O(n^2);
第二種對(duì)法:
class?Solution?{
public:
????int?removeElement(vector<int>&?nums,?int?val)?{
????????int?fast,slow;
????????fast=0,slow=0;
????????while(fast<nums.size()){
????????????if(nums[fast]!=val){
????????????????nums[slow]=nums[fast];
????????????????slow++;
????????????}
????????????fast++;
????????}
????????return?slow;
????}
};
雙指針,slow用來(lái)改變數(shù)組,fast用來(lái)遍歷數(shù)組,時(shí)間復(fù)雜度O(n)