链接:https://ac.nowcoder.com/acm/contest/40/B
来源:牛客网
珂朵莉的值域连续段
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 131072K,其他语言262144K
64bit IO Format: %lld
题目描述
珂朵莉给你一个有根树,求有多少个子树满足其内部节点编号在值域上连续
一些数在值域上连续的意思即其在值域上构成一个连续的区间
输入描述:
第一行有一个整数n,表示树的节点数。
接下来n–1行,每行两个整数x,y,表示存在一条从x到y的有向边。
输入保证是一棵有根树。
输出描述:
输出一个数表示答案
示例1
输入
复制
5
2 3
2 1
2 4
4 5
输出
复制
5
说明
节点1子树中编号为1,值域连续
节点3子树中编号为3,值域连续
节点5子树中编号为5,值域连续
节点4子树中编号为4,5,值域连续
节点2子树中编号为1,2,3,4,5,值域连续
备注:
对于100%的数据,有n <=100000
学算法半年载,一直没有接触图与树,今天开始涉猎树,希望’树先生’手下留情,轻点虐我。
该题应该怎么做呐,我们分析一波,看看<有多少个子树满足其内部节点编号在值域上连续
一些数在值域上连续的意思即其在值域上构成一个连续的区间>这句话的意思。意思是加上自己和所有的子树,最大值减去最小值如果等于结点个数就代表其是连续的,就是符号题意的。DFS就可以了。还是比较简单的。当然我也是借鉴别人的代码才看懂的,我自己就是一个小菜鸡。
#include <bits/stdc++.h>
using namespace std;
#define mem(a,n) memset(a,n,sizeof(a))
#define rep(i,a,n) for(int i=a;i<n;i++)
#define pb push_back
#define fi first
#define se second
typedef long long ll;
typedef unsigned long long ull;
const double eps=1e-6;
const int INF=0x3f3f3f3f;
const int N=1e5+5;
int ans,mx[N],mn[N],sz[N];
bool vis[N];
vector<int>e[N];
void dfs(int rt)
{
mx[rt]=mn[rt]=rt;
sz[rt]=1;
rep(i,0,e[rt].size())
{
int next_rt=e[rt][i];
dfs(next_rt);
mx[rt]=max(mx[rt],mx[next_rt]);
mn[rt]=min(mn[rt],mn[next_rt]);
sz[rt]+=sz[next_rt];
}
if(mx[rt]-mn[rt]+1==sz[rt]) ans++;
}
int main()
{
int n;
while(~scanf("%d",&n))
{
ans=0;
mem(vis,0);
rep(i,1,n)
{
int u,v;
scanf("%d%d",&u,&v);
vis[v]=1;
e[u].pb(v);
}
int root;
rep(i,1,n+1)
if(!vis[i])//防止重复遍历
{
root=i;
dfs(root);
break;
}
printf("%d\n",ans);
}
return 0;
}