原题链接
思路:用结构体储存这个树,向量c记录该节点的子节点,node记录叶节点的数量,遍历所有节点计算出每个节点的叶节点数量node,然后判断是否符合题目的规定,然后输出。
代码:
#include<iostream>
#include<fstream>
#include<cstdio>
#include<iomanip>
#include<algorithm>
#include<vector>
#include<set>
#include<list>
#include<queue>
#include<deque>
#include<unordered_map>
#include<map>
#include<string>
#include<cstring>
#include<stack>
#include<cmath>
#define vint vector<int>
#define vstr vector<string>
#define vll vector<long long>
#define ll long long
#define pf printf
#define sf scanf
#define sfd(n) scanf("%d", &n)
#define sflf(n) scanf("%lf", &n)
#define sfll(n) scanf("%lld", &n)
#define pfd(n) printf("%d", n)
#define pflf(n) printf("%lf", n)
#define pfll(n) printf("%lld", n)
#define pft printf("\t")
#define pfn printf("\n")
#define pfk printf(" ")
#define PI 3.1415926
#define MAX 100000
#define M 1e18
using namespace std;
struct TREE {
vector<int> c;
int node;
};
TREE t[101000];
int main() {
int n, m;
sfd(n);
bool f = true;
for( int i=0; i<101000; i++ ) {
t[i].node = 0;
}
for( int i=2; i<=n; i++ ) {
sfd(m);
t[m].c.push_back(i);
}
for( int i=1; i<=n; i++) {
if( !t[i].c.empty() ) {
int len = t[i].c.size();
for( int j=0; j<len; j++ ) {
if( t[t[i].c[j]].c.empty() ) {
t[i].node++;
}
}
}
if( t[i].node<3&&t[i].c.size()>0 ) {
f = false;
break;
}
}
if( f ) {
pf("Yes\n");
}else {
pf("No\n");
}
return 0;
}