题意:给定一棵多叉树,若每个非叶子节点的儿子里都有至少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;
}