[bzoj4281][ONTAK2015]Związek Harcerstwa Bajtockiego_倍增LCA
Związek Harcerstwa Bajtockiego bzoj-4281 ONTAK-2015
题目大意:给定一棵有n个点的无根树,相邻的点之间的距离为1,一开始你位于m点。之后你将依次收到k个指令,每个指令包含两个整数d和t,你需要沿着最短路在t步之内(包含t步)走到d点,如果不能走到,则停在最后到达的那个点。请在每个指令之后输出你所在的位置。
注释:$1\le n,m,k\le 10^6$,$m\le 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 */
小结:水题啊。
| 欢迎来原网站坐坐! >原文链接<