给定一整数数组,如何保持非0元素相对顺序前提下,使值为0的元素全部移动到数组末尾。
这道题在面试的时候没有做出来,手写代码总感觉不对,而且有一些函数也记不得了。= =果然还是C++写的代码太少
现在复盘一下
#include <algorithm>
#include <vector>
#include <iostream>
auto latest_iterator(std::vector<int>::iterator begin, std::vector<int>::iterator end) {
while (end != begin && *end == 0) {
end--;
}
return end;
}
void sort(std::vector<int> &list) {
auto i = list.begin();
auto j = latest_iterator(i, (++list.rbegin()).base());
while (i != j) {
if (*i == 0) {
*i = *j;
*j = 0;
j = latest_iterator(i, j);
if (i == j) {
break;
}
}
i++;
}
}
int main(int argc, char **argv) {
std::vector<int> list{1, 23, 4, 0, 4, 5, 6, 0, 4, 3, 0, 0};
sort(list);
for (auto p : list) {
std::cout << p;
}
return 0;
}
其中思想不难就是有两个指针ij,i指向第一个元素,j指向最后一个非0的元素,从i开始遍历,发现有值为0的元素就和j交换,并把j的位置变为下一个非0的元素,如此反复。
主要问题在于如何把反向迭代器变为正向迭代器。因为如果要用迭代器做的话需要拿到最后一个元素的地址。
这里使用的是(++rbegin()).base()。 然后在改变了j迭代器的位置之后要判断i == j 放置错位导致程序BUG