@codeforces - 418D@ Big Problems for Organizers


@description@

n 个点连成一棵树,经过每条边需要花费 1 个单位时间。
现给出 m 次询问,每次询问给出两个点,需要求所有点同时出发,最终所有点到达这两个点之一的最小花费时间。

input
第一行包含一个整数 n (2 ≤ n ≤ 100000) ,表示点数。
接下来 n-1 行每行两个 1~n 的整数,描述一条边。
接下来包含一个整数 m,表示询问数。
接下来 m 行每行两个整数,表示我们每次询问的两个点。

output
对于每组询问,输出最少的时间。

Examples
Input1
3
2 3
3 1
3
2 1
2 3
3 1
Output1
1
1
1
Input2
4
1 4
1 2
2 3
3
1 4
1 3
2 3
Output2
2
1
2

@solution@

假设询问给出的是 u, v 两个点,每个点肯定是往 u, v 中离自己更近的点移动。
因此,我们总可以选择 u, v 的中心边,中心边连着的包含 u 的连通块向 u 移动,中心边连着的包含 v 的连通块向 v 移动。
假如偶数长度,有多个中心边,任取一条即可。

实际上,相当于将中心边割断,然后分别以 u, v 为根求最远点。
这个显然可以 lct 搞,不过我们还是不要搞那么复杂(其实也不复杂)

考虑使用树链剖分能否维护(其实因为没有修改可以直接树上倍增)。不妨假设 u 的深度大于等于 v 的深度。

可以发现我们总可以找一条中心边使这个中心边在 u 到 lca(u, v) 的路径上。
因此,对于 u 来说,它的路径仅分为直接往下和先上后下两种。先上后下可以通过记录 -dep[x] + x不经过重儿子的最远点距离 的最大值即可。
而上面那个可以通过记录 x 向下的最长路径、次长路径即可得到,可以发现这样我们跳轻边时也容易得到答案。

而对于 v,我们需要分类讨论:
(1)假如 v 是 u 的祖先,v 的路径分为向上,向下但不经过中心边两种。这个时候我们再统计 dep[x] + x不经过重儿子的最远点距离 的最大值即可。
(2)否则,v 的路径分为直接往下、向上但不经过 lca 再向下、向上到达 lca 再向下、向上穿过 lca 继续向上、向上到达 lca 再转向 u 的方向向下但不经过中心边五种。
注意我们要把 “向上到达 lca 再向下” 这种情况单独提出来求解,因为 lca 向下有两条禁止通行的路径。此时我们还需要记录第三长的路径。

维护最大值直接写 st 表,可以做到 O((n + m)logn),因为树剖本身常数小所以跑得比 lct 和倍增都要快。

@accepted code@

#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN = 100000;
const int INF = (1<<30);
struct edge{
	edge *nxt; int to;
}edges[2*MAXN + 5], *adj[MAXN + 5], *ecnt=&edges[0];
void addedge(int u, int v) {
	edge *p = (++ecnt);
	p->to = v, p->nxt = adj[u], adj[u] = p;
	p = (++ecnt);
	p->to = u, p->nxt = adj[v], adj[v] = p;
}
int siz[MAXN + 5], dep[MAXN + 5], hvy[MAXN + 5], fa[MAXN + 5];
void dfs1(int x, int f) {
	siz[x] = 1, dep[x] = dep[f] + 1, hvy[x] = 0, fa[x] = f;
	for(edge *p=adj[x];p;p=p->nxt) {
		if( p->to == f ) continue;
		dfs1(p->to, x);
		siz[x] += siz[p->to];
		if( siz[p->to] > siz[hvy[x]] )
			hvy[x] = p->to;
	}
}
int top[MAXN + 5], dfn[MAXN + 5], tid[MAXN + 5], dcnt = 0;
void dfs2(int x, int tp) {
	top[x] = tp, dfn[++dcnt] = x, tid[x] = dcnt;
	if( hvy[x] ) dfs2(hvy[x], tp);
	for(edge *p=adj[x];p;p=p->nxt) {
		if( p->to == fa[x] || p->to == hvy[x] ) continue;
		dfs2(p->to, p->to);
	}
}
int f[MAXN + 5], g[MAXN + 5], h[MAXN + 5];
int pf[MAXN + 5], pg[MAXN + 5];
void dfs3(int x) {
	f[x] = g[x] = h[x] = 0;
	for(edge *p=adj[x];p;p=p->nxt) {
		if( p->to == fa[x] ) continue;
		dfs3(p->to);
		if( f[p->to] + 1 > f[x] ) {
			h[x] = g[x], g[x] = f[x], f[x] = f[p->to] + 1;
			pg[x] = pf[x], pf[x] = p->to;
		}
		else if( f[p->to] + 1 > g[x] ) {
			h[x] = g[x], g[x] = f[p->to] + 1;
			pg[x] = p->to;
		}
		else if( f[p->to] + 1 > h[x] )
			h[x] = f[p->to] + 1;
	}
}
int lca(int u, int v) {
	while( top[u] != top[v] ) {
		if( dep[top[u]] < dep[top[v]] ) swap(u, v);
		u = fa[top[u]];
	}
	if( dep[u] < dep[v] ) swap(u, v);
	return v;
}
int get_fa(int u, int d) {
	while( dep[top[u]] > d )
		u = fa[top[u]];
	return dfn[tid[u] - (dep[u] - d)];
}
int st[2][20][MAXN + 5], lg[MAXN + 5];
void get_st() {
	for(int i=1;i<=dcnt;i++) {
		if( pf[dfn[i]] == hvy[dfn[i]] )
			st[0][0][i] = g[dfn[i]] - dep[dfn[i]], st[1][0][i] = g[dfn[i]] + dep[dfn[i]];
		else st[0][0][i] = f[dfn[i]] - dep[dfn[i]], st[1][0][i] = f[dfn[i]] + dep[dfn[i]];
	}
	for(int j=1;j<20;j++) {
		int t = (1<<(j-1));
		for(int i=1;i+t<=dcnt;i++)
			st[0][j][i] = max(st[0][j-1][i], st[0][j-1][i+t]), st[1][j][i] = max(st[1][j-1][i], st[1][j-1][i+t]);
	}
	for(int i=2;i<=dcnt;i++)
		lg[i] = lg[i>>1] + 1;
}
int rmq(int le, int ri, bool type) {
	if( le > ri ) return -INF;
	int k = lg[ri-le+1], l = (1<<k);
	return max(st[type][k][le], st[type][k][ri-l+1]); 
}
int query(int u, int v, bool type) {
	int ret = -INF;
	while( top[u] != top[v] ) {
		ret = max(ret, rmq(tid[top[u]], tid[u]-1, type));
		u = top[u];
		if( pf[fa[u]] == u )
			ret = max(ret, g[fa[u]] + (type ? 1 : -1)*dep[fa[u]]);
		else ret = max(ret, f[fa[u]] + (type ? 1 : -1)*dep[fa[u]]);
		u = fa[u];
	}
	return max(ret, rmq(tid[v], tid[u]-1, type));
}
int main() {
	int n, m; scanf("%d", &n);
	for(int i=1;i<n;i++) {
		int u, v; scanf("%d%d", &u, &v);
		addedge(u, v);
	}
	dfs1(1, 0), dfs2(1, 1), dfs3(1), get_st();
	scanf("%d", &m);
	for(int i=1;i<=m;i++) {
		int a, b; scanf("%d%d", &a, &b);
		if( dep[a] < dep[b] ) swap(a, b);
		int l = lca(a, b), dis = dep[a] + dep[b] - 2*dep[l];
		int mid = get_fa(a, dep[a] - (dis - 1)/2), ans = max(f[a], dep[a] + query(a, mid, 0));
		if( b == l )
			ans = max(ans, max(dep[b] + query(b, 1, 0), -dep[b] + query(mid, b, 1)));
		else {
			int p = get_fa(a, dep[l] + 1), q = get_fa(b, dep[l] + 1);
			ans = max(ans, -dep[l] + query(mid, p, 1) + dep[b] - dep[l]);
			ans = max(ans, dep[b] + query(b, q, 0));
			ans = max(ans, dep[b] + query(l, 1, 0));
			ans = max(ans, f[b]);
			if( pf[l] == p ) {
				if( pg[l] == q )
					ans = max(ans, dep[b] - dep[l] + h[l]);
				else ans = max(ans, dep[b] - dep[l] + g[l]);
			}
			else if( pf[l] == q ) {
				if( pg[l] == p )
					ans = max(ans, dep[b] - dep[l] + h[l]);
				else ans = max(ans, dep[b] - dep[l] + g[l]);
			}
			else ans = max(ans, dep[b] - dep[l] + f[l]);
		}
		printf("%d\n", ans);
	}
}

@details@

总之各种分类讨论还是非常令人心烦的。。。
这个可能真的要写对拍,不然就得反复看自己有没有漏情况,或者是某种情况的公式推错了什么的。。。

posted @ 2019-08-10 20:22  Tiw_Air_OAO  阅读(215)  评论(0编辑  收藏  举报