2018.10.09 NOIP训练 最大疯子树（dp+巧妙建图）

描述

给定一棵 n 个结点的树，结点编号为 1~n，i 号结点的权重记为 wi（每个点的权值各不相同）。我们定义一个“疯子树”为：
是一个联通子图。
我们将子图内的点按照权重从小到大排序后序列为 b1,b2,…,bm，对于任意的i(i<m)，bi到bi+1最短路径（不含 b_i和 b_{i+1}）上的结点的权值都小于 等于$w_{bi}$。
输出包含结点最多的“疯子树”的结点数。

输入

数据有多组 case,文件以 EOF 结束，每组第一行输入一个 n 表示树的节点数；
接下来一行包含 n 个整数，第 i 个数表示 i 号结点的权重 wi；接下行 n-1 行，第 i 行包含 2 个整数 ui, vi，表示 ui和 vi有一条边。

输出

对于每组 case 输出一行，包含一个整数，表示包含结点最多的“疯子树” 的结点数。

样例输入

5
1 4 2 3 5
1 2
2 3
3 4
4 5
5
2 5 4 1 3
1 2
2 3
3 4
4 5

样例输出

4
4

提示

对于 10%的数据有所有组的 n 之和 n≤20。
对于 30%的数据有所有组的 n 之和 n≤2000。
对于 100%的数据有所有组的 n 之和 n≤200000，0 < wi≤100000000，n 为正 整数。

标签

UESTCSpring科学营

---

比较妙的建图方法。
对于点$(u,v)$，我们将权值小的连向权值大的。
这样会连出一个有向无环图。
这样建图的好处是什么呢？
这时如果存在路径$i$->$u$和路径$i$->$v$。
那么$u$,$v$可以存在于同一个疯子树中。
于是令$f[i]$表示从$i$出发最多可以到的节点数。
然后有：
$ans=max${$f[i]$}
直接求$f[i]$是$O(n^2)$的。
显然可以记忆化一波。
代码：

#include<bits/stdc++.h>
#define N 200005
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int n,first[N],cnt=0,d[N],f[N],ans;
struct edge{int v,next;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;}
inline void dfs(int p){
	if(~f[p])return;
	f[p]=1;
	for(int i=first[p];i;i=e[i].next)dfs(e[i].v),f[p]+=f[e[i].v];
}
int main(){
	while(scanf("%d",&n)!=EOF){
		memset(f,-1,sizeof(f)),memset(first,0,sizeof(first)),cnt=0,ans=0;
		for(int i=1;i<=n;++i)d[i]=read();
		for(int i=1,u,v;i<n;++i){
			u=read(),v=read();
			if(d[u]<d[v])add(u,v);
			else add(v,u);
		}
		for(int i=1;i<=n;++i)dfs(i),ans=max(ans,f[i]);
		printf("%d\n",ans);
	}
	return 0;
}