传送门
在一棵树上,每个点都有一个权值,现在问我们以 i ( i ∈ 1 — n ) i(i\in1—n) i(i∈1—n)为根时,其子树到其的最长不下降序列的长度是多少,且要求一定要选 i i i
我们用线段树来计算每个叶子节点向上的最长不下降子序列的长度,但对于一个根节点来说,我们会有多个子节点,也就是说我们只能选择其中一种子序列,所以我们通过合并线段树的方式来模拟这种效果
#include
#include
#include
#include
#include
#include
#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;
}