[bzoj4281][ONTAK2015]Zwi?zek Harcerstwa Bajtockiego_倍增LCA

Posted 2021-01-26 shurak

Zwi?zek Harcerstwa Bajtockiego bzoj-4281 ONTAK-2015

题目大意：给定一棵有n个点的无根树，相邻的点之间的距离为1，一开始你位于m点。之后你将依次收到k个指令，每个指令包含两个整数d和t，你需要沿着最短路在t步之内（包含t步）走到d点，如果不能走到，则停在最后到达的那个点。请在每个指令之后输出你所在的位置。

注释：$1le n,m,kle 10^6$，$mle n$。

---

想法：

对于每一个指令，用倍增LCA判一下能不能到达即可。

Code：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 1000010 
using namespace std;
int f[22][N],dep[N];
int head[N],to[N<<1],nxt[N<<1],tot;
inline char nc() {static char *p1,*p2,buf[100000]; return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;}
int rd() {int x=0; char c=nc(); while(!isdigit(c)) c=nc(); while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=nc(); return x;}
inline void add(int x,int y) {to[++tot]=y; nxt[tot]=head[x]; head[x]=tot;}
void dfs(int pos,int fa)
{
	dep[pos]=dep[fa]+1;
	f[0][pos]=fa; for(int i=1;i<=20;i++) f[i][pos]=f[i-1][f[i-1][pos]];
	for(int i=head[pos];i;i=nxt[i]) if(to[i]!=fa) dfs(to[i],pos);
}
int lca(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=20;~i;i--) if(dep[f[i][x]]>=dep[y]) x=f[i][x];
	if(x==y) return x;
	for(int i=20;~i;i--) if(f[i][x]!=f[i][y]) x=f[i][x],y=f[i][y];
	return f[0][x];
}
int clm(int x,int d)
{
	for(int i=20;~i;i--) if((1<<i)<=d) x=f[i][x],d-=(1<<i);
	return x;
}
int main()
{
	int n=rd(),pre=rd(),k=rd();
	for(int i=1;i<n;i++) {int x=rd(),y=rd(); add(x,y); add(y,x);}
	dfs(1,1);
	for(int i=1;i<=k;i++)
	{
		int x=rd(),y=rd(),z=lca(x,pre);
		if(dep[pre]-dep[z]>=y) pre=clm(pre,y);
		else if(dep[x]+dep[pre]-2*dep[z]<=y) pre=x;
		else y=dep[x]+dep[pre]-2*dep[z]-y,pre=clm(x,y);
		printf("%d ",pre);
	}
	puts("");
	return 0;
}
/*
3 1 2

1 2

2 3

3 4

1 1
*/

小结：水题啊。