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 人被观察到。
<
4000
li>对于 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;
}