C++——NOIP2016提高组day1 t2——天天爱跑步

题目描述

小 C 同学认为跑步非常有趣，于是决定制作一款叫做《天天爱跑步》的游戏。《天天爱跑步》是一个养成类游戏，需要玩家每天按时上线，完成打卡任务。

这个游戏的地图可以看作一一棵包含 n 个结点和 n-1 条边的树， 每条边连接两个结点，且任意两个结点存在一条路径互相可达。树上结点编号为从 1 到 n 的连续正整数。

现在有 m 个玩家，第 i 个玩家的起点为 Si ，终点为 Ti 。每天打卡任务开始时，所有玩家在第 0 秒同时从自己的起点出发，以每秒跑一条边的速度，不间断地沿着最短路径向着自己的终点跑去， 跑到终点后该玩家就算完成了打卡任务。 (由于地图是一棵树， 所以每个人的路径是唯一的)

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

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

输入格式

第一行有两个整数 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 的观察员可以观察到多少人。

样例数据 1

输入

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

样例数据 2

输入

5 3 
1 2 
2 3 
2 4 
1 5 
0 1 0 3 0 
3 1 
1 4 
5 5

输出

1 2 1 0 1

备注

【样例1说明】

• 对于 1 号点，W1 = 0 ，故只有起点为 1 号点的玩家才会被观察到，所以玩家 1 和玩家 2 被观察到，共有 2 人被观察到。
• 对于 2 号点，没有玩家在第 2 秒时在此结点，共 0 人被观察到。
• 对于 3 号点，没有玩家在第 5 秒时在此结点，共 0 人被观察到。
• 对于 4 号点，玩家 1 被观察到，共 1 人被观察到。
• 对于 5 号点，玩家 1 被观察到，共 1 人被观察到。
• 对于 6 号点，玩家 3 被观察到，共 1 人被观察到。

【数据规模与约定】

    

#include<cstdio>#include<vector>#include<algorithm>using namespace std;int n,m,cnt;int first[300000],deep[300000],worth[300000],ans[300000],next[600000],to[600000],num[1200000],jump[300000][20];struct node{int s,t;};vector <node> dian[300000];inline int readint()  {      int i=0;      char ch;      for(ch=getchar();ch<'0'||ch>'9';ch=getchar());      for(;ch>='0' && ch<='9';ch=getchar())          i=(i<<3)+(i<<1)+ch-'0';      return i;  }inline void put()    {        int num=0;char c[10];  for(int i=1;i<=n;++i){do{c[++num]=(ans[i]%10)+48;ans[i]/=10; }while(ans[i]);    while(num) putchar(c[num--]);      putchar(' ');  }}void dfs(int x){for(int i=1;i<=18;++i){jump[x][i]=jump[jump[x][i-1]][i-1];if(!jump[x][i]) break;}for(int i=first[x];i;i=next[i])if(to[i]!=jump[x][0]){deep[to[i]]=deep[x]+1;jump[to[i]][0]=x;dfs(to[i]);}}int lca(int a,int b){if(deep[a]<deep[b]) swap(a,b);for(int i=18,k=deep[a]-deep[b];i>=0;--i)if(k&(1<<i))a=jump[a][i];if(a==b) return a;for(int i=18;i>=0;--i)if(jump[a][i]!=jump[b][i]){a=jump[a][i];b=jump[b][i];}return jump[a][0];}void work(int x){int last=num[deep[x]+worth[x]]+num[worth[x]-deep[x]+3*n+1];for(int i=0,j=dian[x].size();i<j;++i){node u=dian[x][i];num[u.s]+=u.t;}for(int i=first[x];i;i=next[i])if(to[i]!=jump[x][0])work(to[i]);ans[x]=num[deep[x]+worth[x]]+num[worth[x]-deep[x]+3*n+1]-last;}int main(){n=readint();m=readint();int x,y,k;for(int i=1;i<n;++i){x=readint();y=readint();to[++cnt]=y;next[cnt]=first[x];first[x]=cnt;to[++cnt]=x;next[cnt]=first[y];first[y]=cnt;}dfs(1);for(int i=1;i<=n;++i) worth[i]=readint();node u;for(int i=1;i<=m;++i){x=readint();y=readint();k=lca(x,y);u.s=deep[x];u.t=1;dian[x].push_back(u);u.t=-1;dian[jump[k][0]].push_back(u);u.s=deep[x]-deep[k]*2+3*n+1;u.t=1;dian[y].push_back(u);u.t=-1;dian[k].push_back(u);}work(1);put();return 0;}