【BZOJ4719】[Noip2016]天天爱跑步

最新推荐文章于 2026-01-17 14:21:18 发布

原创最新推荐文章于 2026-01-17 14:21:18 发布 · 495 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文介绍了一种算法，用于解决游戏中玩家在特定时间点被观察者观察的问题。通过深度优先搜索和标记处理，实现了高效计算每个观察者能看到的玩家数量。

http://www.lydsy.com/JudgeOnline/problem.php?id=4719

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 N<=300000

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

HINT

对于1号点,Wi=0，故只有起点为1号点的玩家才会被观察到，所以玩家1和玩家2被观察到，共有2人被观察到。

对于2号点，没有玩家在第2秒时在此结点，共0人被观察到。

对于3号点，没有玩家在第5秒时在此结点，共0人被观察到。

对于4号点，玩家1被观察到，共1人被观察到。

对于5号点，玩家1被观察到，共1人被观察到。

对于6号点，玩家3被观察到，共1人被观察到。

当时在考场上只打了25分的暴力，太弱了……现在回来补一补吧。

我们可以按照dfs的顺序来计算每个点的答案，可以发现，如果一条链i对当前点u有贡献，当且仅当①起点s[i]在u的子树中终点t[i]在u的子树外而且 deep[s[i]] - deep[u] == w[u] ② 终点t[i]在u的子树中起点s[i]在u的子树外而且 deep[t[i]]-deep[u] == len[i] - w[u] ③ 起点s[i]]和终点t[i]都在u的子树中且lca(s[i],t[i])==u

于是我们可以记录一个关于深度的数组d记录当前有哪些链的端点需要考虑

这些东西都可以通过把链拆分成标记来解决,(pos,c)表示在pos处加上c ,c=1 或 -1

所以可以先做一次dfs求出每一对s和t的lca以及每个节点的深度

然后将每一条链拆成4个标记，s[i] -> (deep[s[i]],1) t[i]->(deep[t[i]]-len[i]) lca[i]->(deep[s[i]],-1) fa[lca[i]]->(deep[t[i]],-1)

再跑一次dfs就好了

然后就是有一些细节比如w[u]==0的时候答案会被统计两次，所以要/=2

复杂度O(nα(n)+m)

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cassert>
	using namespace std;
typedef pair<int,int> pii;
struct tag{
	int pos,c;
	tag(int pos=0,int c=0):pos(pos),c(c) {};
};
const int N=600010;
int n,m,d[4*N];
int s[N],t[N],lca[N],w[N],ans[N];
int ihead[N],to[N*2],inext[2*N],cnt,f[N],fa[N],dep[N];
vector<pii> G[N];
vector<tag> T[N];
inline int getint(){
	char c=getchar(); int r=0;
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) r=r*10+c-'0',c=getchar();
	return r;
}
void addedge(int u,int v) {
	inext[++cnt]=ihead[u]; ihead[u]=cnt; to[cnt]=v;
	inext[++cnt]=ihead[v]; ihead[v]=cnt; to[cnt]=u;
}
int father(int x) {return x==f[x]?x:f[x]=father(f[x]);}
void tarjan(int u){
	f[u]=u;
	for (int i=ihead[u];i;i=inext[i]) {
		int v=to[i];
		if (fa[u]!=v) {
			dep[v]=dep[u]+1;
			fa[v]=u;
			tarjan(v);
			f[v]=u;
		}
	}
	for (int i=0;i<int(G[u].size());i++) {
		int v=G[u][i].first;
		if (f[v]) lca[G[u][i].second]=father(v);
	}
}

void dfs(int u){
	int up=d[dep[u]+w[u]],down=d[dep[u]-w[u]];
	for (int i=ihead[u];i;i=inext[i]) {
		int v=to[i];
		if (fa[u]!=v) {
			dfs(v);
		}
	}
	for (int i=0;i<int(T[u].size());i++) {
			d[T[u][i].pos]+=T[u][i].c;
	}
	ans[u]+=d[dep[u]+w[u]]-up;
	ans[u]+=d[dep[u]-w[u]]-down;
	if (!w[u]) ans[u]/=2;
}
int main(){
	n=getint(); m=getint();
	for (int i=1;i<n;i++) {
		int u=getint(),v=getint();
		addedge(u,v);
	}
	for (int i=1;i<=n;i++) w[i]=getint();
	for (int i=1;i<=m;i++) {
		s[i]=getint(); t[i]=getint();
		G[s[i]].push_back(pii(t[i],i));
		G[t[i]].push_back(pii(s[i],i));
	}
	dep[1]=n+1;
	tarjan(1);
	for (int i=1;i<=m;i++) {
		int l=lca[i],du=dep[s[i]],dv=dep[t[i]],dl=dep[l],dis=du+dv-2*dl;
		T[s[i]].push_back(tag(du,1)); T[l].push_back(tag(du,-1));
		T[t[i]].push_back(tag(dv-dis,1)); T[fa[l]].push_back(tag(dv-dis,-1));
	}
	cnt=0;
	dfs(1);
	for (int i=1;i<n;i++) printf("%d ",ans[i]);
	printf("%d",ans[n]);
	return 0;
}