CF868F Yet Another Minimization Problem 题解

一、题目：

洛谷原题

codeforces原题

二、思路：

首先有一个非常简单的DP思路：设DP状态为 (dp[i, j])，表示把前 (j) 个元素分成 (i) 个部分所需要的最小花费。则有状态转移方程

[dp[i,j]=minlimits_{ileq j'leq j} {dp[i-1,j'-1]+cost(j',j)} ]

在这里，(i) 是阶段，(i) 和 (j) 共同构成状态，(j') 是决策。

接下来我们证明该DP具有决策单调性。首先让我们固定 (i)，然后设 (p_j) 是 (j) 的最优决策点（如果有多个一样优的决策点，取最左边的一个）。要证明 (p_j) 具有决策单调性，只需证 (cost) 具有四边形不等式。

即证

[forall a<b,cost(a,b+1)+cost(a+1,b)geq cost(a,b)+cost(a+1,b+1) ]

即证

[forall a<b,cost(a, b+1)-cost(a,b)geq cost(a+1,b+1)-cost(a+1,b) ]

不等式左边可以看成是将元素 (b+1) 放入区间 ([a,b]) 对 (cost) 的影响，不等式右边可以看成是将元素 (b+1) 放入区间 ([a+1,b]) 对 (cost) 的影响，显然左边大于等于右边。因此原命题得证。

于是这道题就可以使用整体二分解决了！当然，为了保证复杂度，我们要保证每次计算 (cost) 的复杂度。这其实可以通过维护一个桶 (cnt)，每次计算区间端点改变对 (cnt) 的影响即可。

时间复杂度 (O(knlog n))。

三、代码：

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;
#define FILEIN(s) freopen(s".in", "r", stdin);
#define FILEOUT(s) freopen(s".out", "w", stdout)
#define mem(s, v) memset(s, v, sizeof s)

inline int read(void) {
    int x = 0, f = 1; char ch = getchar();
    while (ch < '0' || ch > '9') { if (ch == '-') f = -1; ch = getchar(); }
    while (ch >= '0' && ch <= '9') { x = x * 10 + ch - '0'; ch = getchar(); }
    return f * x;
}

const int maxn = 1e5 + 5, maxk = 25;
const long long inf = 0x3f3f3f3f3f3f3f3f;

int n, K;
int cnt[maxn];
long long cost, dp[maxk][maxn];
int a[maxn];
int lastl, lastr;

void get(int l, int r) {
    while (lastr > r) -- cnt[a[lastr]], cost -= cnt[a[lastr]], -- lastr;
    while (lastr < r) ++ lastr, cost += cnt[a[lastr]], ++ cnt[a[lastr]];

    while (lastl < l) -- cnt[a[lastl]], cost -= cnt[a[lastl]], ++ lastl;
    while (lastl > l) -- lastl, cost += cnt[a[lastl]], ++ cnt[a[lastl]];
}

void solve(int L, int R, int l, int r, int id) {
    if (L > R) return;
    if (l > r) return;
    int mid = (l + r) >> 1;
    int pos = 0; long long minn = inf;
    for (int i = max(L, id); i <= min(R, mid); ++ i) {
        get(i, mid);
        if (dp[id - 1][i - 1] + cost < minn) {
            minn = dp[id - 1][i - 1] + cost; pos = i;
        }
    }
    dp[id][mid] = minn;
    solve(L, pos, l, mid - 1, id); solve(pos, R, mid + 1, r, id);
}

int main() {
    n = read(); K = read();
    for (int i = 1; i <= n; ++ i) {
        a[i] = read();
    }
    mem(dp, 0x3f);
    dp[0][0] = 0;
    lastl = 1; lastr = 0;
    for (int id = 1; id <= K; ++ id) {
        solve(1, n, 1, n, id);
    }
    printf("%lld
", dp[K][n]);
    return 0;
}