poj 2377

題意:

最大生成樹

思路:

最大生成樹

code:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<string>
#include<queue>
#include<map>
#include<set>
#include<cmath>
using namespace std;

#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define eps 1e-4
#define maxd 10e4
#define mem(a, b) memset(a, b, sizeof(a))
typedef pair<int,int> pii;
typedef long long LL;
//------------------------------
const int maxn = 1005;
const int maxm = 20005;

struct Edge{
    int u,v,w;
    bool operator < (const Edge et) const{
        return w > et.w;
    }
}e[maxm];
int n,m;
void init(){
    int u,v,w;
    for(int i = 0; i < m; i++){
        scanf("%d%d%d",&u,&v,&w);
        e[i].u = u;
        e[i].v = v;
        e[i].w = w;
    }
}
int p[maxn];
int find(int x){
    if(x == p[x]) return p[x];
    else return p[x] = find(p[x]);
}
void solve(){
    sort(e, e+m);
    for(int i = 0; i <= n; i++) p[i] = i;
    int cnt = 1;
    LL ans = 0;
    for(int i = 0; i < m; i++){
        int pu = find(e[i].u);
        int pv = find(e[i].v);
        if(pu != pv){
            ans += e[i].w;
            p[pu] = pv;
            cnt++;
        }
        if(cnt == n){
            break;
        }
    }
    if(cnt < n) printf("-1\n");
    else cout << ans << endl;
}
int main(){
    scanf("%d%d",&n,&m);
    init();
    solve();
    return 0;
}


這道題沒有題解一定是因爲特太簡單了T_T 我居然因爲 ans += e[i].w 寫錯位置wa了一發 微笑  b l g l

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章