题意
给出一个数组和一个数,原地删除数组中与给出数值相等的数,即不能另开数组,求操作完后的数组长度。元素的顺序可以改变,超出数组长度的数不考虑。
思考
最开始的时候想到 vector.erase()
,但是因为未考虑到删除元素后 vector
长度发生改变,所以出错。然后看了别人的代码才意识到错误。
第二个思路是将末尾不等于所给数值的元素移动到前面,但未考虑到数组为空、只有一个数值、如何判断退出循环等边界条件,屡次出错。最后加入了很多条件判断,导致代码看上去很丑陋。
看了别人的代码后,发现思路很清晰,代码很优雅。之后在做题时应尽可能全面的思考问题,找出实现比较简洁的写法。
代码
测试代码:
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<int>& v) {
for (auto& i : v) {
cout << i << " ";
}
cout << endl;
}
int main(void) {
Solution s;
vector<int> nums{ 0, 1, 2, 2, 3, 0, 4, 2 };
cout << s.removeElement(nums, 2) << endl;
print_vector(nums);
return 0;
}
代码 1:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
for (int i = 0; i < nums.size(); i++) {
if (nums[i] == val) {
nums.erase(nums.begin() + i);
i--;
}
}
return nums.size();
}
};
代码 2:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int pos = nums.size() - 1;
for (int i = 0; i <= pos; i++) {
if (nums[i] == val) {
while (pos >= 0 && nums[pos] == val) pos--;
if (pos < 0 || i > pos) break;
swap(nums[i], nums[pos--]);
}
if (i > pos) break;
}
return pos + 1;
}
};
题解
拷贝覆盖:一个指针指向当前的位置,另一个指针来判断是否与所给数值相等。用后面的值覆盖前面的值。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int pos = 0;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] != val) {
nums[pos++] = nums[i];
}
}
return pos;
}
};
交换移除:一个指针指向当前位置,一个指针指向末尾,如果当前元素与所给数值相同,则与末尾交换。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int pos = 0, ans = nums.size() - 1;
while (pos <= ans) {
if (nums[pos] == val) {
nums[pos] = nums[ans--];
} else {
pos++;
}
}
return pos;
}
};