【bzoj 4719】[Noip2016]天天爱跑步

[Noip2016]天天爱跑步

Time Limit: 40 Sec Memory Limit: 512 MB
Submit: 737 Solved: 241
[Submit][Status][Discuss]
Description

小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 Input

6 3

2 3

1 2

1 4

4 5

4 6

0 2 5 1 2 3

1 5

1 3

2 6
Sample Output

1 2 1 0 1
终于写了个正常的

#include<cstdio>
#include<cstring>

using namespace std;

const int MAXN = 309000;
const int MAXM = 309000;

#define cnt2(x) cntttt[(x)+MAXN]
#define cnt1(x) cntt[(x)]

int ti[MAXN];
int h[MAXN],to[MAXN<<1],nx[MAXN<<1],cnt;
int top[MAXN],fa[MAXN],ch[MAXN],sz[MAXN],dep[MAXN];
int len[MAXM],begin[MAXM],end[MAXM],lca[MAXM];
int adt1[MAXN],adt2[MAXN],des1[MAXN],des2[MAXN];
int nxb[MAXN<<2],vb[MAXN<<2],cntb;
int n,m;
int cntt[MAXN<<2],cntttt[MAXN<<2],ans[MAXN];

int read(){
    int rt=0,fl=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')fl=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){rt=rt*10+ch-'0';ch=getchar();}
    return rt*fl;
}

void add_edge(int _u,int _v){
    to[++cnt]=_v;nx[cnt]=h[_u];h[_u]=cnt;
}

void link(int _u,int _v){
    add_edge(_u,_v);add_edge(_v,_u);
}

void dfs1(int _rt,int _pr){
    fa[_rt]=_pr;sz[_rt]=1;dep[_rt]=dep[_pr]+1;
    for(int i=h[_rt];i;i=nx[i]){
        if(to[i] == _pr)continue;
        dfs1(to[i],_rt);
        sz[_rt]+=sz[to[i]];
        if(sz[to[i]]>sz[ch[_rt]])ch[_rt]=to[i];
    }
}

void dfs2(int _rt,int _tp){
    top[_rt]=_tp;
    if(ch[_rt]!=0)dfs2(ch[_rt],_tp);
    for(int i=h[_rt];i;i=nx[i]){
        if(to[i]==fa[_rt] || to[i]==ch[_rt])continue;
        dfs2(to[i],to[i]);
    }
}

int get_lca(int _u,int _v){
    while(top[_u]!=top[_v])
        dep[top[_u]] > dep[top[_v]] ? _u=fa[top[_u]] : _v=fa[top[_v]];
    return dep[_u]>dep[_v]?_v:_u;
}

void init_data(){
    n=read();m=read();
    for(int i=1;i<n;i++){
        int a=read(),b=read();
        link(a,b);
    }
    dfs1(1,0);dfs2(1,1);
    for(int i=1;i<=n;i++){
        ti[i]=read();
    }
    for(int i=1;i<=m;i++){
        begin[i]=read();
        end[i]=read();
        lca[i]=get_lca(begin[i],end[i]);
        len[i] = dep[begin[i]] - dep[lca[i]] + dep[end[i]] - dep[lca[i]];
    } 
}

void add_ins_tag_1(int _x,int _v){
    cntb++;
    nxb[cntb] = adt1[_x];
    adt1[_x] = cntb;
    vb[cntb] = _v;
}

void add_des_tag_1(int _x,int _v){
    cntb++;
    nxb[cntb] = des1[_x];
    des1[_x] = cntb;
    vb[cntb] = _v;
}

void add_ins_tag_2(int _x,int _v){
    cntb++;
    nxb[cntb] = adt2[_x];
    adt2[_x] = cntb;
    vb[cntb] = _v;
}

void add_des_tag_2(int _x,int _v){
    cntb++;
    nxb[cntb] = des2[_x];
    des2[_x] = cntb;
    vb[cntb] = _v;
}

void init_map(){
    for(int i=1;i<=m;i++){
        add_ins_tag_1(begin[i],dep[begin[i]]);
        add_des_tag_1(fa[lca[i]],dep[begin[i]]);
        add_ins_tag_2(end[i],dep[end[i]] - len[i]);
        add_des_tag_2(lca[i],dep[end[i]] - len[i]);
    }
}

void solve(int _x){
    ans[_x] = -cnt1(dep[_x]+ti[_x])-cnt2(dep[_x]-ti[_x]);
    for(int i=h[_x];i;i=nx[i]){
        if(to[i]==fa[_x])continue;
        solve(to[i]);
    }
    for(int i=adt1[_x];i;i=nxb[i])
        cnt1(vb[i])++;
    for(int i=des1[_x];i;i=nxb[i])
        cnt1(vb[i])--;
    for(int i=adt2[_x];i;i=nxb[i])
        cnt2(vb[i])++;
    for(int i=des2[_x];i;i=nxb[i])
        cnt2(vb[i])--;
    ans[_x] += cnt1(dep[_x]+ti[_x]) + cnt2(dep[_x]-ti[_x]);
}

int main(){
    init_data();
    init_map();
    solve(1);
    for(int i=1;i<n;i++){
        printf("%d ",ans[i]);
    }printf("%d",ans[n]);
    return 0;
}