L2-051 满树的遍历

一棵“k 阶满树”是指树中所有非叶结点的度都是 k 的树。给定一棵树，你需要判断其是否为 k 阶满树，并输出其前序遍历序列。

注：树中结点的度是其拥有的子树的个数，而树的度是树内各结点的度的最大值。

输入格式：

输入首先在第一行给出一个正整数 n（≤105），是树中结点的个数。于是设所有结点从 1 到 n 编号。
随后 n 行，第 i 行（1≤i≤n）给出第 i 个结点的父结点编号。根结点没有父结点，则对应的父结点编号为 0。题目保证给出的是一棵合法多叉树，只有唯一根结点。

输出格式：

首先在一行中输出该树的度。如果输入的树是 k 阶满树，则加 1 个空格后输出 yes，否则输出 no。最后在第二行输出该树的前序遍历序列，数字间以 1 个空格分隔，行首尾不得有多余空格。
注意：兄弟结点按编号升序访问。

输入样例 1：

7
6
5
5
6
6
0
5

输出样例 1：

3 yes
6 1 4 5 2 3 7

输入样例 2：

7
6
5
5
6
6
0
4

输出样例 2：

3 no
6 1 4 7 5 2 3

一开始用的是

void printtree(int start){
    while(tree[start].size() != 0){
        cout << " " << tree[start].front();
        if(tree[tree[start].front()].size() != 0){ 
            printtree(tree[start].front());
        }
        tree[start].erase(tree[start].begin());
    }
}

没想到代码超时，后面才知道erase 操作是 O(n) 复杂度的，下次还是尽量少用erase吧

完整代码

#include <bits/stdc++.h>
using namespace std;
const int N = 100000;
int n,k;
vector<vector<int>> tree(N+1);

void printtree(int start){
    for(auto x:tree[start]){
        cout << " " << x ;
        if(tree[x].size() != 0){ 
            printtree(x);
        }
    }

}

int main(){
    cin >>n;
    int node,start;
    for(int i=1;i<=n;i++){
        cin >> node;
        if(node == 0){
            start = i;
        }
        tree[node].push_back(i);
    }
    bool treejudge = true;
    k = tree[start].size();
    for(int i=1;i<=n;i++){
        //cout << tree[i].size() << endl;
        if(tree[i].size() != 0 && tree[i].size() != k)
            treejudge = false;
    }
    if(treejudge == true){
        cout << k << " yes" << endl;
    }
    else{
        cout<< k << " no" << endl;
    }

    cout << start;
    printtree(start);
}