poj1330lca入门题

直接套模板，dfs的时候注意起点

#include<map>

#include<set>

#include<cmath>

#include<queue>

#include<stack>

#include<vector>

#include<cstdio>

#include<cassert>

#include<iomanip>

#include<cstdlib>

#include<cstring>

#include<iostream>

#include<algorithm>

#define C 0.5772156649

#define pi acos(-1.0)

#define ll long long

#define mod 1000000007

#define ls l,m,rt<<1

#define rs m+1,r,rt<<1|1

#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

const double g=10.0,eps=1e-;

const int N=+,maxn=+,inf=0x3f3f3f;

vector<int>v[N];

int depth[N],father[][N];

void dfs(int u,int f,int d)

{

    depth[u]=d;

    for(int i=;i<v[u].size();i++)

        if(v[u][i]!=f)

           dfs(v[u][i],u,d+);

}

void init(int n)

{

    int x=;

    while(father[][x]!=-)x=father[][x];

    dfs(x,-,);

    for(int i=;i<;i++)

        for(int j=;j<=n;j++)

            father[i][j]=father[i-][father[i-][j]];

}

int lca(int x,int y)

{

    if(depth[x]>depth[y])swap(x,y);

    for(int i=;i<;i++)

        if((depth[y]-depth[x])>>i&)

            y=father[i][y];

    if(x==y)return x;

    for(int i=;i>=;i--)

    {

        if(father[i][x]!=father[i][y])

        {

            x=father[i][x];

            y=father[i][y];

        }

    }

    return father[][x];

}

int main()

{

    ios::sync_with_stdio(false);

    cin.tie();

    int t,n;

    cin>>t;

    while(t--){

        cin>>n;

        for(int i=;i<=n;i++)v[i].clear();

        memset(father,-,sizeof father);

        memset(depth,,sizeof depth);

        for(int i=;i<n;i++)

        {

            int a,b;

            cin>>a>>b;

            father[][b]=a;

            v[a].push_back(b);

        }

        init(n);

        int a,b;

        cin>>a>>b;

        cout<<lca(a,b)<<endl;

    }

    return ;

}

/********************

********************/