p3379——LCA模版题

题目链接

#include<bits/stdc++.h>
using namespace std;
//LCA（Least Common Ancestors），即最近公共祖先，是这样一个问题： 在有根树中，找出某两个结点 u和 v最近的公共祖先（或者说，离树根最远的公共祖先）。
//倍增解法 倍增解法的核心是分治思想。
//当已知两个点在树中的深度时，先让较深的结点向上走，直到两个结点深度一样；再二分找出离他们最近的公共祖先。
//我们记一个结点的父结点为它的 2^0=1 倍祖先，它的父结点的父结点为它的 2 倍祖先， 以此类推。
const int NN=1000005;
struct edge
{
    int next;
    int to;
    edge()
    {
        next=-1;
    };
} edge[NN];
int head[NN];
int d[NN];
int f[NN][21];
bool vis[NN];
int n,m,s;
int cnt;
void add(int u,int v)//链式前向星构图
{
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
//一、在树上预处理每个结点的深度和 1倍祖先，也就是每个结点的父结点。
//用d数组来表示每个结点的深度， f[v][h]表示结点v的h倍祖先的结点编号。
//d数组中的元素初始为 −1。
//初始化完成后，f[a][0]保存的是a的 第 2^0=1 倍祖先结点，即它的父结点。时间复杂度 O(V)。
void dfs(int u)
{
    for(int i=head[u]; i!=-1; i=edge[i].next)
    {
        int v=edge[i].to;
        if(f[v][0]==u)continue;
        if(vis[v]==0)
        {
            vis[v]=1;
            d[v]=d[u]+1;
            f[v][0]=u;
            dfs(v);
        }
    }
}
//二、倍增计算各个点的2^j倍祖先是谁，
//其中，1倍祖先就是爸爸，2倍祖先是爷爷（爸爸的爸爸），以此类推。
//该点2^j祖先也就是该点2^j-1祖先的2^j-1祖先。
void ST(int n)
{
    f[s][0]=s;
    d[s]=0;
    for(int j=1; (1<<j)<=n; j++)
    {
        for(int i=1; i<=n; i++)
            f[i][j]=f[f[i][j-1]][j-1];
    }
}
//三（查询）：首先，深度大的点往上爬，直到与深度小的点在同一层。
//若此时两结点相同直接返回此结点，即最近公共祖先LCA。
//  这意味着其中一个结点是另一个结点的祖先结点。
//否则，利用倍增思想，同时让 x和 y二分地向上找，直到找到深度相等且最小的 x和 y的祖先 x',y'，满足 x'不等于y'。此时他们的父结点 f[x'][0] 即为 x 和 y 的最近公共祖先 LCA。
//代码的时间复杂度为O(logV).
int LCA(int u,int v)
{
    //cout<<u<<" "<<v<<endl;
    if(d[u]<d[v])swap(v,u);// 确保u的深度比v大
    //cout<<"d[u]:"<<d[u]<<"d[v]:"<<d[v]<<endl;
    int h=d[u]-d[v];
    //cout<<"h:"<<h<<endl;
    int i,j;
    for(i=0; (1<<i)<=d[u]; i++);
    i--;
    for(j=i; j>=0; j--)
    {
        if(d[u]-(1<<j)>=d[v])
        {
            u=f[u][j];
        }
    }
    //cout<<"u:"<<u<<endl;
    if(u==v)return u;
    for(int j=i; j>=0; j--)//u和v到达同一层之后的从最大的爸爸开始找
    {
        if(f[u][j]!=f[v][j])
        {
            u=f[u][j];
            v=f[v][j];
        }
    }
    //cout<<"u:"<<u<<"f[u][0]:"<<f[u][0]<<endl;
    return f[u][0];
}
int main()
{
//    cin>>n>>m>>s;
    scanf("%d%d%d",&n,&m,&s);
    int x,y;
    cnt=1;
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
    memset(d,0,sizeof(d));
    for(int i=1; i<n; i++)
    {
//        cin>>x>>y;
        scanf("%d%d",&x,&y);
        add(y,x);
        add(x,y);
    }
    dfs(s);//对根节点开始dfs将所有的孩子们和爸爸们的关系确认->f数组里面
    ST(n);
    for(int i=0; i<m; i++)
    {
//        cin>>x>>y;
        scanf("%d%d",&x,&y);
//        cout<<LCA(x,y)<<endl;
        printf("%d\n",LCA(x,y));
    }
    return 0;
}

气死我了气死我了debug好久最后是cin和cout导致的TLE
我是菜菜 烂地里了