正題

---

題目大意

求兩個點的祖孫關系。

---

解題思路

直接$LCA$，然后暴力判斷。

---

$code$

#include<cstdio> #include<algorithm> #include<queue> #include<cmath> #define N 41000 using namespace std; struct line{int to,next,w; }a[N*5]; int tot,n,m,s,x,y,ls[N],dep[N],f[N][30],dis[N][30],t; queue<int> q; inline int readn() {int X=0,w=0; char c=0;while(c<'0'||c>'9') {w|=c=='-';c=getchar();}while(c>='0'&&c<='9') X=(X<<3)+(X<<1)+(c^48),c=getchar();return w?-X:X; } inline void addl(int x,int y,int w) {a[++tot].to=y;a[tot].next=ls[x];a[tot].w=w;ls[x]=tot; } inline void bfs(int s) {q.push(s);dep[s]=1;while(!q.empty()){int x=q.front();q.pop();for (int i=ls[x];i;i=a[i].next){int y=a[i].to;if (dep[y]) continue;q.push(y);f[y][0]=x;dep[y]=dep[x]+1;dis[y][0]=a[i].w;}}t=(int)(log(n)/log(2))+1;for (int j=1;j<=t;j++){for (int i=1;i<=40000;i++){f[i][j]=f[f[i][j-1]][j-1];dis[i][j]=min(dis[i][j-1],dis[f[i][j-1]][j-1]);}} } inline int LCA(int x,int y) {if (dep[x]>dep[y]) swap(x,y);for (int i=t;i>=0;i--)if (dep[f[y][i]]>=dep[x]) y=f[y][i];if (x==y) return x;for (int i=t;i>=0;i--)if (f[y][i]!=f[x][i]) {x=f[x][i];y=f[y][i];}return f[x][0]; } int main() {n=readn();for(int i=1;i<=n;i++){x=readn();y=readn();if(y==-1) s=x;else{addl(x,y,1);addl(y,x,1);}}m=readn();bfs(s);for(int i=1;i<=m;i++){x=readn();y=readn();int lca=LCA(x,y);if(lca==x) printf("1\n");else if(lca==y) printf("2\n");else printf("0\n");} }

總結

以上是生活随笔為你收集整理的jzoj3054-祖孙询问【LCA】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。