最小生成树 poj 1679

poj 1679 The Unique MST http://poj.org/problem?id=1679

求最小生成树是否唯一

把构成最小生成树的边一条条的删

#include<iostream>
#include<stdio.h>
#include<queue>
#include<vector>
#include<string.h>
using namespace std;

struct E{
    friend bool operator< (E n1, E n2)
    {
        return n1.w > n2.w;
    }
    int u;int v;int w;
}edge[10000];

vector <int > g[102];
priority_queue < E > q;
int n,f[102],vis[102];

int find(int a)
{
    return f[a]==a?a:f[a]=find(f[a]);
}

int klu()
{
    E temp;
    int ans=0,i,ff,a,b,cnt;
    for(i=0;i<=n;i++)
        f[i]=i;
    for( i=1;i<n;)      //  找n-1 个
    {
        temp=q.top ();
        a=find(temp.u) ;
        b=find(temp.v) ;
        if(a!=b)
        {
            if(i==n-1)
                ff=temp.w;
            else
                f[a]=b;
            ans+=temp.w;
            i++;
        }
        q.pop ();
    }
    cnt=0;
    while(!q.empty ())
    {    
        temp=q.top(); q.pop ();
        if(temp.w >ff)
            break;
        a=find(temp.u );
        b=find(temp.v);
        if(a!=b)
            cnt++;
        if(cnt>0)
            return -1;
    }
    return ans;
    
}

void dfs(int s)
{
    vis[s]=1;
    for(int i=0;i<g[s].size ();i++)
        if(!vis[g[s][i]])
            dfs(g[s][i]);
}


int main()
{
    
    int t,m,i,a,b,c,ans;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            edge[i].u=a; edge[i].v=b; edge[i].w=c;
            q.push(edge[i]);
            g[a].push_back (b);
            g[b].push_back (a);
        }
        int p=0;
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++)
            if(!vis[i])
            {
                dfs(i);
                p++;
            }
            if(p>1)
                ans=0;
            else 
                ans=klu();
            if(ans==-1)
                printf("Not Unique!
");
            else printf("%d
",ans);
            while(!q.empty ())
                q.pop ();
            for(i=1;i<=n;i++)
                g[i].clear ();
    }
    return 0;
}

poj 3522 找最大边-最小边的值最小的生成树

http://poj.org/problem?id=3522

 1 #include<iostream>
 2 #include<stdio.h>
 3 #include<string.h>
 4 #include<algorithm>
 5 using namespace std;
 6 struct node{int x;int y;int w;}no[10002];
 7 
 8 int minn,f[102];
 9 int cmp(node a,node b)
10 {
11     return a.w<b.w;
12 }
13 int find(int x)
14 {
15     return f[x]==x?x:f[x]=find(f[x]);
16 }
17 int main()
18 {
19     int i,j,a,b,c,aa,bb,maxx=0,num=0,n,m,ans,final;
20     while(scanf("%d%d",&n,&m))
21     {
22         if(n==0&&m==0)
23             break;
24         for(i=0;i<m;i++)
25         {
26             scanf("%d%d%d",&a,&b,&c);
27             no[i].x=a; 
28             no[i].y=b;
29             no[i].w=c;
30         }
31         sort(no,no+m,cmp);
32         final=1000000;
33         for(i=0;i<m;i++)
34         {
35             minn=no[i].w;     //枚举最小的            
36             ans=0;
37             for(j=1;j<=n;j++)
38                 f[j]=j;
39             int k=1;
40             for(j=i;j<m;j++)
41             {
42                 aa=find(no[j].x);
43                 bb=find(no[j].y);
44                 if(aa!=bb)
45                 {
46                     k++;
47                     
48                     if(no[j].w-minn>ans)
49                         ans=no[j].w-minn;
50                     if(k==n)
51                         break;
52                     f[aa]=bb;
53                 }
54             }
55             if(j==m)
56                 break;
57             if(ans<final)
58                 final=ans;
59         }
60         if(final==1000000)
61             printf("-1
");
62         else
63             printf("%d
",final);
64     }
65     return 0;
66 }

View Code