Codeforces 983E NN country

传送门

先考虑树是一条链怎么做?

最最最最经典的做法就是倍增。

f i , x f_{i,x} fi,x表示从x出发,用2^i的条路径,最远能够走到哪里。

首先预处理 f 0 , x f_{0,x} f0,x,这个相当于把一段区间的f取max。

假设有一路径[x…y],那么要对[x…y]的f取 m a x ( f i , y ) max(f_i,y) max(fi,y),但是由于 f i f_{i} fi的初值为i,所以也可以说是对[x…n]取 m a x ( f i , y ) max(f_i,y) max(fi,y),这样就简单了,可以 O ( n ) O(n) O(n)扫一遍。

然后查询区间[x…y],那么就从x开始倍增,倍增到 &lt; y &lt;y <y的最远的地方,然后走一步就到了。

现在问题转换到了树上。

然后倍增就可以想象是在树上跳father,唯一的问题就是在lca拐弯处。

那么我们分类讨论:
1.在lca处换了路径,那么显然和链上没有什么区别。

2.在lca处没有换路径,那么把x,y都倍增到lca的下处,然后就是判断有没有一条路径覆盖剩下那一小段,这个就是判断有没有路径的两个点分在两棵子树下,转dfs序后就变成了二维偏序问题,不带修主席树或离线扫描线+树状数组即可解决。

Code:

#include<cstdio>
#include<algorithm>
#define i0 i << 1
#define i1 i << 1 | 1
#define fo(i, x, y) for(int i = x; i <= y; i ++)
#define fd(i, x, y) for(int i = x; i >= y; i --)
using namespace std;

const int N = 2e5 + 5;

int n, fa[N], x, y, m;
int fi[N], to[N], nt[N], tot;
int dep[N], siz[N], son[N], top[N];
int p[N], q[N], tp;

void link(int x, int y) {
	nt[++ tot] = fi[x], to[tot] = y, fi[x] = tot;
}

void dg(int x) {
	p[x] = ++ tp;
	siz[x] = 1;
	for(int i = fi[x]; i; i = nt[i]) {
		dep[to[i]] = dep[x] + 1;
		dg(to[i]);
		siz[x] += siz[to[i]];
		son[x] = siz[to[i]] > siz[son[x]] ? to[i] : son[x];
	}
	q[x] = tp;
}

void dfs(int x) {
	if(son[x]) top[son[x]] = top[x], dfs(son[x]);
	for(int i = fi[x]; i; i = nt[i]) if(to[i] != son[x])
		top[to[i]] = to[i], dfs(to[i]);
}

int lca(int x, int y) {
	while(top[x] != top[y])
		if(dep[top[x]] > dep[top[y]])
			 x = fa[top[x]]; else y = fa[top[y]];
	return dep[x] < dep[y] ? x : y;
}

int f[20][N];

struct nod {
	int x, y;
} b[N * 2];
int cmp(nod a, nod b) { return a.x < b.x;}

struct tree {
	int l, r, x;
} a[N * 40];
int g[N], tt, pl, pr, px;
void ad(int &i, int x, int y) {
	if(y < pl || x > pr) return;
	a[++ tt] = a[i]; i = tt;
	if(x == y) {a[i].x ++; return;}
	int m = x + y >> 1;
	ad(a[i].l, x, m); ad(a[i].r, m + 1, y);
	a[i].x = a[a[i].l].x + a[a[i].r].x;
}
void ff(int i, int x, int y) {
	if(y < pl || x > pr || !i) return;
	if(x >= pl && y <= pr) { px += a[i].x; return;}
	int m = x + y >> 1; ff(a[i].l, x, m); ff(a[i].r, m + 1, y);
}

int sum(int x1, int y1, int x2, int y2) {
	pl = x2, pr = y2; px = 0;
	ff(g[x1 - 1], 1, n);
	int y = px; px = 0;
	ff(g[y1], 1, n);
	return px - y;
}

int main() {
	scanf("%d", &n);
	fo(i, 2, n) scanf("%d", &fa[i]), link(fa[i], i);
	dep[1] = 1; dg(1); top[1] = 1; dfs(1);
	scanf("%d", &m);
	fo(i, 1, n) f[0][i] = i;
	fo(i, 1, m) {
		scanf("%d %d", &x, &y);
		int z = lca(x, y);
		if(dep[z] < dep[f[0][x]]) f[0][x] = z;
		if(dep[z] < dep[f[0][y]]) f[0][y] = z;
		b[i].x = p[x]; b[i].y = p[y];
		b[i + m].x = p[y]; b[i + m].y = p[x];
	}
	fd(i, n, 1) for(int j = fi[i]; j; j = nt[j])
		if(dep[f[0][to[j]]] < dep[f[0][i]]) f[0][i] = f[0][to[j]];
	sort(b + 1, b + 2 * m + 1, cmp); int l = 0;
	fo(i, 1, n) {
		g[i] = ++ tt;
		a[g[i]] = a[g[i - 1]];
		while(l < 2 * m && b[l + 1].x == i) l ++, pl = pr = b[l].y, ad(g[i], 1, n);
	}
	fo(j, 1, 17) fo(i, 1, n) f[j][i] = f[j - 1][f[j - 1][i]];
	scanf("%d", &m);
	fo(ii, 1, m) {
		scanf("%d %d", &x, &y);
		if(dep[x] > dep[y]) swap(x, y);
		if(x == y) { printf("0\n"); continue;} 
		int z = lca(x, y);
		int ans = 2e9;
		if(x == z) {
			int s = 0;
			fd(i, 17, 0)
				if(f[i][y] != y && dep[f[i][y]] > dep[z]) y = f[i][y], s += 1 << i;
			if(dep[f[0][y]] <= dep[z]) printf("%d\n", s + 1); else printf("-1\n");
			continue;
		}
		int u = x, v = y, s = 0;
		fd(i, 17, 0) {
			if(f[i][u] != u && dep[f[i][u]] > dep[z]) u = f[i][u], s += 1 << i;
			if(f[i][v] != v && dep[f[i][v]] > dep[z]) v = f[i][v], s += 1 << i;
		}
		if(sum(p[u], q[u], p[v], q[v])) ans = min(ans, s + 1);
		if(dep[f[0][u]] <= dep[z] && dep[f[0][v]] <= dep[z]) ans = min(ans, s + 2);
		printf("%d\n", ans == 2e9 ? -1 : ans);
	}
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值