不止代码：保留道路（ybtoj 最小生成树）

题目描述

解析

其实就是修建道路
我一开始只能想到枚举g去跑最小生成树
是m^2的算法（50pts）
但是其实每次加入的边只有一条
而且之前都不在最小生成树上的边以后也肯定不会在
所以可以建一个新的边的集合存当前生成树的边
这个集合的边最多只有n条
这样复杂度就降为n*m
可以通过本题

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 600;
const int M = 1e5 + 100;
int n, m;
ll wwg, wws;
struct node {
    int x, y;
    ll g, s;
    bool operator<(const node y) const { return g < y.g; }
} p[M], p2[N], now[N];
int tot;
int fa[N];
int find(int x) {
    if (fa[x] == x)
        return x;
    return fa[x] = find(fa[x]);
}
ll ans = 2e18;
bool cmp(node x, node y) { return x.s < y.s; }
void kruscal(int k) {
    for (int i = 1; i <= tot; i++) {
        p2[i] = now[i];
    }
    p2[tot + 1] = p[k];
    sort(p2 + 1, p2 + 1 + tot + 1, cmp);
    for (int i = 1; i <= n; i++) fa[i] = i;
    ll num = 0, gmax = 0, smax = 0;
    for (int i = 1; i <= tot + 1; i++) {
        int xx = find(p2[i].x), yy = find(p2[i].y);
        if (xx != yy) {
            fa[xx] = yy;
            gmax = max(gmax, p2[i].g);
            smax = max(smax, p2[i].s);
            now[++num] = p2[i];
        }
    }
    tot = num;
    if (num == n - 1) {
        ans = min(ans, gmax * wwg + smax * wws);
    }
}
int main() {
    scanf("%d%d%lld%lld", &n, &m, &wwg, &wws);
    for (int i = 1; i <= m; i++) {
        scanf("%d%d%lld%lld", &p[i].x, &p[i].y, &p[i].g, &p[i].s);
    }
    sort(p + 1, p + 1 + m);
    for (int i = 1; i <= m; i++) {
        kruscal(i);
        //		printf("i=%d tot=%d\n",i,tot);
    }
    if (ans != 2e18)
        printf("%lld", ans);
    else
        printf("-1");
    return 0;
}
/*
3 3
2 1
1 2 10 15
1 2 4 20
1 3 5 1
*/

thanks for reading!