从树上下来的 for 循环

For-loop going down a tree

本文关键字:for 循环 上下      更新时间:2023-10-16

我有一个k-ary树存储在一个名为treestd::vector中:

0,                          1,                          2
3,       4,       5         6,       7,       8         9,       10,      11
12,13,14 15,16,17 18,19,20  21,22,23 24,25,26 27,28,29  30,31,32 33,34,35 36,37,38

tree.size() == 39如上所示

我正在寻找类似于以下内容的循环:

for (size_t i=0,j=tree.size(); i!=j; i=some_magic()) // <<---here
{
  std::cout << i << std::endl;
}

哪个应该输出:

0
3
12
13
14
4
15
...

对于深度优先搜索,您需要某种堆栈。

int child(int i, int c) { return (i+1)*3+c; }
int main()
{
    std::vector<int> parents = { 2, 1, 0 };
    while(!parents.empty())
    {
        int i = parents.back();
        parents.pop_back();
        std::cout << tree[i] << std::endl;
        for(int c = 0; c != 3; ++c)
        {
            if(child(i, 2-c) < tree.size())
                parents.push_back(child(i, 2-c));
        }
    }
}