【BZOJ4281】[ONTAK2015]Zwi?zek Harcerstwa Bajtockiego LCA
阿新 • • 發佈:2017-08-26
() string mil dfs data pri stream oid std
輸出一行,包含k個正整數,即執行每條指令後你所在的位置。
1 2
2 3
3 4
1 1
【BZOJ4281】[ONTAK2015]Zwi?zek Harcerstwa Bajtockiego
Description
給定一棵有n個點的無根樹,相鄰的點之間的距離為1,一開始你位於m點。之後你將依次收到k個指令,每個指令包含兩個整數d和t,你需要沿著最短路在t步之內(包含t步)走到d點,如果不能走到,則停在最後到達的那個點。請在每個指令之後輸出你所在的位置。Input
第一行包含三個正整數n,m,k(1<=m<=n<=1000000,1<=k<=1000000)。 接下來n-1行,每行包含兩個正整數x,y(1<=x,y<=n),描述一條樹邊。 接下來k行,每行兩個整數d,t(1<=d<=n,0<=t<=10^9),描述一條指令。Output
Sample Input
3 1 21 2
2 3
3 4
1 1
Sample Output
3 2題解:先倍增求出LCA,判斷能否走到,然後倍增模擬走的過程即可。
#include <cstdio> #include <cstring> #include <iostream> using namespace std; const int maxn=1000010; int n,m,S,mx,cnt; int to[maxn<<1],next[maxn<<1],head[maxn],fa[21][maxn],dep[maxn]; inline int rd() { int ret=0,f=1; char gc=getchar(); while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘)f=-f; gc=getchar();} while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+gc-‘0‘,gc=getchar(); return ret*f; } void add(int a,int b) { to[cnt]=b,next[cnt]=head[a],head[a]=cnt++; } void dfs(int x) { for(int i=head[x];i!=-1;i=next[i]) if(to[i]!=fa[0][x]) fa[0][to[i]]=x,dep[to[i]]=dep[x]+1,dfs(to[i]); } int lca(int a,int b) { if(dep[a]<dep[b]) swap(a,b); for(int i=mx;i>=0;i--) if(dep[fa[i][a]]>=dep[b]) a=fa[i][a]; if(a==b) return a; for(int i=mx;i>=0;i--) if(fa[i][a]!=fa[i][b]) a=fa[i][a],b=fa[i][b]; return fa[0][a]; } int find(int x,int y) { for(int i=mx;i>=0;i--) if(y&(1<<i)) x=fa[i][x]; return x; } int main() { n=rd(),S=rd(),m=rd(); int i,a,b,c; memset(head,-1,sizeof(head)); for(i=1;i<n;i++) a=rd(),b=rd(),add(a,b),add(b,a); dep[1]=1,dfs(1); for(mx=1;(1<<mx)<=n;mx++) for(i=1;i<=n;i++) fa[mx][i]=fa[mx-1][fa[mx-1][i]]; mx--; for(i=1;i<=m;i++) { a=rd(),b=rd(),c=lca(S,a); if(dep[S]+dep[a]-2*dep[c]<=b) S=a; else { if(dep[S]-dep[c]>=b) S=find(S,b); else S=find(a,dep[S]+dep[a]-2*dep[c]-b); } printf("%d",S); if(i<m) printf(" "); } return 0; }//3 1 2 1 2 2 3 3 4 1 1
【BZOJ4281】[ONTAK2015]Zwi?zek Harcerstwa Bajtockiego LCA