【NOIP2016提高组复赛day2】天天爱跑步

题目

小 C 同学认为跑步非常有趣，于是决定制作一款叫做《天天爱跑步》的游戏。 《天天爱跑步》是一个养成类游戏，需要玩家每天按时上线，完成打卡任务。 这个游戏的地图可以看作一棵包含 n 个结点和 n − 1 条边的树，每条边连接两个结点，且任意两个结点存在一条路径互相可达。树上结点编号为从 1 到 n 的连续正整数。 现在有 m 个玩家，第 i 个玩家的起点为 Si ，终点为 Ti 。每天打卡任务开始时，所 有玩家在第 0 秒同时从自己的起点出发，以每秒跑一条边的速度，不间断地沿着 最短 路径向着自己的终点跑去，跑到终点后该玩家就算完成了打卡任务。（由于地图是一棵树，所以每个人的路径是唯一的）

小 C 想知道游戏的活跃度，所以在每个结点上都放置了一个观察员。在结点 j 的 观 察员会选择在第 Wj 秒观察玩家，一个玩家能被这个观察员观察到当且仅当该玩家 在第 Wj 秒也正好到达了结点 j 。小 C 想知道每个观察员会观察到多少人？

注意：我们认为一个玩家到达自己的终点后该玩家就会结束游戏，他不能等待一 段时间后再被观察员观察到。即对于把结点 j 作为终点的玩家：若他在第 Wj 秒前到达 终点，则在结点 j 的观察员不能观察到该玩家；若他正好在第 Wj 秒到达终点，则在结 点 j 的观察员可以观察到这个玩家。

分析

假设有一条路径(x->y)，最近公共祖先lca，

考虑i这个点是否能观察到这个玩家，

才成两条路径，分两种情况，

一、（x->lca）

如果要使i可以观察到，

i一定在（x->lca）上

那么deep[i]+w[i]=deep[x]。

建一个桶，

在x入栈时将deep[x]加入到桶中，

在lca退栈时，将deep[x]踢掉。

二、(lca的某个儿子且为j的祖先->y)

如果要使i可以观察到，

i一定在(lca的某个儿子且为j的祖先->y)上

那么deep[i]-w[i]=deep[y]-t（t=deep[x]+deep[y]-deep[lca]*2）。

再建一个桶，

同样

在y入栈时将deep[y]-t加入到同中，

在lca的某个儿子且为j的祖先退栈时，将deep[y]-t踢掉。

因为会算重，所以将做完以i为根的子树后的值-i进栈时的值就是i的答案。

#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
using namespace std;
const int N=700005;
int deep
,next
,last
,to
,deep1
,n,m,w
,tot,ans
,a
[2],g
[20],sum,mxd,t[N*4],t1[N*4];
int next1
,last1
,to1
,tot1;
int next2
,last2
,to2
,tot2;
int next3
,last3
,to3
,tot3;
int next4
,last4
,to4
,tot4;
int bj(int x,int y){next[++tot]=last[x];last[x]=tot;to[tot]=y;}
int bj1(int x,int y){next1[++tot1]=last1[x];last1[x]=tot1;to1[tot1]=y;}
int bj2(int x,int y){next2[++tot2]=last2[x];last2[x]=tot2;to2[tot2]=y;}
int bj3(int x,int y){next3[++tot3]=last3[x];last3[x]=tot3;to3[tot3]=y;}
int bj4(int x,int y){next4[++tot4]=last4[x];last4[x]=tot4;to4[tot4]=y;}
int dg(int x)
{
for(int i=last[x];i;i=next[i])
{
int j=to[i];
if(j!=g[x][0])
{
g[j][0]=x;
deep[j]=deep[x]+1;
dg(j);
}
}
}
int lca(int x,int y)
{
if(deep[x]>deep[y])
{
int o=x;
x=y;
y=o;
}
for(int j=log2(n);j>=0;j--)
{
if(deep[g[y][j]]>=deep[x])
y=g[y][j];
}
for(int j=log2(n);j>=0;j--)
{
if(g[y][j]!=g[x][j])
y=g[y][j],x=g[x][j];
}
if(x!=y) y=g[y][0],x=g[x][0];
return x;
}
int dg1(int x)
{
int xx=t[deep[x]+w[x]]+t1[deep[x]-w[x]+N];
for(int i=last1[x];i;i=next1[i]) t[to1[i]]++;
for(int i=last3[x];i;i=next3[i]) t1[to3[i]+N]++;
for(int i=last[x];i;i=next[i])
if(to[i]!=g[x][0]) dg1(to[i]);
ans[x]=t[deep[x]+w[x]]+t1[deep[x]-w[x]+N]-xx;
for(int i=last2[x];i;i=next2[i]) t[to2[i]]--;
for(int i=last4[x];i;i=next4[i]) t1[to4[i]+N]--;
}
int main()
{
freopen("running.in","r",stdin);
freopen("running.out","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
bj(x,y);
bj(y,x);
}
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
deep[1]=1;
dg(1);
for(int j=1;j<=log2(n);j++)
for(int i=1;i<=n;i++)
g[i][j]=g[g[i][j-1]][j-1];
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
int top=lca(x,y),t=deep[x]+deep[y]-2*deep[top];
if(top==x)
{
bj3(y,deep[y]-t);
bj4(x,deep[y]-t);
}
else
if(top==y)
{
bj1(x,deep[x]);
bj2(y,deep[x]);
}
else
{
bj1(x,deep[x]);
bj2(top,deep[x]);
bj3(y,deep[y]-t);
for(int j=last[top];j;j=next[j])
{
if(to[j]!=g[top][0] && lca(to[j],y)==to[j])
{
bj4(to[j],deep[y]-t);
break;
}
}
}
}
dg1(1);
for(int i=1;i<=n;i++) printf("%d ",ans[i]);
return 0;
}