P6524「Wdoi-1」贪心

题意

传送门 P6524「Wdoi-1」托卡马克

题解

使选取的方案按照坐标有序 $x_0,x_1,\cdots ,x_{M-1}$，索引为 $i$ 的位置对总距离的贡献为 $i-(M-1-i)$。已知 $2|M$，则各位置对总距离的贡献的绝对值为奇数，即贡献非零且互不相同。使 $X$ 有序，按照贪心策略，贡献为正的位置，依据贡献值从小到大取 $X_{N-M/2},\cdots ,X_{M-1}$；反之，依据贡献值从小到大取 $X_0,\cdots ,X_{M/2-1}$。则得到总花费最大的方案。

求严格第二大的方案，所谓不存在并列情况，是指严格第二大不等于第一大方案的总花费，而非取到严格第二大时方案数需要唯一（这可以通过统计各个坐标值的数量求解）。需要使用 $X_{M/2},\cdots ,X_{N-M/2-1}$ 中（称之为中部）的某些元素替换总花费最大的方案。那么当 $N=M$ 以及 $X_0=X_{N-1}$ 时不存在严格第二大方案。

若中间位置的元素等于最大方案选取的元素，则只能等于贡献值为负的元素的最大值，或贡献值为正的元素的最小值。仅讨论贡献值为正的元素，其余同理。容易证明严格次小方案的一组构造为，仅从中部选取一个元素，替换最大方案中的某个元素。若选取的中部元素等于最大方案中的元素，则取严格次小值的元素替换；若选取的中部元素小于最大方案中的元素，则取最小值的元素替换。取共 $4$ 种情况的最大值即可。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(i, l, r) for (int i = l, _ = r; i < _; ++i)
#define _rep(i, r, l) for (int i = r - 1, _ = l; i >= _; --i)
const int maxn = 300005;
int N, M, K, X[maxn], chs[maxn], coef[maxn];

ll calc()
{
    sort(X, X + N);
    int nl = M >> 1, nr = M >> 1, k = 0;
    rep(i, 0, nl) coef[k] = i - (M - (i + 1)), chs[k++] = X[i];
    rep(i, N - nr, N) coef[k] = (M - (N - i)) - (N - 1 - i), chs[k++] = X[i];
    ll mx = 0;
    rep(i, 0, M) mx += (ll)coef[i] * chs[i];
    if (K == 1)
        return mx;
    if (N == M || X[0] == X[N - 1])
        return -1;
    set<int> st;
    ll mx2 = -1;
    rep(i, nl, N - nr) st.insert(X[i]);
    _rep(i, nl, 0)
    {
        auto it = st.upper_bound(chs[i]);
        if (it == st.end())
            continue;
        mx2 = max(mx2, mx + (ll)coef[i] * (*it - chs[i]));
        if (chs[i] != chs[nl - 1])
            break;
    }
    rep(i, M - nr, M)
    {
        auto it = st.lower_bound(chs[i]);
        if (it == st.begin())
            continue;
        --it;
        mx2 = max(mx2, mx + (ll)coef[i] * (*it - chs[i]));
        if (chs[i] != chs[M - nr])
            break;
    }
    return mx2;
}

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> N >> M >> K;
    rep(i, 0, N) cin >> X[i];
    cout << calc() << '\n';
    return 0;
}