Codeforces 913B Christmas Spruce

Codeforces 913B Christmas Spruce

题目在这里呀！

题意：给定一棵多叉树，若每个非叶子节点的儿子里都有至少3个叶子节点，则这棵树是漂亮的，否则就是不漂亮的。

问这棵树是不是漂亮的！（樱花树最漂亮呀！）

树上模拟。size[u]记录以u为根的子树大小（好像并没有用）

sum[u]表示以u为根有多少个儿子节点是叶子节点。

然后...就没有然后了呀！

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#define N 1010
using namespace std;
int n,p,cnt,head[N],vet[N],Next[N],size[N],sum[N];inline void add_edge(int u,int v){vet[++cnt]=v;Next[cnt]=head[u];head[u]=cnt;
}void dfs(int u){size[u]=1;if(!head[u]) return;for(int i=head[u];i;i=Next[i]){int v=vet[i];dfs(v);if(size[v]==1) sum[u]++;size[u]+=size[v];}
}int main()
{scanf("%d",&n);for(int i=2;i<=n;i++){scanf("%d",&p);add_edge(p,i);}dfs(1);for(int i=1;i<=n;i++) if(size[i]>1) if(sum[i]<3){puts("No");return 0;}puts("Yes");return 0;
}