题目大意:
每次对一条链进行区间修改,最后询问每个点的点权
树剖裸题,似乎dfs序也可以做?有时间再补估计不会写了
dp是类似差分的思路
//树剖
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=300000+10;
typedef long long ll;
int last[N],len=0,l[N],r[N],
ls[N<<2],rs[N<<2];ll tag[N<<2],sum[N<<2],a[N];
int dep[N],s[N],pos[N],bl[N],f[N],size=0,
n,q;
struct Edge{int to,next;Edge(int to=0,int next=0):to(to),next(next){}}e[N<<1];
void add_edge(int u,int v){e[++len]=Edge(v,last[u]);last[u]=len;}
void dfs1(int u)
{
s[u]=1;
for(int i=last[u];i;i=e[i].next) {
int id=e[i].to;
if(id==f[u])continue;
dep[id]=dep[u]+1;f[id]=u;
dfs1(id);
s[u]+=s[id];
}
}
void dfs2(int u,int chain)
{
bl[u]=chain;pos[u]=++size;l[u]=r[u]=size;
int k=0;
for(int i=last[u];i;i=e[i].next) {
int id=e[i].to;
if(dep[id]>dep[u]&&s[id]>s[k])k=id;
}
if(k==0)return;
dfs2(k,chain);
for(int i=last[u];i;i=e[i].next) {
int id=e[i].to;
if(dep[id]>dep[u]&&id!=k)dfs2(id,id);
}r[u]=size;
}
void build(int k,int l,int r)
{
ls[k]=l;rs[k]=r;sum[k]=tag[k]=0;
if(l==r)return;
int mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
}
void pushdown(int k)
{
int len=(rs[k]-ls[k]+1);
tag[k<<1]+=tag[k];
tag[k<<1|1]+=tag[k];
sum[k<<1]+=(len-(len>>1))*tag[k];
sum[k<<1|1]+=(len>>1)*tag[k];
tag[k]=0;
}
void change(int k,int ql,int qr,ll delta)
{
if(ql==ls[k]&&qr==rs[k]){sum[k]+=delta*(rs[k]-ls[k]+1);tag[k]+=delta;return;}
if(tag[k])pushdown(k);
int mid=(ls[k]+rs[k])>>1;
if(qr<=mid)change(k<<1,ql,qr,delta);
else if(ql>mid)change(k<<1|1,ql,qr,delta);
else {
change(k<<1,ql,mid,delta);
change(k<<1|1,mid+1,qr,delta);
}
sum[k]=sum[k<<1]+sum[k<<1|1];
}
ll querysum(int k,int ql,int qr)
{
if(ls[k]==ql&&rs[k]==qr){return sum[k];}
if(tag[k])pushdown(k);
int mid=(ls[k]+rs[k])>>1;
if(qr<=mid)return querysum(k<<1,ql,qr);
else if(ql>mid)return querysum(k<<1|1,ql,qr);
else return (querysum(k<<1,ql,mid)+querysum(k<<1|1,mid+1,qr));
}
void solveadd(int x,int y,ll val)
{
while(bl[x]!=bl[y]) {
if(dep[bl[x]]<dep[bl[y]])swap(x,y);
change(1,pos[bl[x]],pos[x],val);
x=f[bl[x]];
}if(dep[x]>dep[y])swap(x,y);
change(1,pos[x],pos[y],val);
}
int main()
{
scanf("%d",&n);
fo(i,1,n)scanf("%lld",&a[i]);
for(int x,y,i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
add_edge(x,y);add_edge(y,x);
}
dfs1(1);dfs2(1,1);
build(1,1,n);
solveadd(a[1],a[1],1);
for(int i=1;i<n;i++)
{solveadd(a[i],a[i+1],1);solveadd(a[i],a[i],-1);}
solveadd(a[n],a[n],-1);
fo(i,1,n)printf("%lld\n",querysum(1,pos[i],pos[i]));
return 0;
}