P3258 [JLOI2014]松鼠的新家

题目描述
松鼠的新家是一棵树，前几天刚刚装修了新家，新家有n个房间，并且有n-1根树枝连接，每个房间都可以相互到达，且俩个房间之间的路线都是唯一的。天哪，他居然真的住在”树“上。

松鼠想邀请小熊维尼前来参观，并且还指定一份参观指南，他希望维尼能够按照他的指南顺序，先去a1，再去a2，…，最后到an，去参观新家。可是这样会导致维尼重复走很多房间，懒惰的维尼不停地推辞。可是松鼠告诉他，每走到一个房间，他就可以从房间拿一块糖果吃。

维尼是个馋家伙，立马就答应了。现在松鼠希望知道为了保证维尼有糖果吃，他需要在每一个房间各放至少多少个糖果。

因为松鼠参观指南上的最后一个房间an是餐厅，餐厅里他准备了丰盛的大餐，所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。

输入输出格式
输入格式：
第一行一个整数n，表示房间个数第二行n个整数，依次描述a1-an

接下来n-1行，每行两个整数x，y，表示标号x和y的两个房间之间有树枝相连。

输出格式：
一共n行，第i行输出标号为i的房间至少需要放多少个糖果，才能让维尼有糖果吃。

输入输出样例
输入样例#1：
5
1 4 5 3 2
1 2
2 4
2 3
4 5

输出样例#1：
1
2
1
2
1

说明
2<= n <=300000

【题目分析】
根据树的性质，每一次两点间的走动，都是唯一路径，这条路径可以通过LCA得到：

这样一来，每一条路径我们都能搞出来，对于每条路径，显然只要将路径上的点全部加1即可，最后经过所有路径的修改，将每个点的权值输出，于是不难想到树上差分。将根节点到每一个节点的一条链作为差分的序列，例如从3号点到8号点，就先拆成[3,1]、[8,1]两条链分别进行差分即可，还有个细节，这题规定了an是不需要加1的，其实是为了打代码方便，因为每次差分时有意漏掉终点，就刚好对齐，无需处理最后一个终点了。

#include<cstdio>
#include<cstring>
#define maxn 300039
using namespace std;
struct NODE{
	int to, nxt;
}e[maxn<<1];
void swap(int& x, int& y){int tmp = x;x = y;y = tmp;}
int tot, head[maxn], lg[maxn], pre[maxn];
void add(int from, int to){
	e[++tot].to = to;
	e[tot].nxt = head[from];
	head[from] = tot;
}
int deep[maxn], l[maxn][20];
void dfs(int num, int fa){
	pre[num] = fa;
	deep[num] = deep[fa]+1;
	l[num][0] = fa;
	for(int i = 1; (1<<i) <= deep[num]; i++)l[num][i] = l[l[num][i-1]][i-1];
	for(int i = head[num]; i; i = e[i].nxt)if(e[i].to!=fa)dfs(e[i].to, num);
}
int lca(int x, int y){
	if(deep[x]<deep[y])swap(x, y);
	while(deep[x]>deep[y]){x = l[x][lg[deep[x]-deep[y]]-1];}
	if(x==y)return x;
	for(int i = lg[deep[x]]; i >= 0; i--)
		if(l[x][i]!=l[y][i])x = l[x][i], y = l[y][i];
	return l[x][0];
}
int n, a[maxn], f[maxn];
void cal(int u, int fa){
	for(int i = head[u]; i!=-1; i=e[i].nxt){
		int v = e[i].to;
		if(v!=fa){
			cal(v, u);
			f[u] += f[v];
		}
	}
}
int main(){
	//freopen("1.in", "r", stdin);
	scanf("%d", &n);
	memset(head, -1, sizeof(head));
	for(int i = 1; i < n+1; i++)lg[i] = lg[i-1] + (i==(1<<lg[i-1]));
	int g, x, y;
	for(int i = 1; i < n+1; i++)scanf("%d", a+i);
	for(int i = 1; i < n; i++){
		scanf("%d%d", &x, &y);
		add(x, y);
		add(y, x);
	}
	dfs(1, 0);
	for(int i = 1; i < n; i++){
		g = lca(a[i], a[i+1]);
		if(g==a[i+1]){
			f[a[i]]++;
			f[a[i+1]]--;
		}
		else if(g==a[i]){
			f[pre[a[i+1]]]++;
			f[pre[a[i]]]--;
		}
		else{
			f[a[i]]++;
			f[g]--;
			f[pre[a[i+1]]]++;
			f[g]--;
			f[g]++;
			f[pre[g]]--;
		}
	}
	cal(1, 0);
	for(int i = 1; i < n+1; i++)
		printf("%d\n", f[i]);
	return 0;
}