JZOJ 3338. 【NOI2013模拟】法法塔的奖励【线段树】

---

题目：

传送门

---

题意：

在一棵树上，每个点都有一个权值，现在问我们以$i(i\in 1—n)$为根时，其子树到其的最长不下降序列的长度是多少，且要求一定要选$i$

---

分析：

我们用线段树来计算每个叶子节点向上的最长不下降子序列的长度，但对于一个根节点来说，我们会有多个子节点，也就是说我们只能选择其中一种子序列，所以我们通过合并线段树的方式来模拟这种效果

---

代码：

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define LL long long 
#define LZX IMU
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}   
struct node{
	int to,next;
}e[100005];
int ls[100005],cnt=0;
struct imu{
	int w,l,r;
}tree[2000005];
void add(int x,int y)
{
	e[cnt]=(node){y,ls[x]};
	ls[x]=cnt++;
	return;
}
int w[100005],n=read(),u=read(),ans[100005],root[100005];
void change(int &k,int l,int r,int a,int b)
{
	if(!k) k=++cnt;
	if(l==r) {tree[k].w=max(tree[k].w,b);return;}
	int mid=(l+r)>>1;
	if(a<=mid) change(tree[k].l,l,mid,a,b);
	else change(tree[k].r,mid+1,r,a,b);
	tree[k].w=max(tree[tree[k].l].w,tree[tree[k].r].w);
	return;
}
int query(int k,int l,int r,int a,int b)
{
	if(!k) return 0;
	if(l==a&&r==b) return tree[k].w;
	int mid=(l+r)>>1;
	if(b<=mid) return query(tree[k].l,l,mid,a,b);
	else if(a>mid) return query(tree[k].r,mid+1,r,a,b);
	else return max(query(tree[k].l,l,mid,a,mid),query(tree[k].r,mid+1,r,mid+1,b));
}
int merge(int a,int b,int l,int r)
{
	if(!a||!b) return a+b;
	if(l==r) {tree[a].w=max(tree[a].w,tree[b].w);return a;}
	int mid=(l+r)>>1;
	tree[a].l=merge(tree[a].l,tree[b].l,l,mid);
	tree[a].r=merge(tree[a].r,tree[b].r,mid+1,r);
	tree[a].w=max(tree[a].w,tree[b].w);
	return a;
}
void work(int k)
{
	int king=0;
	for(int i=ls[k];~i;i=e[i].next)
	{
		work(e[i].to);
		king=merge(king,root[e[i].to],1,n);
	}
	ans[k]=query(king,1,n,1,w[k])+1;
	change(king,1,n,w[k],ans[k]);
	root[k]=king;
	return;
}
int main()
{
	memset(ls,-1,sizeof(ls));
	for(int i=2;i<=n;i++)
	{
		u=read();
		add(u,i);
	}
	for(int i=1;i<=n;i++) w[i]=read();
	cnt=0;
	work(1);
	for(int i=1;i<=n;i++) printf("%d ",ans[i]);
	return 0;
}