題目描述

小c同學認為跑步非常有趣,於是決定制作一款叫做《天天愛跑步》的遊戲。《天天愛跑步》是一個養成類遊戲,需要玩家每天按時上線,完成打卡任務。

這個遊戲的地圖可以看作一一棵包含 n個結點和 n?1 條邊的樹, 每條邊連接兩個結點,且任意兩個結點存在一條路徑互相可達。樹上結點編號為從 1 到 n 的連續正整數。

現在有 m 個玩家,第 i 個玩家的起點為 Si? ,終點為 Ti? 。每天打卡任務開始時,所有玩家在第 0 秒同時從自己的起點出發, 以每秒跑一條邊的速度, 不間斷地沿著最短路徑向著自己的終點跑去, 跑到終點後該玩家就算完成了打卡任務。 (由於地圖是一棵樹, 所以每個人的路徑是唯一的)

小c想知道遊戲的活躍度, 所以在每個結點上都放置了一個觀察員。 在結點 j 的觀察員會選擇在第 Wj? 秒觀察玩家, 一個玩家能被這個觀察員觀察到當且僅當該玩家在第 Wj? 秒也理到達了結點j 。 小C想知道每個觀察員會觀察到多少人?

註意: 我們認為一個玩家到達自己的終點後該玩家就會結束遊戲, 他不能等待一 段時間後再被觀察員觀察到。 即對於把結點 jjj 作為終點的玩家: 若他在第 WjW_jWj? 秒前到達終點,則在結點 j的觀察員不能觀察到該玩家;若他正好在第 Wj 秒到達終點,則在結點 j 的觀察員可以觀察到這個玩家。

輸入輸出格式

輸入格式：

第一行有兩個整數 n 和 m 。其中 n 代表樹的結點數量, 同時也是觀察員的數量, m 代表玩家的數量。

接下來 n?1 行每行兩個整數 u 和 v ,表示結點 u 到結點 v 有一條邊。

接下來一行 n 個整數,其中第 j 個整數為 Wj? , 表示結點 j 出現觀察員的時間。

接下來 m 行,每行兩個整數 Si,和 Ti? ,表示一個玩家的起點和終點。

輸出格式：

輸出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

輸出樣例#1：

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

輸出樣例#2：

1 2 1 0 1

說明

題解：

樹上差分,暫且留坑

code:

//By Menteur_Hxy
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#define M(a,b) memset(a,(b),sizeof(a))
#define F(i,a,b) for(register int i=(a);i<=(b);i++)
#define E(i,u) for(register int i=head[u];i;i=nxt[i])
#define _E(i,u) for(register int i=_head[u];i;i=_nxt[i])
#define V(it,a) for(vector<int>::iterator it=a.begin();it!=a.end();it++)
using namespace std;

int rd() {
    int x=0,f=1; char c=getchar();
    while(!isdigit(c)) {if(c==‘-‘)f=-f; c=getchar();}
    while(isdigit(c)) x=(x<<3)+(x<<1)+c-48,c=getchar();
    return x*f; 
}
const int N=600010;
int n,m,cnt,md;
int nxt[N],head[N],to[N],w[N],fa[N],dep[N],num[N],vis[N];
int _cnt,_nxt[N],_head[N],_to[N];
struct players{int s,t,lca,dis;}pl[N];

int getf(int x) {return fa[x]==x?x:fa[x]=getf(fa[x]);}
void tarjan(int u,int pre) { 
    vis[u]=1; dep[u]=dep[pre]+1;
    _E(i,u) { int v=_to[i];   
        if(pl[v].s==u&&vis[pl[v].t]) pl[v].lca=getf(pl[v].t);
        if(pl[v].t==u&&vis[pl[v].s]) pl[v].lca=getf(pl[v].s);
        // cout<<pl[v].s<<" "<<pl[v].t<<" "<<pl[v].lca<<" "<<u<<endl;
    }
    E(i,u) { int v=to[i];
        if(v==pre) continue;
        tarjan(v,u);
        fa[v]=u;
        if(dep[v]>md) md=dep[v];
    }
}

vector <int> t1[N],t2[N],t3[N];
int tu[N],td[N<<1],ans[N];
void dfs(int u,int pre) {
    int a=td[dep[u]+w[u]],b=tu[dep[u]-w[u]+N];
    td[dep[u]]+=num[u]; 
    V(i,t1[u]) tu[*i+N]++;  
    E(i,u) if(to[i]!=pre) dfs(to[i],u);
    ans[u]=td[dep[u]+w[u]]-a+tu[dep[u]-w[u]+N]-b;
    V(i,t2[u]) {td[*i]--; if(*i==dep[u]+w[u]) ans[u]--;}
    V(i,t3[u]) tu[*i+N]--;
} 

#define _add(a,b) _nxt[++_cnt]=_head[a],_to[_cnt]=b,_head[a]=_cnt;
#define add(a,b) nxt[++cnt]=head[a],to[cnt]=b,head[a]=cnt
int main() {
    n=rd(),m=rd();
    F(i,1,n-1) {int a=rd(),b=rd(); add(a,b);add(b,a);}
    F(i,1,n) w[i]=rd(),fa[i]=i;
    F(i,1,m) {pl[i].s=rd(),pl[i].t=rd(),num[pl[i].s]++;
    // cout<<pl[i].s<<" "<<pl[i].t<<" "<<i<<endl;
    _add(pl[i].s,i);_add(pl[i].t,i);}
    tarjan(1,0);
    // for(int i=1;i<=n;i++) cout<<pl[i].lca<<" ";cout<<endl;
    F(i,1,m) {
        pl[i].dis=dep[pl[i].s]+dep[pl[i].t]-(dep[pl[i].lca]<<1);
        t2[pl[i].lca].push_back(dep[pl[i].s]);
        t3[pl[i].lca].push_back(dep[pl[i].t]-pl[i].dis);
        t1[pl[i].t].push_back(dep[pl[i].t]-pl[i].dis);
        // cout<<dep[pl[i].s]<<" "<<dep[pl[i].t]<<" "<<dep[pl[i].lca]<<" "<<pl[i].dis<<" "<<endl;
    }
    dfs(1,0);
    for(int i=1;i<=n;i++) printf("%d ",ans[i]);
    return 0;
}

[luogu1600 noip2016] 天天愛跑步 （樹上差分）