[BZOJ3631]JLOI2014松鼠的新家|樹上差分

看起來是一個裸的樹鏈剖分，其實不用那麼麻煩。。給每個點一個標記表示它到root上所有點權值+多少，把每一次將一條鏈上權值+1的操作差分就相當於把L,R標記+1，把lca和fa[lca]標記-1，最後把標記往上傳就行了。。

#include<cstdio>
#include<iostream>
#define N 300005
using namespace std;
struct edge{
	int e,xu,next;
}ed[N*4];
int n,i,ne=0,s,e,a[N],A[N],h[N],d[N],pre[N],lca[N],fa[N],u[N];
int getfa(int x){return x==fa[x]?x:fa[x]=getfa(fa[x]);}
void adde(int s,int e)
{
	ed[++ne].e=e;ed[ne].next=a[s];a[s]=ne;
}
void addq(int s,int e,int xu)
{
	ed[++ne].e=e;ed[ne].xu=xu;
	ed[ne].next=h[s];h[s]=ne;
}
void dfs(int x)
{
	fa[x]=x;
	for (int j=a[x];j;j=ed[j].next)
		if (ed[j].e!=pre[x])
		{
			pre[ed[j].e]=x;
			dfs(ed[j].e);
			fa[ed[j].e]=x;
			d[x]+=d[ed[j].e];
		}
	u[x]=1;
	for (int j=h[x];j;j=ed[j].next)
		if (u[ed[j].e]) lca[ed[j].xu]=getfa(ed[j].e);
}
void dp(int x)
{
	for (int j=a[x];j;j=ed[j].next)
		if (ed[j].e!=pre[x]) 
			dfs(ed[j].e),d[x]+=d[ed[j].e];
}
int main()
{
	scanf("%d",&n);
	for (i=1;i<=n;i++) a[i]=h[i]=0,fa[i]=i,u[i]=0,d[i]=0;
	for (i=1;i<=n;i++) 
	{
		scanf("%d",&A[i]);
		if (i!=1)addq(A[i],A[i-1],i-1);addq(A[i-1],A[i],i-1);
	}
	for (i=1;i<n;i++) scanf("%d%d",&s,&e),adde(s,e),adde(e,s);
	pre[A[1]]=0;
	dfs(A[1]);
	for (i=1;i<n;i++)
		d[A[i]]++,d[A[i+1]]++,d[lca[i]]--,d[pre[lca[i]]]--;	
	dp(A[1]);
	for (i=2;i<=n;i++) d[A[i]]--;
	for (i=1;i<=n;i++) printf("%d\n",d[i]);
}