为什么最后两个样例过不了
查看原帖
为什么最后两个样例过不了
568312
xlak47楼主2022/8/27 17:17

看题目描述说卡暴力跳,那我这代码还能怎么优化?

#include<bits/stdc++.h>
using namespace std;
const int N=500010,M=1000100;
int h[N],ne[M],b[M],idex,dist[N],gp[N][18],n,m,k;
bool st[N];
void add(int x,int y){
    ne[idex]=h[x];
    b[idex]=y;
    h[x]=idex++;
}
void dfs(int x){
    memset(dist,0x3f,sizeof dist);
    dist[0]=0;
    dist[x]=1;
    queue<int> qu;
    qu.push(x);
    while(qu.size()){
        auto t=qu.front();
        qu.pop();
        if(st[t]) continue;
        st[t]=true;
        for(int i=h[t];i!=-1;i=ne[i]){
            int lu=b[i];
            if(dist[lu]>dist[t]+1){
                dist[lu]=dist[t]+1;
                qu.push(lu);
                gp[lu][0]=t;
                for(int i=1;i<=17;i++){
                    gp[lu][i]=gp[gp[lu][i-1]][i-1];
                }
            }
        }
    }
}
int lca(int a,int b){
    if(dist[a]<dist[b]) swap(a,b);
    for(int i=17;i>=0;i--){
        if(dist[gp[a][i]]>=dist[b]) a=gp[a][i];
    }
    if(a==b) return a;
    for(int i=17;i>=0;i--){
        if(gp[a][i]!=gp[b][i]){
            a=gp[a][i];
            b=gp[b][i];
        }
    }
    return gp[a][0];
}
int main(){
    cin>>n>>m>>k;
    memset(h,-1,sizeof h);
    for(int i=0;i<n-1;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(k);
    
    while(m--){
        int x,y;
        scanf("%d%d",&x,&y);
        int p=lca(x,y);
        cout<<p<<'\n';
    }
    return 0;
}
2022/8/27 17:17
加载中...