码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj 1196

时间:2018-06-17 20:52:02      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:生成   root   clu   amp   for   second   cond   air   print   

思路:最小生成树水题

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define pii pair<int, int>

using namespace std;

const int N = 1e5 + 7;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 +7;

int n, m, k, fa[N];

struct node {
    int u, v, c1, c2;
}edge[N];

bool cmp1(const node &a, const node &b) {
    return a.c1 < b.c1;
}

bool cmp2(const node &a, const node &b) {
    return a.c2 < b.c2;
}

int getRoot(int x) {
    return fa[x] == x ? x : getRoot(fa[x]);
}

int kruskal() {
    for(int i = 1; i <= n; i++) {
        fa[i] = i;
    }

    sort(edge, edge + m, cmp2);

    int cnt = 0, ans = 0;

    for(int i = 0; i < m; i++) {
        int u = edge[i].u;
        int v = edge[i].v;
        int c = edge[i].c2;

        int x = getRoot(u);
        int y = getRoot(v);

        if(x != y) {
            fa[x] = y;
            cnt++;
            ans = max(ans, c);
            if(cnt == k) break;
        }
    }

    sort(edge, edge + m, cmp1);

    for(int i = 0; i < m; i++) {
        int u = edge[i].u;
        int v = edge[i].v;
        int c = edge[i].c1;

        int x = getRoot(u);
        int y = getRoot(v);

        if(x != y) {
            fa[x] = y;
            cnt++;
            ans = max(ans, c);
            if(cnt == n - 1) break;
        }
    }
    return ans;
}
int main() {
    scanf("%d%d%d", &n, &k, &m);
    m--;
    for(int i = 0; i < m; i++) {
        scanf("%d%d%d%d", &edge[i].u, &edge[i].v, &edge[i].c2, &edge[i].c1);
    }
    int ans = kruskal();


    printf("%d\n", ans);
    return 0;
}
/*
*/

 

bzoj 1196

标签:生成   root   clu   amp   for   second   cond   air   print   

原文地址:https://www.cnblogs.com/CJLHY/p/9193617.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!