UVA 1395 苗条的生成树(最小生成树+并查集)

苗条的生成树

紫书P358 这题最后坑了我20分钟，怎么想都对了啊，为什么就wa了呢，最后才发现，是并查集的编号搞错了。
题目编号从1开始，我并查集编号从0开始 = =
图论这种题真的要记住啊！！题目大部分都是从1开始编号，而代码大部分是从0开始编号，所以要把输入减减。

【题目链接】苗条的生成树

【题目类型】最小生成树+并查集

&题解：

如果你看懂了Kruskal算法，那么这个也就很好懂了。
首先按边排序，之后找个连续的区间[L,R] 判断是否已经是生成树，如果是，更新答案就好。
判断是否为生成树这里有点意思：cnt=0 当合并并查集的时候，cnt++ 当cnt==n-1时，代表正好进入了n-1条边，所以就一定是生成树了。

【时间复杂度】O(\(m^2\))

&代码：

#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxm = 5000 + 5 ;
int n,m,u[maxm],v[maxm],w[maxm];
int r[maxm],p[maxm];
int cmp(int i,int j) {return w[i]<w[j];}
int find(int x) {return p[x]==x?x:p[x]=find(p[x]);}
void Solve()
{
    while(cin>>n>>m){
        if (n==0&&m==0) break;
        for(int i=0;i<m;i++){
            cin>>u[i]>>v[i]>>w[i];
            u[i]--,v[i]--,w[i]--;
        }
        for(int i=0;i<m;i++)
            r[i]=i;
        sort(r,r+m,cmp);
        int ans=INF;
        for(int L=0;L<m;L++){
            for(int i=0;i<n;i++)
                p[i]=i;
            int cnt=0,R;
            for(R=0;L+R<m;R++){
                int e=R+L;
                int x=find(u[r[e]]),y=find(v[r[e]]);
                if (x!=y){
                    cnt++;
                    p[x]=y;
                }
                if (cnt==n-1)
                    break;
            }
            if(cnt==n-1)
                ans=min(ans,w[r[R+L]]-w[r[L]]);
        }
        if (ans!=INF) printf("%d\n",ans);
        else puts("-1");
    }
}
int main()
{
    Solve();
    return 0;
}

转载于:https://www.cnblogs.com/s1124yy/p/6056501.html