题目链接:移除元素
暴力解法
思路
暴力解法,时间复杂度O(n2)
注意在移动元素时,下标指针i也要跟着减少
题解
//暴力解法
class Solution{
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++) {
if (nums[i] == val) {//如果找到,将所有元素向前移动一位
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
size--;//数组大小-1
i--;//下标i以后的值都往前移动一位,i也需要往前移动一位
}
}
return size;
}
};
双指针法
思路
两个指针,一个slow,一个fast,fast指向新数组需要的元素,slow指向新数组的下标值,这就是为什么是if(val!=nums[fastIndex])
nums[slowIndex++] = nums[fastIndex] 是先赋值,再让slowIndex加一
题解
class Solution{
public:
int removeElement(vector<int>& nums, int val) {
//两个指针,一个slow,一个fast,fast指向新数组需要的元素,slow指向新数组的下标值
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
if (val != nums[fastIndex]) {
nums[slowIndex++] = nums[fastIndex];
}
}
return slowIndex;
}
};