给定一整数数组,如何保持非0元素相对顺序前提下,使值为0的元素全部移动到数组末尾。

给定一整数数组,如何保持非0元素相对顺序前提下,使值为0的元素全部移动到数组末尾。

这道题在面试的时候没有做出来,手写代码总感觉不对,而且有一些函数也记不得了。= =果然还是C++写的代码太少
现在复盘一下

#include 
#include 
#include 

auto latest_iterator(std::vector::iterator begin, std::vector::iterator end) {
    while (end != begin && *end == 0) {
        end--;
    }
    return end;
}

void sort(std::vector &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 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

你可能感兴趣的:(C++)