【bzoj4719】[Noip2016]天天爱跑步

4719: [Noip2016]天天爱跑步

Time Limit: 40 Sec   Memory Limit: 512 MB
Submit: 1432   Solved: 479
[ 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

2 0 0 1 1 1

HINT

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

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

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

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

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

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

Source

[ Submit][ Status][ Discuss]

考场上没能想到。。。。。。。。实在太菜。。。

这一题我们这样来考虑：
对于一个玩家跑的路线，我们拆成两条

一条为从s 到 lca，另一条为lca 到 t

对于s 到 lca上的点u，如果要观察到玩家，肯定会满足：

w[u] = dep[s] - dep[u]

移项得

w[u] + dep[u] = dep[s]

对于lca 到 t上的点v，如果要观察到玩家，肯定会满足：

w[v] = dep[s] + dep[t] - 2 * dep[lca] - (dep[t] - dep[v])

化简得

w[v] - dep[v] = dep[s] - 2 * dep[lca]

于是我们只需要对于每一个点找出满足上述两式的玩家数即可，可用树上差分外带一个桶实现

PS.学长使用线段树合并过的。。？？？？

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<vector>
#include<map>
using namespace std;

typedef long long LL;

const int maxn = 300010;
const int p = 300010;

struct data{
	int pos,x;
};

vector<int> e[maxn];
vector<data> a[maxn],b[maxn];
int n,m,ans[maxn],w[maxn],s[maxn],t[maxn],before[maxn];
int dep[maxn],fa[maxn][25];
int cnt[2 * maxn];

inline LL getint()
{
	LL ret = 0;
	char c = getchar();
	while (c < '0' || '9' < c) c = getchar();
	while ('0' <= c && c <= '9')
		ret = ret * 10 + c - '0',c = getchar();
	return ret;
}

inline void init()
{
	for (int j = 1; j <= 20; j++)
		for (int i = 1; i <= n; i++)
			fa[i][j] = fa[fa[i][j - 1]][j - 1];
}

inline int lca(int u,int v)
{
	if (dep[u] < dep[v]) swap(u,v);
	for (int j = 20; j >= 0; j--)
		if (dep[fa[u][j]] >= dep[v])
			u = fa[u][j];
	if (u == v) return u;
	for (int j = 20; j >= 0; j--)
		if (fa[u][j] != fa[v][j])
			u = fa[u][j] , v = fa[v][j];
	return fa[u][0];
}

inline void dfsfirst(int u)
{
	dep[u] = dep[fa[u][0]] + 1;
	for (int i = 0; i < e[u].size(); i++)
	{
		int v = e[u][i];
		if (v == fa[u][0]) continue;
		fa[v][0] = u;
		dfsfirst(v);
	}
}

inline void dfs1(int u)
{
	before[u] = cnt[w[u] + dep[u]];
	for (int i = 0; i < e[u].size(); i++)
	{
		int v = e[u][i];
		if (v == fa[u][0]) continue;
		dfs1(v);
	}
	for (int i = 0; i < a[u].size(); i++)
		cnt[a[u][i].pos] += a[u][i].x;
	ans[u] += cnt[w[u] + dep[u]] - before[u];
}

inline void dfs2(int u)
{
	before[u] = cnt[w[u] - dep[u] + p];
	for (int i = 0; i < e[u].size(); i++)
	{
		int v = e[u][i];
		if (v == fa[u][0]) continue;
		dfs2(v);
	}
	for (int i = 0; i < b[u].size(); i++)
		cnt[b[u][i].pos + p] += b[u][i].x;
	ans[u] += cnt[w[u] - dep[u] + p] - before[u];
}

int main()
{
	n = getint(); m = getint();
	for (int i = 1; i <= n - 1; i++)
	{
		int u = getint(),v = getint();
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dfsfirst(1);
	init();
	for (int i = 1; i <= n; i++) w[i] = getint();
	for (int i = 1; i <= m; i++)
	{
		s[i] = getint();
		t[i] = getint();
		int anc = lca(s[i],t[i]);
		
		if (fa[anc][0]) a[fa[anc][0]].push_back((data){dep[s[i]],-1});
		a[s[i]].push_back((data){dep[s[i]],1});
		
		b[anc].push_back((data){dep[s[i]] - 2 * dep[anc],-1});
		b[t[i]].push_back((data){dep[s[i]] - 2 * dep[anc],1});
	}
	memset(before,0,sizeof(before));
	memset(cnt,0,sizeof(cnt));
	dfs1(1);
	memset(before,0,sizeof(before));
	memset(cnt,0,sizeof(cnt));
	dfs2(1);
	printf("%d",ans[1]);
	for (int i = 2; i <= n; i++)
		printf(" %d",ans[i]);
	return 0;
}