AtCoder Beginner Contest 165 F LIS on Tree 树的遍历+最长上升子序列nlogn+二分+回溯

AtCoder Beginner Contest 165   比赛人数11730  比赛开始后15分钟看到所有题

AtCoder Beginner Contest 165  F  LIS on Tree   树的遍历+最长上升子序列nlogn+二分+回溯

总目录详见https://blog.csdn.net/mrcrack/article/details/104454762

在线测评地址https://atcoder.jp/contests/abc165/tasks/abc165_f

样例配图如下

Input:
10
1 2 5 3 4 6 7 3 2 4
1 2
2 3
3 4
4 5
3 6
6 7
1 8
8 9
9 10
Output:
1
2
3
3
4
4
5
2
2
3

第一次能在赛后，独立做出F题，可喜可贺。

借助样例，弄懂了题，几个知识点涌了出来，

树不外乎遍历，子树节点数，深度。本题考察树的遍历。

最长上升子序列(若对此不熟悉，请看此文最长不下降子序列 最长不升子序列 最长上升子序列 最长下降子序列 n^2 nlogn),从本题提供的数据范围看，考察的是O(nlogn)算法，这里面需要二分。

在编写调试代码的过程中，发现该题还需加入回溯，这样树上不同分支的数据，就不会互相影响了。

该题若有树的遍历，最长上升子序列O(nlogn)，结合题目看代码，还是容易看懂的。

AC代码如下

#include <stdio.h>
#define maxn 200010
int a[maxn],head[maxn],tot,n,d[maxn],ans,mx[maxn];
struct node{
	int to,next;
}e[maxn<<1];
void add_edge(int u,int v){//邻接表
	tot++,e[tot].to=v,e[tot].next=head[u],head[u]=tot;
}
void dfs(int u,int fa){//树的遍历,u是当前遍历节点,fa是u的父亲节点
	int b,v,save,flag,l,r,mid;;
	for(b=head[u];b;b=e[b].next){
		v=e[b].to;
		if(v==fa)continue;
		save=ans,flag=0;//为回溯作准备
		if(d[ans]<a[v])d[++ans]=a[v];
		else{//二分
			l=0,r=ans;
			while(l+1<r){
				mid=(l+r)/2;
				if(d[mid]<a[v])l=mid;
				else r=mid;//a[st[mid]]>=a[v]
			}
			flag=d[r],d[r]=a[v];//记录d[]数组中被改变的数据,为回溯作准备
		}
		mx[v]=ans;//记录最大值
		dfs(v,u);
		ans=save;//回溯
		if(flag)d[r]=flag;//回溯
	}
}
void init(){
	int i,u,v;
	scanf("%d",&n);
	for(i=1;i<=n;i++)scanf("%d",&a[i]);
	for(i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		add_edge(u,v),add_edge(v,u);//无向图
	}
}
int main(){
	init();
	ans=1,d[1]=a[1],mx[1]=1;//记录根节点数据
	dfs(1,0);
	for(int i=1;i<=n;i++)printf("%d\n",mx[i]);
	return 0;
}