此题的vector 完全可以做数组用。一开始的常规思路当然是依次把数字前移,然而由于考虑不全面,错了几次。
一开始的错误代码是这样:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
vector<int>& vec = nums;
int len=vec.size();
for(int i=0;i<len;i++){
if(vec[i]==0){
for(int j=i;j<len;j++){
vec[j]=vec[j+1];
vec[len]=0;
}
}
}
}
};
发现由于每次遇到某个元素,需要前移,移动过后,i是需要在原地继续判断的。因为此时的原地判断值就是前移之前的那个i+1的值。
修改后正确的代码如下,可以超过大概7%的伙伴。由于是两次for循环么,时间复杂度太高。(如不加len--这种,大概只超过5%)
class Solution {
public:
void moveZeroes(vector<int>& nums){
vector<int>& vec = nums;
int len=vec.size();
for(int i=0;i<len;i++){
if(vec[i]==0){
for(int j=i+1;j<len;j++){
vec[j-1]=vec[j];
//vec[j]=0;
}
vec[len-1]=0;
i--;len--;
}
}
}
};
转换思路考虑,发现可以逆向想。直接将非0元素一起前移,最后在数组后填0就可以了。
按思路写了一下 超过了20%多的答案,剩余答案大部分也都是一样的复杂度了
class Solution {
public:
void moveZeroes(vector<int>& nums){
vector<int>& vec = nums;
int len=vec.size();
int head=0;
for(int i=0;i<len;i++){
if(vec[i]!=0){
vec[head]=vec[i];
head++;
}
}
//int tail=len-head-1;
for(int j=head;j<len;j++){
vec[j]=0;
}
}
};