题解 | 旅游

旅游

https://www.nowcoder.com/practice/70f173646f0a4d8daa0eac8a789e906e

#include <bits/stdc++.h>
using namespace std;
#define endl "\n"
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define int long long
#define pb push_back
using vi = vector<int>;
int find(vi& root, int x) {
    if (root[x] == x)return x;
    return root[x] = find(root, root[x]);
}

struct edge {
    int x, y, c;
};

void solve() {
    int n, m, c;cin >> n >> m >> c;
    vector<edge>a(m);
    for (auto& [x, y, c] : a) {
        cin >> x >> y >> c;
    }
    sort(all(a), [](edge a1, edge a2) {
        return a1.c < a2.c;
        });
    int l = 0, r = 1e9;
    int ans = 0;
    while (l <= r) {
        int mid = l + (r - l) / 2;
        vi root(n+1);
        iota(all(root), 0);
        int sum = 0;
        vi b;
        int t = n - 1;
        int idx = 0;
        while (t) {
            auto [xx, yy, aa] = a[idx++];
            int xxx = find(root, xx);
            int yyy = find(root, yy);
            if (xxx != yyy) {
                if (aa > mid) {
                    b.pb(aa);
                }
                root[xxx] = yyy;
                t--;
            }
        }
        sort(rall(b));
        int cnt = 1;
        for (int i : b) {
            sum += cnt * i;
            cnt++;
        }
        if (sum <= c) {
            ans = mid;
            r = mid - 1;
        }
        else {
            l = mid + 1;
        }
    }
    cout << ans << endl;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    int t = 1;
    //cin >> t;
    for (int i = 1; i <= t; i++) {
        //cout << "----Test " << i << "----" << endl;
        solve();
    }
    return 0;
}

二分最小生成树(并查集解法),小于等于mid的不计入sum,根据排序不等式,先修损坏值最大的使得sum最小

全部评论

相关推荐

故事和酒66:小米现在校招很多都是在高校搞小米训练营,然后直接挑人,大四就去实习,所以实际上校招总名额是变少了的。同学211本无经验经过两周培训直接签了
秋招,不懂就问
点赞 评论 收藏
分享
面了100年面试不知...:今年白菜这么多,冬天可以狂吃了
点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务