Codeforces Round #620 (Div. 2) E

LCA的倍增

模板：

 int fath[maxn][], depth[maxn];
 int dist[maxn],head[maxn];
 void add(int u,int v,int dist0){
     a[tot].next=head[u];
     a[tot].dist=dist0;
     a[tot].v=v;
     head[u]=tot++;
 }
 void dfs(int u,int fa,int d) {
     fath[u][]=fa; depth[u]=d;
     for(int i=;i<;i++) fath[u][i]=fath[fath[u][i-]][i-];
     for (int i=head[u];~i;i=a[i].next){
         int v=a[i].v;if(v==fa)continue;
         dist[v]=dist[u]+a[i].dist;
         dfs(v,u,d+);
     }
 }
 void init(int n){
     for(int i=;i<=n;i++)fath[i][]=,dist[i]=,head[i]=-,depth[i]=;
     tot=;
 }
 inline int lca(int x,int y){
     if(depth[x]<depth[y])swap(x,y);
     int h=depth[x]-depth[y];
     for(it i=;h>;i++){
         if(h&){
             x=fath[x][i];
         }
         h>>=;
     }
     if(x==y)return x;
     for(it i=;i>=;i--){
         if(fath[x][i]!=fath[y][i]){
             x=fath[x][i];
             y=fath[y][i];
         }
     }
     return fath[x][];
 }
 inline int dis(int u,int v){
     int d=lca(u,v);
     return dist[u]+dist[v]-*dist[d];
 }

题意：

给一个n点的数，以1为根的树，

询问m个x,y,a,b,k

问x，y两点暂时联通，a点到b点能不能刚好满足过k条线

能输出yes，不能输出no

思路：

树上两点之间的距离用lca算出来，因为可以反复横跳，所以只要两点之间的距离小于等于k，然后与k的奇偶性一样就是yes

一共有三条路

dis（a，b）

dis（a，x）+dis（y，b）+1

dis（a，y）+dis（x，b）+1

 #include<bits/stdc++.h>
 using namespace std;
 #define ll long long
 #define il inline
 #define it register int
 #define inf 0x3f3f3f3f
 #define lowbit(x) (x)&(-x)
 #define mem(a,b) memset(a,b,sizeof(a))
 #define mod 998244353
 const int maxn=1e5+;
 struct node{
     int next,v;
 }a[maxn<<];
 int t,m,n,head[maxn],tot,x,y,a1,b,k,f,depth[maxn],fa[maxn][];
 void add(int u,int v){
     a[tot].next=head[u];
     a[tot].v=v;head[u]=tot++;
 }
 void build(int uu){
     queue<int>q;
     depth[uu]=;
     q.push();
     while(!q.empty()){
         int u=q.front();q.pop();
         for(it i=head[u];~i;i=a[i].next){
             int v=a[i].v;
             if(depth[v]==-){
                 depth[v]=depth[u]+;q.push(v);
                 fa[v][]=u;
                 for(it j=;j<=;j++){
                     int arc=fa[v][j-];
                     fa[v][j]=fa[arc][j-];
                 }
             }
         }
     }
 }
 int lca(int x,int y){
     if(depth[x]<depth[y])swap(x,y);
     for(it i=;i>=;i--){
         if(depth[fa[x][i]]>=depth[y]){
             x=fa[x][i];
         }
     }
     if(x==y)return x;
     for(it i=;i>=;i--){
         if(fa[x][i]!=fa[y][i]){
             x=fa[x][i];
             y=fa[y][i];
         }
     }
     return fa[x][];
 }
 int dis(int u,int v){
     int d=lca(u,v);
     return depth[u]+depth[v]-*depth[d];
 }
 int main(){
     mem(head,-);mem(depth,-);tot=;
     scanf("%d",&n);
     for(it i=;i<n-;i++){
         int u,v;
         scanf("%d%d",&u,&v);add(u,v);add(v,u);
     }
     depth[]=;build();
     scanf("%d",&m);
     while(m--){
         scanf("%d%d%d%d%d",&x,&y,&a1,&b,&k);
         int kk=k;k%=;
         int d1=dis(a1,b),d2=dis(a1,x)+dis(y,b)+,d3=dis(a1,y)+dis(x,b)+;
         if((d1%==k && kk>=d1) || (d2%==k && kk>=d2)||(d3%==k && kk>=d3)){printf("YES\n");}
         else{printf("NO\n");}
     }
     return ;
 }
 /*
 5
 1 2
 2 3
 3 4
 4 5
 5
 1 3 1 2 2
 1 4 1 3 2
 1 4 1 3 3
 4 2 3 3 9
 5 2 3 3 9
 YES
 YES
 NO
 YES
 NO
 */