給你一個(gè)數(shù)組 nums 和一個(gè)值 val,你需要 原地 移除所有數(shù)值等于 val 的元素,并返回移除后數(shù)組的新長(zhǎng)度。
不要使用額外的數(shù)組空間,你必須僅使用 O(1) 額外空間并 原地 修改輸入數(shù)組。
元素的順序可以改變。你不需要考慮數(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)度后面的元素。
來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/remove-element
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán),非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。
一、暴力法
1.首先在數(shù)組中逐個(gè)元素判斷是否和val相等,如果不相等,則needDo加1。
for(int i = 0;i<numsSize;i++){
if(nums[i] != val) needDo++;
}
2.比較 eqVal和needDo的大小。
這樣便于確定需要交換元素的最大次數(shù)。當(dāng)交換達(dá)到最大次數(shù)的時(shí)候,就沒(méi)有必要進(jìn)行交換了。因?yàn)閿?shù)組已經(jīng)達(dá)到了最終的狀態(tài)。
代碼
int removeElement(int* nums, int numsSize, int val){
int j = 0;
int eq = 0;
int needDo = 0;
for(int i = 0;i<numsSize;i++){
if(nums[i] != val) needDo++;
}
if(needDo == 0) return 0;
int eqVal = numsSize - needDo;//其中,needDo就是最終需要返回的元素,至于何時(shí)返回,取決于數(shù)組中元素的狀態(tài)。
for(int i = 0; i < numsSize; i++){//該for循環(huán)是為了調(diào)整數(shù)組中元素的位置。
if(nums[i] == val){
for(j = i+1;j<numsSize;j++){
if(nums[j] != val){
nums[i] = nums[j];
nums[j] = val;
eq++;
if(eq == needDo) return needDo;//當(dāng)交換的次數(shù)達(dá)到的時(shí)候,數(shù)組就達(dá)到了最終的狀態(tài)。
break;
}
}
}
}
return needDo;
}
二、雙指針?lè)?/h3>
int removeElement(int* nums, int numsSize, int val){
int i=0,j=0;
while(j<numsSize){
if(nums[j] == val){//等于的時(shí)候j增加,i不變
j++;
}else{//不等于的時(shí)候,j增加,nums[i]被賦值,i增加
nums[i] = nums[j];
i++;
j++;
}
}
return i;
}
640.gif
int removeElement(int* nums, int numsSize, int val){
int i=0,j=0;
while(j<numsSize){
if(nums[j] == val){//等于的時(shí)候j增加,i不變
j++;
}else{//不等于的時(shí)候,j增加,nums[i]被賦值,i增加
nums[i] = nums[j];
i++;
j++;
}
}
return i;
}
