在线LCA 倍增法 Codeforces Round #294 (Div. 2) E - A and B and Lecture Rooms

倍增法:http://www.cnblogs.com/OUSUO/p/3805715.html

RMQ-LCA:http://dongxicheng.org/structure/lca-rmq/

题解:http://blog.csdn.net/u014800748/article/details/44007129


#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<utility>
#include<queue>
#include<set>
#include<map>
#include<math.h>
using namespace std;
#define inf 0x3f3f3f3f
#pragma warning(disable:4996)
#pragma comment(linker, "/STACK:102400000,102400000")
#define ll long long
const double eps=1e-9;

vector<int> f[100010];
int deep[100010];
int p[17][100010],lev;
int num[100010];

void dfs(int x,int fu)
{
    deep[x]=deep[fu]+1;
    num[x]=1;
    p[0][x]=fu;
    for(int i=0;i<f[x].size();i++)
    {
        int v=f[x][i];
        if(v!=fu)
        {
            dfs(v,x);
            num[x]+=num[v];
        }
    }
}
int LCA(int a,int b)
{
    int i,j;
    if(deep[a]<deep[b]) swap(a,b);
    for(j=lev;j>=0;j--)
        if(deep[a]-(1<<j)>=deep[b])
            a=p[j][a];
    if(a==b) return a;
    for(j=lev;j>=0;j--)
        if(p[j][a]!=p[j][b])
        {
            a=p[j][a];
            b=p[j][b];
        }
    return p[0][a];
}
int get(int x,int k)
{
    for(int i=lev;i>=0;i--)
        if(k>>i&1)
            x=p[i][x];
    return x;
}
int main()
{
    int n,m,i,j,k;
    cin>>n;
    for(i=0;i<=n;i++)
        f[i].clear();
    for(i=0;i<n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        f[a].push_back(b);
        f[b].push_back(a);
    }
    deep[0]=0;
    dfs(1,0);
    for(i=1;(1<<i)<n;i++)
    {
        for(j=1;j<=n;j++)
            p[i][j]=p[i-1][p[i-1][j]];
    }
    lev=i-1;
    cin>>m;
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        if(deep[a]<deep[b]) swap(a,b);
        int c=LCA(a,b);
        int tmpa=a,tmpb=b;
        if((deep[a]+deep[b]-2*deep[c])%2==1)
        {
            printf("0\n");
            continue;
        }
        int d=(deep[a]+deep[b]-2*deep[c])/2-1;
        tmpa=get(tmpa,d);
        if(deep[a]==deep[b])
        {
            tmpb=get(tmpb,d);
            printf("%d\n",n-num[tmpa]-num[tmpb]);
        }
        else
            printf("%d\n",num[p[0][tmpa]]-num[tmpa]);
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值