NOIP2013 货车运输 kruscal+LCA

LCA改了一周。毒瘤题。参考了3位dalao的代码,改出来了。加油哇,拿下LCA QAQ

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cmath>
#include<queue>
#define il inline
using namespace std;
const int N=50030;
int k,v,p,q,e,r,x,y,n,m,s,t,inde,ind,maxn,cnt=1,maxlog=19;
int head[2*N],deep[N],fa[N];
bool vis[N];
int f[10009][21],f2[10009][317],head1[N];
struct Edge{
    int f,to,w,nxt;
}a[2*N];
Edge b[2*N];
il int read(){
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
il void add(int x,int y,int w){
    a[++inde].f=x;
	a[inde].to=y;
	a[inde].w=w;
	a[inde].nxt=head[x];
	head[x]=inde;
}
il int find(int x){
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
int cmp(Edge x,Edge y){return x.w>y.w;}
il void kruscal(){
   // int cei=min(n-1,m);
    sort(b+1,b+m+1,cmp);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++){
        int u=b[i].f,v=b[i].to,w=b[i].w;
        if(find(u)!=find(v)){
        	add(u,v,w);
            add(v,u,w); 
			fa[find(u)]=find(v);
        }
    }
}
il void dfs(int x,int fa,int dep){
    f[x][0]=fa;
    deep[x]=dep;
    vis[x]=1;
    for(int i=head[x];i;i=a[i].nxt){
        int y=a[i].to;
       // f2[y][0]=a[i].w;
        if(!vis[y]&&fa!=y) {
		    
			dfs(y,x,dep+1);f2[y][0]=a[i].w; 
		}
    }
} 
il void work(){
    for(int i=1;i<=n;i++){
        if(!vis[i]) dfs(i,0,1);
    }
    for(int k=1;k<=19;k++)
    for(int i=1;i<=n;i++){
        f[i][k]=f[f[i][k-1]][k-1];
        f2[i][k]=min(f2[i][k-1],f2[f[i][k-1]][k-1]);
    }
}
int LCA(int x,int y){
	int ans=999999999;
    if(deep[y]>deep[x]) swap(x,y);//x>y
    for(int i=19;i>=0;i--){
        if(deep[y]<=deep[f[x][i]]){
            ans=min(ans,f2[x][i]);
			x=f[x][i];
        }
    }
    if(x==y) return ans;
    for(int i=19;i>=0;i--){
        if(f[x][i]!=f[y][i]){
        	ans=min(ans,f2[x][i]);
        	ans=min(ans,f2[y][i]);
            x=f[x][i]; 
            y=f[y][i]; 
        }
    }
    ans=min(ans,f2[x][0]);
    ans=min(ans,f2[y][0]);
    return ans;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;i++) {	b[i].f=read(); b[i].to=read(); b[i].w=read();}  
    kruscal();//cout<<"we"<<endl;
    work();
    q=read();
  //  for(int i=1;i<=n;i++) cout<<f2[i][0];
    for(int i=1;i<=q;i++){
        e=read();r=read();
        if(find(e)!=find(r)) printf("-1\n");
        else printf("%d\n",LCA(e,r));
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值