问题 G: 天天爱跑步
时间限制: 2 Sec 内存限制: 512 MB
提交: 24 解决: 20
[提交][状态][讨论版]
题目描述
小C同学认为跑步非常有趣，于是决定制作一款叫做《天天爱跑步》的游戏。《天天爱跑步》是一个养成类游戏，需要玩家每天按时上线，完成打卡任务。
这个游戏的地图可以看作一棵包含n个结点和n-1条边的树，每条边连接两个结点，且任意两个结点存在一条路径互相可达。树上结点编号为从1到n的连续正整数。
现在有m个玩家，第i个玩家的起点为Si，终点为Ti。每天打卡任务开始时，所有玩家在第0秒同时从自己的起点出发，以每秒跑一条边的速度，不间断地沿着最短路径向着自己的终点跑去，跑到终点后该玩家就算完成了打卡任务。（由于地图是一棵树，所以每个人的路径是唯一的）
小C想知道游戏的活跃度, 所以在每个结点上都放置了一个观察员。 在结点的观察员会选择在第Wj秒观察玩家, 一个玩家能被这个观察员观察到当且仅当该玩家在第Wj秒也理到达了结点J 。 小C想知道每个观察员会观察到多少人?
注意: 我们认为一个玩家到达自己的终点后该玩家就会结束游戏, 他不能等待一 段时间后再被观察员观察到。 即对于把结点J作为终点的玩家: 若他在第Wj秒重到达终点,则在结点J的观察员不能观察到该玩家;若他正好在第Wj秒到达终点,则在结点的观察员可以观察到这个玩家。
输入
第一行有两个整数n和m。其中n代表树的结点数量，同时也是观察员的数量，m代表玩家的数量。
接下来n-1行每行两个整数u和v，表示结点u到结点v有一条边。
接下来一行n个整数，其中第j个整数为Wj，表示结点j出现观察员的时间。
接下来m行,每行两个整数Si和Ti，表示一个玩家的起点和终点。
对于所有的数据，保证1≤Si,Ti≤n，0≤ Wj ≤n。
输出
输出1行n个整数,第j个整数表示结点j的观察员可以观察到多少人。
样例输入
6 3
2 3
1 2
1 4
4 5
4 6
0 2 5 1 2 3
1 5
1 3
2 6
样例输出
2 0 0 1 1 1
很显然要用到LCA，但正解比LCA多很多。
一条路该分成两段，起点x到LCA，LCA到终点y。这里就要分别考虑了。对于第一段，对于其路径上的点i，t[i]==dep[x]-dep[i]时，就是dep[x]==t[i]+dep[i]。所以对于起点，就看对于点i有多少人起步于t[i]+dep[i]就好了。 那么对于第二段，len-(dep[y]-dep[i])==t[i]，就是t[i]-dep[i]==len-depy，以同样的方式统计就行了。
这只是主体思路，离实现还差很远。因为同一深度不同路径会对答案产生影响，所以可以再来一发深搜。这样判断就变得较容易了。 可以开两个桶记录先前所述的第一段和第二段对答案的贡献，但之前的会产生影响，可以这样考虑，记录搜自己和子节点之前的值，用搜完的之后的值减去他，就是自己子树要过自己的数量了。。。而为了消除LCA对答案的影响，回溯回来时，要分别把以他为LCA的起点和终点的点在两个桶中分别减减
在这里搜答案时，我学（zhao）习(chao)神犇ljm的做法，把每个点向他所能对应的答案练了一条边。方便在dfs时查找答案，或是在回溯时减去。。
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<iostream>
#define N 300000
using namespace std;
int read()
{
int sum=0,f=1;char x=getchar();
while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}
while(x>='0'&&x<='9'){sum=(sum<<1)+(sum<<3)+x-'0';x=getchar();}
return sum*f;
}
int n,m,adj[N],f[N],vis[N],dep[N],anc[N],ans[N];
int tim[N],A[N],sum[N],up[N*2],down[N*2];
int e,e1,e2,e3,e4;
int adj1[N*2],adj2[N*2],adj3[N*2],head[N];
struct Q{int v,next,id;}q[N*2];
struct road{int v,next;}lu[N*2],lu1[N*2],lu2[N*2],lu3[N*2];
void add(int u,int v){lu[++e].v=v;lu[e].next=adj[u];adj[u]=e;}
void add1(int u,int v){lu1[++e1].v=v;lu1[e1].next=adj1[u];adj1[u]=e1;}
void add2(int u,int v){lu2[++e2].v=v;lu2[e2].next=adj2[u];adj2[u]=e2;}
void add3(int u,int v){lu3[++e3].v=v;lu3[e3].next=adj3[u];adj3[u]=e3;}
void add_q(int u,int v,int id){q[e4].v=v;q[e4].id=id;q[e4].next=head[u];head[u]=e4++;}
int find(int x){return f[x]==x? x:f[x]=find(f[x]);}
void hb(int x,int y){int fx=find(x),fy=find(y);if(fx!=fy)f[fx]=fy;}
void dfs1(int x,int fa)
{
dep[x]=dep[fa]+1;anc[x]=x;vis[x]=1;
for(int i=adj[x];i;i=lu[i].next)
{
int to=lu[i].v;
if(vis[to])continue;
dfs1(to,x);
hb(x,to);anc[find(x)]=x;
}
for(int i=head[x];i!=-1;i=q[i].next)
{
int to=q[i].v;
if(vis[to])ans[q[i].id]=anc[find(to)];
}
}
void dfs2(int x,int y)
{
int t1=up[dep[x]+tim[x]],t2=down[dep[x]-tim[x]+N];
up[dep[x]]+=A[x];
for(int i=adj1[x];i;i=lu1[i].next)
++down[lu1[i].v+N];
for(int i=adj[x];i;i=lu[i].next)
if(lu[i].v!=y)dfs2(lu[i].v,x);
sum[x]=up[dep[x]+tim[x]]+down[dep[x]-tim[x]+N]-t1-t2;
for(int i=adj2[x];i;i=lu2[i].next)
{
int to=lu2[i].v;
up[to]--;
if(to==dep[x]+tim[x])
sum[x]--;
}
for(int i=adj3[x];i;i=lu3[i].next)
down[lu3[i].v+N]--;
}
int main()
{
// freopen("runninga.in","r",stdin);
// freopen("runninga.out","w",stdout);
int __size__=64<<20;
char *__p__=(char*)malloc(__size__)+__size__;
__asm__("movl %0, %%esp
"::"r"(__p__));
n=read();m=read();int x,y,a,b;
memset(head,-1,sizeof(head));
for(int i=1;i<=n;i++)f[i]=i;
for(int i=1;i<n;i++)x=read(),y=read(),add(x,y),add(y,x);
for(int i=1;i<=n;i++)tim[i]=read();
for(int i=0;i<m;i++)x=read(),y=read(),add_q(x,y,i),add_q(y,x,i);
dfs1(1,0);
for(int i=0;i<m;i++)
{
x=q[i<<1].v,y=q[i<<1|1].v;
a=dep[x]+dep[y]-dep[ans[i]]*2;b=dep[x]-a;
++A[y];add1(x,b);add2(ans[i],dep[y]);add3(ans[i],b);
}
dfs2(1,0);
for(int i=1;i<=n;i++)printf("%d ",sum[i]);
}