5221. 【GDOI2018模拟7.10】A

题目大意：

给你一棵有根树，问你在这棵树上总共有多少棵子树的节点构成了一个完整的整数区间。

考试想法：

考试时就想到了正解，正解就是从下到上遍历整一棵树，每一个节点记录一下它的最小值min、最大值max、和节点的个数sum，如果(max-min+1)==sum那么ans++,否则继续遍历。

代码：

#include<bits/stdc++.h>
using namespace std;
int ans,s[100005],l[100005],r[100005],n,u,v,rt,cnt=1,head[100005],fa[100005];
struct Edge
{
	int to,next;
}edge[100005];
void add_edge()
{
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt;
	cnt++;
}
void dp(int x)
{
	s[x]=1;
	for(int i=head[x];i!=0;i=edge[i].next)
	{
		dp(edge[i].to);
		if(l[edge[i].to]<l[x]) l[x]=l[edge[i].to];
		if(r[edge[i].to]>r[x]) r[x]=r[edge[i].to];
		s[x]+=s[edge[i].to];
	}
	if(s[x]!=r[x]-l[x]+1) return;
	ans++;
}
int main()
{
	scanf("%d",&n);
	for(register int i=1;i<n;i++)
	{
		l[i]=r[i]=i;
		scanf("%d%d",&u,&v);
		fa[v]=u;
		add_edge();
	}
	l[n]=r[n]=n;
	for(register int i=1;i<=n;i++) 
		if(fa[i]==0)
		{
			rt=i;
			break;
		}
	dp(rt);
	printf("%d",ans);
	return 0;
}