[BZOJ]4719: [Noip2016]天天爱跑步 LCA

Description

小c同学认为跑步非常有趣,于是决定制作一款叫做《天天爱跑步》的游戏。?天天爱跑步?是一个养成类游戏,需要玩家每天按时上线,完成打卡任务。这个游戏的地图可以看作一一棵包含 N个结点和N-1 条边的树, 每条边连接两个结点,且任意两个结点存在一条路径互相可达。树上结点编号为从1到N的连续正整数。现在有个玩家,第个玩家的起点为Si ,终点为Ti 。每天打卡任务开始时,所有玩家在第0秒同时从自己的起点出发, 以每秒跑一条边的速度,不间断地沿着最短路径向着自己的终点跑去, 跑到终点后该玩家就算完成了打卡任务。 (由于地图是一棵树, 所以每个人的路径是唯一的)小C想知道游戏的活跃度, 所以在每个结点上都放置了一个观察员。 在结点的观察员会选择在第Wj秒观察玩家, 一个玩家能被这个观察员观察到当且仅当该玩家在第Wj秒也理到达了结点J 。小C想知道每个观察员会观察到多少人?注意: 我们认为一个玩家到达自己的终点后该玩家就会结束游戏, 他不能等待一 段时间后再被观察员观察到。 即对于把结点J作为终点的玩家: 若他在第Wj秒重到达终点,则在结点J的观察员不能观察到该玩家;若他正好在第Wj秒到达终点,则在结点的观察员可以观察到这个玩家。
##题解：
这题太神了……网上一位大神的做法：对于一个玩家，起点为x，终点为y，lca为z，可以发现，x->z的路径上的每个点i，dep[i]+到达i的时刻为定值；同理，z->y的路径上的每个点j，dep[j]-到达j的时刻也为定值。所以我们可以把两条路径上的点某个值的出现次数+1（我语文不好也不知道怎么表述），然后问题就转化为询问每个点特定值出现的次数。我们可以这样解决这个问题：首先用差分的思想，路径加转化为两个点加，然后对于一个点，只要统计子树和就能得到答案，子树和可以这样做：递归子树后的和-递归子树前的和。问题就迎刃而解了。
##代码：

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=600010;
struct Edge{int y,next;}e[maxn*2];
int last[maxn],len=0;
void ins(int x,int y)
{
	int t=++len;
	e[t].y=y;e[t].next=last[x];last[x]=t;
}
int n,m,w[maxn],dep[maxn],fa[maxn][20];
vector<int>v1[maxn],v2[maxn],v3[maxn],v4[maxn];
void dfs(int x,int f)
{
	fa[x][0]=f;
	dep[x]=dep[f]+1;
	for(int i=1;(1<<i)<=dep[x];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].y;
		if(y==f)continue;
		dfs(y,x);
	}
}
int LCA(int x,int y)
{
	if(dep[x]<dep[y])swap(x,y);
	for(int i=19;i>=0;i--)
	if(dep[x]-dep[y]>=(1<<i))x=fa[x][i];
	if(x==y)return x;
	for(int i=19;i>=0;i--)
	if(dep[x]>=(1<<i)&&fa[x][i]!=fa[y][i])
	x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int ta[maxn],tb[maxn],ans[maxn];
void solve(int x)
{
	ans[x]=-ta[w[x]+dep[x]]-tb[w[x]-dep[x]+n];
	for(int i=0;i<v1[x].size();i++)
	ta[abs(v1[x][i])]+=v3[x][i];
	for(int i=0;i<v2[x].size();i++)
	tb[abs(v2[x][i])]+=v4[x][i];
	for(int i=last[x];i;i=e[i].next)
	if(e[i].y!=fa[x][0])solve(e[i].y);
	ans[x]+=ta[w[x]+dep[x]]+tb[w[x]-dep[x]+n];
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		ins(x,y);ins(y,x);
	}
	dep[0]=0;dfs(1,0);
	for(int i=1;i<=n;i++)scanf("%d",&w[i]);
	for(int i=1;i<=m;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		int z=LCA(x,y);
		v1[x].push_back(dep[x]);v1[fa[z][0]].push_back(-dep[x]);
		v2[y].push_back(dep[x]-2*dep[z]+n);v2[z].push_back(2*dep[z]-dep[x]-n);//+n防负数 
		v3[x].push_back(1);v3[fa[z][0]].push_back(-1);
		v4[y].push_back(1);v4[z].push_back(-1);
	}
	solve(1);
	for(int i=1;i<n;i++)
	printf("%d ",ans[i]);printf("%d",ans[n]);
}