Parametric MST（贪心）

最新推荐文章于 2026-06-23 22:46:37 发布

原创最新推荐文章于 2026-06-23 22:46:37 发布 · 517 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #图论 #leetcode

本文探讨了一个涉及完全图和动态边权的参数化最小生成树问题，通过对节点权重进行排序并利用前缀和技巧，提出了求解最小生成树代价随参数变化规律的有效算法。

开发板推荐：天空星STM32F407VET6开发板

超高性价比 STM32主控 | 超高主频 | 一板兼容百芯 | 比赛神器 | 沉金彩色丝印

点击查看

Parametric MST

[Link](Problem - F - Codeforces)

题意

给你一个 $n$ 个结点的完全图，第 $i$ 个点的权重为 $a_i$ ，对于任意两个点的边权 $w_{i,j}(t)=a_ia_j+t(a_i+a_j)$ ，其中 $t$ 是任意一个实数， $f (t) 是最小生成树的花费$ ，问你 $f (t)$ 最小可能是多少。

思路

首先将 $a$ 从小到大排序，设 $s$ 为 $a$ 的前缀和，对于 $t$ 非常大，则权值取决于 $a_i+a_j)t$ ，则此时权值为 $a_1*(n-1)+s_n-s_1)t$ ，若 $a_1*(n-1)+s_n-s_1)>0$ 则为 $I N F$ ，同理若 $t$ 是非常小的负数，若 $a_1*(n-1)+s_n-s_1)<0$ 则为 $I N F$

其它情况我们观察一下式子 $w_{i,j}(t)=a_ia_j+t(a_i+a_j)=(a_i+t)(a_j+t)-t^2$ ，令 $b_i=a_i+t$ ，则权值为 $w_{i,j}(t)=b_ib_j-t^2$ ，令 $w_{i,j}=b_ib_j$ 则 $f(t)=mst-(n-1)t^2$ ，对于所有的对于任意一个 $t$ ,一定满足 $1,i]\ b_i<0$ ， $i+1,n]\ b_i>0$ ，对于 $b_i<0$ 我们将其和 $max(b_i)$ 相连，对于 $b_i>0$ 我们将其和 $min(b_i)$ 相连，这样求出来的就是最小值，通过枚举 $t$ 求解即可

Code

#include <bits/stdc++.h>
#define x first
#define y second
#define debug(x) cout<<#x<<":"<<x<<endl;
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 2e5 + 10, M = 2 * N, INF = 0x3f3f3f3f, mod = 1e9 + 7;
const double eps = 1e-8, pi = acos(-1), inf = 1e20;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
    e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
int n, m, k;
LL a[N], sum[N];
int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    int T;
    cin >> T;
    while (T -- ) {
        cin >> n;
        for (int i = 1; i <= n; i ++) cin >> a[i];
        sort(a + 1, a + 1 + n);
    for (int i = 1; i <= n; i ++) sum[i] = sum[i - 1] + a[i];
        if (sum[n] - sum[1] + a[1] * (n - 1) > 0 || 
            sum[n - 1] + a[n] * (n - 1) < 0) {
                cout << "INF\n";
                continue ;
            }

        LL res = -1e18;
        for (int i = 1; i <= n; i ++) {
            LL t = -a[i];
            LL mx = a[n] + t, mn = a[1] + t;
            LL ls = sum[i] + i * t, rs = sum[n] - sum[i] + (n - i) * t;

            res = max(res, ls * mx + rs * mn - mn * mx - (n - 1) * t * t);
        }

        cout << res << '\n';                    
    }
    return 0;
}