#启发式合并,LIS,平衡树#洛谷 4577 [FJOI2018]领导集团问题

题目

在一棵树上选择最多的点,使得存在祖先关系的点满足\(w_x\leq w_y\),其中\(x\)\(y\)的祖先


分析

祖先链上要满足\(LIS\),考虑将子节点的LIS序列合并至节点\(x\)
用启发式合并就可以做到\(O(nlog^2n)\),同时还要将\(w_x\)插入,
由于需要查询后继,所以用\(\text{STL::set}\)实现即可


代码

#include <cstdio>
#include <cctype>
#include <algorithm>
#include <set>
#define rr register
using namespace std;
const int N=200011; typedef long long lll;
struct node{int y,next;}e[N]; multiset<int>dp[N];
multiset<int>::iterator it; int k=1,as[N],a[N],n;
inline signed iut(){
	rr int ans=0; rr char c=getchar();
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
	return ans;
}
inline void dfs(int x,int fa){
	for (rr int i=as[x];i;i=e[i].next){
	    dfs(e[i].y,x);
	    if (dp[x].size()<dp[e[i].y].size())
		    swap(dp[x],dp[e[i].y]);
		for (it=dp[e[i].y].begin();it!=dp[e[i].y].end();)
		    dp[x].insert(*it),dp[e[i].y].erase(it++);
	}
	dp[x].insert(a[x]),it=dp[x].lower_bound(a[x]);
	if (it!=dp[x].begin()) dp[x].erase(--it);
}
signed main(){
	n=iut();
	for (rr int i=1;i<=n;++i) a[i]=iut();
	for (rr int i=2;i<=n;++i){
		rr int x=iut();
		e[++k]=(node){i,as[x]},as[x]=k;
	}
	dfs(1,0);
	return !printf("%d",dp[1].size());
} 
posted @ 2021-03-05 18:49  lemondinosaur  阅读(58)  评论(0编辑  收藏  举报