P1600 天天爱跑步
题面
小c同学认为跑步非常有趣,于是决定制作一款叫做《天天爱跑步》的游戏。《天天爱跑步》是一个养成类游戏,需要玩家每天按时上线,完成打卡任务。
这个游戏的地图可以看作一一棵包含 n个结点和 n-1条边的树, 每条边连接两个结点,且任意两个结点存在一条路径互相可达。树上结点编号为从1到n的连续正整数。
现在有m个玩家,第ii个玩家的起点为 S_i ,终点为 T_i。
每天打卡任务开始时,所有玩家在第00秒同时从自己的起点出发, 以每秒跑一条边的速度, 不间断地沿着最短路径向着自己的终点跑去, 跑到终点后该玩家就算完成了打卡任务。 (由于地图是一棵树, 所以每个人的路径是唯一的)
小c想知道游戏的活跃度, 所以在每个结点上都放置了一个观察员。 在结点jj的观察员会选择在第W_j秒观察玩家, 一个玩家能被这个观察员观察到当且仅当该玩家在第W_j秒也理到达了结点 j。 小C想知道每个观察员会观察到多少人?
注意: 我们认为一个玩家到达自己的终点后该玩家就会结束游戏, 他不能等待一 段时间后再被观察员观察到。 即对于把结点jj作为终点的玩家: 若他在第W_j 秒前到达终点,则在结点jj的观察员不能观察到该玩家;若他正好在第W_j 秒到达终点,则在结点j的观察员可以观察到这个玩家。
输入
第一行有两个整数n和m 。其中n代表树的结点数量, 同时也是观察员的数量, m代表玩家的数量。
接下来 n- 1行每行两个整数u和 v,表示结点u到结点v有一条边。
接下来一行 n个整数,其中第jj个整数为W_j , 表示结点jj出现观察员的时间。
接下来 m行,每行两个整数S_i,和T_i,表示一个玩家的起点和终点。
对于所有的数据,保证1≤Si,Ti≤n,0≤Wj≤n1≤Si,Ti≤n,0≤Wj≤n1\leq S_i,T_i\leq n, 0\leq W_j\leq n1≤S i ,T i ≤n,0≤W j ≤n1≤Si,Ti≤n,0≤Wj≤n1≤Si,Ti≤n,0≤Wj≤n
输出
输出1行 n个整数,第j个整数表示结点j的观察员可以观察到多少人。
样例输入
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
SOL
对于每一对点<s,t><s,t><s,t>,考虑lca(s,t)lca(s,t)lca(s,t)对答案的影响
有两种情况:观察员在 s→lca(s,t)s → lca(s, t)s→lca(s,t) 的路径上和 lca(s,t)→tlca(s, t) → tlca(s,t)→t 的路径上。
1.考虑起点为 s,观察员为 v,且 s 在 v 的子树内,被 v 观察到,
即满足 dep[s]−dep[v]=W[v]dep[s]-dep[v] = W[v]dep[s]−dep[v]=W[v],
移项可得 dep[s]=dep[v]+W[v]dep[s] = dep[v] + W[v]dep[s]=dep[v]+W[v]。
2.考虑起点为 s,终点为 t,观察员为 v,且 t 在 v 的子树内,被v 观察到,
即满足 dis(s,t)=W[v]+dep[t]−dep[v]dis(s, t) = W[v] + dep[t] - dep[v]dis(s,t)=W[v]+dep[t]−dep[v],
整理得dep[s]−2deplca(s,t)=W[v]−dep[v]dep[s]-2deplca(s,t) = W[v] - dep[v]dep[s]−2deplca(s,t)=W[v]−dep[v]。
权值线段树可以直接维护两个式子的左边。
于是问题就变成了统计每个观察员的子树中分别满足上述式子的人数。
统计子树信息且允许离线,直接从线段树合并即可,预处理lcalcalca倍增即可,树剖也可以做,求lcalcalca更快一些。
这里给出两种的做法,其实区别不大。(我不是同一时间写的,数组、变量名可能有一些区别)
膜xehothxehothxehoth大佬,在他讲了线段树合并之后才写的这道题,受益匪浅……
代码:
倍增
#include<bits/stdc++.h>
using namespace std;
#define re register
inline char nc(){
static char buf[1000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000,stdin),p1==p2)?EOF:*p1++;
}