欢迎访问 生活随笔!

生活随笔

当前位置: 首页 > 人文社科 > 生活经验 >内容正文

生活经验

A and B and Lecture Rooms CodeForces - 519E LCA+dfs序

发布时间:2023/11/27 生活经验 55 豆豆
生活随笔 收集整理的这篇文章主要介绍了 A and B and Lecture Rooms CodeForces - 519E LCA+dfs序 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

看到这个题的第一个思路就是就是统计以每一个点为根的所有节点个数,然后具体就分情况讨论一下即可。

因为刚刚学习了dfs序,这个题就用了dfs序来通过进出时间戳来相减表示其为根的子节点个数。

分情况

我们假设每个节点为根所有节点的数目为num[u];  

1.两点同一间教室 输出n

2.两点深度一样,这个时候找到x,y的最近祖先LCA 点u,将x,y分别往上跑d[x]-d[u]-1次,及对应x.y到u的路上,u点下边那个点,

那么答案就是num[root]-num[x]-num[y];

3.深度不一样,看两点的距离是不是偶数,不是输入-1,是的话设距离为s,那么我们要吧x,y中深度深的那个往上跑s/2的点,就找到x,y最近那个符合题意得点,然后再通过上述方式将num[u]减去x,y中深度深的那个点向上跑s/2-1的num值就得到答案了。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define me(a,b) memset(a,b,sizeof(a))
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
#define pii make_pair
const int dir[4][2]= {0,-1,-1,0,0,1,1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=1e5+5;
int f[maxn][33],d[maxn],head[maxn],L[maxn],R[maxn];
int sign,n,m,t;
void init()
{sign=t=0;for(int i=0;i<=n;i++)head[i]=-1;
}
struct node
{int to,p;
}edge[maxn<<1];
void add(int u,int v)
{edge[sign]=node{v,head[u]};head[u]=sign++;
}
void dfs(int u)
{L[u]=++t;for(int i=1;(1<<i)<=n;i++)f[u][i]=f[f[u][i-1]][i-1];for(int i=head[u];~i;i=edge[i].p){int v=edge[i].to;if(v==f[u][0]) continue;d[v]=d[u]+1;f[v][0]=u;dfs(v);}R[u]=t;
}
int lca(int a,int b)
{if(d[a]<d[b]) swap(a,b);int x=d[a]-d[b];for(int i=0;(1<<i)<=x;i++)if((1<<i)&x) a=f[a][i];if(a!=b){for(int i=(int)log2(n);i>=0;i--)if(f[a][i]!=f[b][i]) a=f[a][i],b=f[b][i];a=f[a][0];}return a;
}
int main()
{int x,y;while(cin>>n){init();for(int i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);f[1][0]=1,d[1]=1;dfs(1);cin>>m;while(m--){scanf("%d %d",&x,&y);if(x==y){cout<<n<<endl;continue;}int u=lca(x,y);if(d[x]==d[y]){int c=d[x]-d[u]-1;for(int i=0;(1<<i)<=c;i++)if((1<<i)&c) x=f[x][i],y=f[y][i];cout<<R[1]-L[1]+1-(R[x]-L[x]+1)-(R[y]-L[y]+1)<<endl;}else{if(d[x]<d[y]) swap(x,y);int s=d[x]+d[y]-2*d[u];if(s%2!=0) cout<<0<<endl;else{s=s/2-1;for(int i=0;(1<<i)<=s;i++)if((1<<i)&s) x=f[x][i];u=f[x][0];cout<<R[u]-L[u]-(R[x]-L[x])<<endl;}}}}return 0;
}

 

总结

以上是生活随笔为你收集整理的A and B and Lecture Rooms CodeForces - 519E LCA+dfs序的全部内容,希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错,欢迎将生活随笔推荐给好友。