UOJ #297. 一样远
生活随笔
收集整理的這篇文章主要介紹了
UOJ #297. 一样远
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
Description本題沒有背景嚶嚶嚶。給一棵樹以及樹上的兩個點,問樹上到這兩個點距離相同的點的個數。
Input第一行一個整數N代表點的個數。接下來N-1 行,每行兩個數字F和T,表示F和T之間有一條邊。接下來一行一個整數 M 代表詢問次數。接下來 M 行,每行兩個數字 A 和 B,表示這次詢問 A 和 B(A 可能與 B 相同)。
Output輸出 M 行,每行一個整數表示到 A 和 B 一樣遠的點個數。
Sample Input4
1 2
2 3
2 4
2
1 2
1 3Sample Output0
2Hint10% 的數據,N,M=030% 的數據:N,M≤1000;100% 的數據:1≤N,M≤100000,保證樹的形態隨機
Time Limit&Memory LimitTime Limit :1sMemory Limit : 256M本題直接找兩點的距離中點即可,注意:若兩點距離為奇數則無解.
若有中點,將兩點跳到中點的子節點,分別即為u,v,當兩點深度相同時,ans=n-size[u]-size[v]
當兩點深度不同時,ans=size[fa[u]]-size[u](這里定義u為較深的那個節點,fa[u]即為兩節點中點).Code:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<stack>
#include<queue>
using namespace std;
const int M=100005;
const int N=17;
int n,m,fa[M][N],dep[M],son[M];
vector<int> G[M];
void dfs(int u){for(int i=0;i<G[u].size();i++){int v=G[u][i];if(v!=fa[u][0]){ dep[v]=dep[u]+1;fa[v][0]=u;for(int j=1;j<17;j++){fa[v][j]=fa[fa[v][j-1]][j-1];}dfs(v);son[u]+=son[v];}}
}
int up(int x,int t){for(int i=0;i<17;i++){if(t&(1<<i)){x=fa[x][i];}}return x;
}
int lca(int u,int v){if(dep[u]>dep[v]){u=up(u,dep[u]-dep[v]);}if(u==v){return u;}for(int i=16;i>=0;i--){if(fa[u][i]!=fa[v][i]){u=fa[u][i];v=fa[v][i];}}return fa[u][0];
}
int result(int u, int v){if(u==v){return n;}else{if(dep[u]<dep[v]){swap(u,v);}int x=lca(u,v);int s=dep[u]+dep[v]-2*dep[x];if(s%2==1){return 0; }if(dep[u]==dep[v]){u=up(u,s/2-1);v=up(v,s/2-1);return n-son[u]-son[v];}else{u=up(u,s/2-1);return son[fa[u][0]]-son[u];}}
}
int main(){scanf("%d",&n);int u,v;for(int i=0;i<n-1;i++){scanf("%d%d",&u,&v);G[u].push_back(v);G[v].push_back(u);}for(int i=1;i<=n;i++){son[i]=1; }for(int j=0;j<17;j++){fa[1][j]=1;}dep[1]=1;dfs(1); scanf("%d",&m);for(int i=0;i<m;i++){scanf("%d%d",&u,&v);printf("%d\n",result(u,v));}return 0;
}
轉載于:https://www.cnblogs.com/ukcxrtjr/p/11503956.html
總結
以上是生活随笔為你收集整理的UOJ #297. 一样远的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 永远不会执行的cron表达式
- 下一篇: UOJ #576. 积的第K小数