陕西OI多校模拟赛Day4t1 A

题目描述

小 A 得到了一棵美丽的有根树。这棵树由 n 个节点以及 n - 1 条有向边构成，每条边都从父亲节点指向儿子节点，保证除了根节点以外的每个节点都有一个唯一的父亲。树上的节点从 1 到n 标号。该树的一棵子树的定义为某个节点以及从该节点出发能够达到的所有节点的集合，显然这棵树共有 n 棵子树。小 A 认为一棵有根树是美丽的当且仅当这棵树内节点的标号构成了一个连续的整数区间。现在小 A 想知道这棵树上共有多少棵美丽的子树。

输入

第一行有一个整数 n，表示树的节点数。 

接下来 n–1 行，每行两个整数 u，v，表示存在一条从 u 到 v 的有向边。 

输入保证该图是一棵有根树。

输出

输出一个整数占一行，表示对应的答案。

         又是一道自己超级弱的树的题目...

       开始打的是枚举+树状数组的nlogn暴力，觉得数据小于等于十万还是可以的，不过思路错了，到考试结束也没有发现......

       首先可以想想对一棵树进行遍历能得到什么，第一个子节点的个数，第二个子树里节点的编号的最小值，第三个子树里节点的编号的最大值，对于大多数题目，也就是使用这三个变量。

       由于编号是不可重复的，那么如果我们对于一个节点，它子节点的最大值-子节点的最小值+1刚好等于其子节点的个数，就能说明这个区间是连续的了。

       对于一棵树用dfs遍历即可...一句话题解系列...

       下附AC代码。

#include<iostream>
#include<vector>
#include<stdio.h>
#include<string.h>
#define maxn 100005
using namespace std;
int n;
int ans=0;
int root;
int fa[maxn];
int son[maxn];
vector<int> edge[maxn];
int getfa(int x)
{
	return fa[x]==x ? x : fa[x]=getfa(fa[x]);
}
int maxx[maxn],minn[maxn];
void dfs(int now,int par)
{
	son[now]=1;
	int len=edge[now].size();
	maxx[now]=minn[now]=now;
	for(int i=0;i<len;i++)
	{
		int nex=edge[now][i];
		dfs(nex,now);
		son[now]+=son[nex];
		maxx[now]=max(maxx[now],maxx[nex]);
		minn[now]=min(minn[now],minn[nex]);
	}
	if((maxx[now]-minn[now]+1)==(son[now]))
	ans++;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	fa[i]=i;
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		edge[x].push_back(y);
		fa[y]=x;
	}
	root=getfa(1);
	dfs(root,-1);
	printf("%d\n",ans);
}