牛客春招刷题训练营-2025.5.27题解

b 活动地址： 牛客春招刷题训练营 - 编程打卡活动

简单题 小红的整数配对

贪心，先从大到小排序，每次取一对最大的数，如果这一对的差大于 ，则放弃较大的数，取下一个数。
注意答案需要 long long。
时间复杂度 。

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n, k;
    cin >> n >> k;
    vector<int> a(n);
    for (int i = 0; i < n; i++)cin >> a[i];
    sort(a.begin(), a.end(), greater<>());
    long long ans = 0;
    for (int i = 0; i + 1 < n;) {
        if (a[i] - a[i + 1] <= k) {
            ans += 1ll * a[i] * a[i + 1];
            i += 2;
        } else i++;
    }
    cout << ans << '\n';
    return 0;
}

中等题 小红的取模构造

分类讨论。
如果 且 ，则 。
如果 ，则无解。
如果 且 ，则 。
如果 且 ，则 。
如果 且 且 ，则 。
如果 且 且 ，则 。
时间复杂度 。

#include <bits/stdc++.h>
using namespace std;
void solve() {
    int a, b;
    cin >> a >> b;
    if (a == b) {
        if (a != 0)
            cout << "-1 -1\n";
        else cout << "1 1\n";
    } else if (a == 0) {
        cout << 2 * b << ' ' << b << '\n';
    } else if (b == 0) {
        cout << a << ' ' << 2 * a << '\n';
    } else {
        if (a == min(a, b))a += b;
        else b += a;
        cout << a << ' ' << b << '\n';
    }
}
int main() {
    int t;
    cin >> t;
    while (t--)solve();
    return 0;
}

困难题 【模板】单源最短路Ⅲ ‖ 非负权图

dijkstra 算法模板。
dijkstra 主要用于解决单源最短路，即一个源点到其余所有点的最短路，但是边权不能为负。
这里采用堆优化，时间复杂度 。

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n, m, s;
    cin >> n >> m >> s;
    vector<vector<pair<int, long long>>> ver(n + 1);
    for (int i = 0; i < m; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        ver[u].push_back({v, w});
    }
    vector<long long> dis(n + 1, 1e18);
    auto dijkstra = [&](int s = 1) -> void {
        using PLI = pair<long long, int>;
        priority_queue<PLI, vector<PLI>, greater<PLI>> q;
        q.emplace(0, s);
        dis[s] = 0;
        vector<int> vis(n + 1);
        while (!q.empty()) {
            int x = q.top().second;
            q.pop();
            if (vis[x]) continue;
            vis[x] = 1;
            for (auto [y, w] : ver[x]) {
                if (dis[y] > dis[x] + w) {
                    dis[y] = dis[x] + w;
                    q.emplace(dis[y], y);
                }
            }
        }
    };
    dijkstra(s);
    for (int i = 1; i <= n; i++) {
        if (dis[i] >= 1e18)dis[i] = -1;
        cout << dis[i] << " \n"[i == n];
    }
    return 0;
}