【团体程序设计天梯赛】L2-051 满树的遍历

思路：

就是一个很简单的树的度数和先序遍历的问题。

代码：

#include <iostream>
#include <vector>
#include <set>
#define N 100010
using namespace std;

int n,f,d,rt,sp;
vector<int> g[N];
set<int> s;
void preorder(int rt){
    if(sp) cout<<" ";
    sp=1;
    cout<<rt;
    for(auto k: g[rt]) preorder(k);
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>f;
        if(f) g[f].push_back(i);
        else rt=i;
    }
    if(n==1){
        cout<<0<<" yes"<<endl<<1;
        return 0;
    }
    for(int i=1;i<=n;i++){
        if(g[i].size()!=0) {
            s.insert(g[i].size());
            d=max(d,(int)g[i].size());
        }
    }
    cout<<d<<" ";
    if(s.size()==1) cout<<"yes";
    else cout<<"no";
    cout<<endl;
    preorder(rt);
    return 0;
}