NOIP 2016 D1T2 天天爱跑步

胡雨菲菲

小c同学认为跑步非常有趣,于是决定制作一款叫做《天天爱跑步》的游戏。天天爱跑步是一个养成类游戏,需要
玩家每天按时上线,完成打卡任务。这个游戏的地图可以看作一一棵包含 N个结点和N-1 条边的树, 每条边连接两
个结点,且任意两个结点存在一条路径互相可达。树上结点编号为从1到N的连续正整数。现在有个玩家,第个玩家的
起点为Si ,终点为Ti  。每天打卡任务开始时,所有玩家在第0秒同时从自己的起点出发, 以每秒跑一条边的速度,
不间断地沿着最短路径向着自己的终点跑去, 跑到终点后该玩家就算完成了打卡任务。 (由于地图是一棵树, 所以
每个人的路径是唯一的)小C想知道游戏的活跃度, 所以在每个结点上都放置了一个观察员。 在结点的观察员会选
择在第Wj秒观察玩家, 一个玩家能被这个观察员观察到当且仅当该玩家在第Wj秒也理到达了结点J  。 小C想知道
每个观察员会观察到多少人?注意: 我们认为一个玩家到达自己的终点后该玩家就会结束游戏, 他不能等待一 段时
间后再被观察员观察到。 即对于把结点J作为终点的玩家: 若他在第Wj秒重到达终点,则在结点J的观察员不能观察
到该玩家;若他正好在第Wj秒到达终点,则在结点的观察员可以观察到这个玩家。

Input第一行有两个整数N和M 。其中N代表树的结点数量, 同时也是观察员的数量, M代表玩家的数量。
接下来n-1 行每行两个整数U和V ,表示结点U 到结点V 有一条边。
接下来一行N 个整数,其中第个整数为Wj , 表示结点出现观察员的时间。
接下来 M行,每行两个整数Si和Ti,表示一个玩家的起点和终点。
对于所有的数据,保证 。
1<=Si,Ti<=N,0<=Wj<=N


Output输出1行N 个整数,第个整数表示结点的观察员可以观察到多少人。


Sample Input6 3
2 3
1 2
1 4
4 5
4 6
0 2 5 1 2 3
1 5
1 3
2 6
Sample Output1 2 1 0 1

胡雨菲菲

1. #include <algorithm>
   
2. #include <cstdio>
   
3. using namespace std;
   
4. const int N=300005;
   
5. int f[N][20],to[N*5],ne[N*5],fi[N],dep[N],fir[N],v[N*5],w[N],
   
6.     cnt1[N],cnt2[N],ans[N],f1[N*2],f2[N*2],tot=0,cnt=0,n,m;
   
7. void add(int x,int y){
   
8.     to[++tot]=y;ne[tot]=fi[x];fi[x]=tot;
   
9. }
   
10. void ADD(int x,int y,int z){
    
11.     to[++tot]=y;v[tot]=z;ne[tot]=fir[x];fir[x]=tot;
    
12. }
    
13. void dfs(int x){
    
14.     dep[x]=dep[f[x][0]]+1;
    
15.     for (int i=f[x][0],j=0;f[i][j];i=f[i][j++]) f[x][j+1]=f[i][j];
    
16.     for (int i=fi[x];i;i=ne[i])
    
17.         if (to[i]!=f[x][0]) f[to[i]][0]=x,dfs(to[i]);
    
18. }
    
19. int lca(int x,int y){
    
20.     if (dep[x]<dep[y]) std::swap(x,y);
    
21.     for (int i=18;dep[x]>dep[y];x=f[x][i])
    
22.         for (;i && dep[f[x][i]]<dep[y];i--);
    
23.     for (int i=18;x!=y;x=f[x][i],y=f[y][i])
    
24.         for (;i && f[x][i]==f[y][i];i--);
    
25.     return x;
    
26. }
    
27. void insert(int x){
    
28.     for (int i=fir[x];i;i=ne[i])
    
29.     if (!v[i]){
    
30.         cnt1[f1[to[i]-dep[x]+N]]--;
    
31.         cnt2[f2[to[i]+dep[x]]]--;
    
32.     }
    
33.     else if (v[i]==1) cnt2[f2[to[i]]]++;
    
34.     else cnt1[f1[to[i]+N]]++;
    
35. }
    
36. void dfs2(int x){
    
37.     int d1=w[x]-dep[x]+N,d2=w[x]+dep[x],y=f1[d1],z=f2[d2];
    
38.     f1[d1]=f2[d2]=x;
    
39.     for (int i=fi[x];i;i=ne[i])
    
40.         if (to[i]!=f[x][0]) dfs2(to[i]);
    
41.     insert(x);
    
42.     ans[x]+=cnt1[x]+cnt2[x];
    
43.     cnt1[y]+=cnt1[x];cnt2[z]+=cnt2[x];
    
44.     f1[d1]=y;f2[d2]=z;
    
45. }
    
46. int main(){
    
47.     scanf("%d%d\n",&n,&m);
    
48.     for (int i=1;i<n;i++){
    
49.         int x,y;
    
50.         scanf("%d%d\n",&x,&y);
    
51.         add(x,y);add(y,x);
    
52.     }
    
53.     dfs(1);
    
54.     for (int i=1;i<=n;i++) scanf("%d",&w[i]);
    
55.     for (int i=1;i<=m;i++){
    
56.         int x,y,z;
    
57.         scanf("%d%d\n",&x,&y);
    
58.         z=lca(x,y);
    
59.         if (dep[x]-dep[z]==w[z]) ans[z]++;
    
60.         ADD(z,dep[x]-dep[z],0);
    
61.         ADD(x,dep[x],1);
    
62.         ADD(y,dep[x]-dep[z]*2,-1);
    
63.     }
    
64.     dfs2(1);
    
65.     for (int i=1;i<=n;i++) printf("%d ",ans[i]);
    
66.     return 0;
    
67. }

复制代码