[poj1797]Heavy Transportation<最大生成树prim&kruskal>

题目链接：http://poj.org/problem?id=1797

题意：给定n个点，m条边，每条边连接两点切有权值。求点1到点n的路径的上的最小边的值最大。。。

翻别人博客找到的题，方法挺多的，直接跑一个最大生成树就行，或者是一个最短路算法也行

我自己用了prim和kruskal 的方法来做

虽然用最短路也可以轻松过，但是我还是选择了生成树

prim

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<queue>
#define maxn 1005
using namespace std;
struct node{
    int u,v,w,nxt;
}e[maxn*maxn];

int n,m,low[maxn],mst[maxn],head[maxn],vis[maxn],ans;

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;
}

int tot;
void adde(int u,int v,int w){
    e[tot]=(node){u,v,w,head[u]};
    head[u]=tot++;
}

void first(){
    ans=1000000001;
    memset(low,0,sizeof(low));
    memset(mst,0,sizeof(mst));
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));tot=0    ;    
}

void prim(int num){
    int u=1,maxx=-ans,maxid;
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].v;
        mst[v]=u;low[v]=e[i].w;
    }vis[u]=1;
    int nn=n-1;
    while(nn--){
        for(int i=1;i<=n;i++){
            if(low[i]&&mst[i]){
                if(low[i]>maxx){
                    maxx=low[i];maxid=i;
                }
            }
        }
        ans=min(maxx,ans);maxx=0;
        low[maxid]=0;mst[maxid]=0;vis[maxid]=1;
        if(maxid==n)break;
        for(int i=head[maxid];i!=-1;i=e[i].nxt){
            int v=e[i].v;
            if(e[i].w>low[v]&&!vis[v]){
                mst[v]=maxid;low[v]=e[i].w;
            }
        }    
    }
    printf("Scenario #%d:
%d

",num,ans);
}

int main(){
    int T;T=read();int h=0;
    while(T--){
        first();h++;
        n=read();m=read();
        for(int i=1;i<=m;i++){
            int u,v,w;
            u=read();v=read();w=read();
            adde(u,v,w);adde(v,u,w);
        }
        prim(h);    
    }
}

kruskal

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<queue>
#define maxn 1005
using namespace std;

struct node{
    int u,v,w,nxt;
}e[maxn*maxn];

int n,m,fa[maxn],sum,ans,vis[maxn];

int tot;
void adde(int u,int v,int w){
    e[tot]=(node){u,v,w};
    tot++;
}

void first(){
    ans=1000000001;sum=0;tot=0;
    for(int i=1;i<=n;i++)fa[i]=i;
}

int find(int x){
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}

int comp(const void*a,const void*b){
    return (*(struct node*)a).w<(*(struct node*)b).w?1:-1;
}

int read(){
    int xx=0,ff=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')ff=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){xx=xx*10+ch-'0';ch=getchar();}
    return xx*ff;
}

int can(int x,int y){
    int fx=find(x),fy=find(y);
    if(fx!=fy)return 0;
    return 1;
}

int main(){
    int T,k=0;T=read();
    while(T--){
        k++;n=read();m=read();
        first();memset(vis,0,sizeof(vis));
        for(int i=1;i<=m;i++){
            int u=read(),v=read(),w=read();
            adde(u,v,w);adde(v,u,w);
        }e[0].w=ans;
        qsort(e+1,tot+1,sizeof(e[0]),comp);
        for(int i=1;i<=tot;i++){
            int u=e[i].u,v=e[i].v;
            int fu=find(u),fv=find(v);
            if(fu!=fv){
                fa[fv]=fu;ans=min(ans,e[i].w);
            }
            if(can(1,n)){
                printf("Scenario #%d:
%d

",k,ans);break;
            }
        }
    }
}

PS：注意一下输出的格式，容易错的。

还有就是判断算法的结束条件，并不是当所有的点都加入了，而是当1能到n 的时候就可以跳出了