POJ 3709 K-Anonymous Sequence - 斜率优化dp

描述

给定一个数列 $a$， 分成若干段，每段至少有$k$个数， 将每段中的数减少至所有数都相同， 求最小的变化量

题解

易得到状态转移方程 $F_i = \min(F_j  + sum_i - sum_j - (i - j ) \times a_(j+1) ) $ $ 0 <= j <= i - k$。

把只含$j$ 放在一边， 其他的放在另一边得到：$F_j + j \times a_(j+1)  - sum_j =  i \times a_(j+1)  F_i - sum_i$

然后就可以愉快地套上斜率优化的板子了QuQ

还有这道题我用叉积挂了唔， 数列$a_i$是不严格递增的，点重合就挂了（也有可能是我自己打挂了

 

代码

#include<cstring>
#include<cstdio>
#include<algorithm>
#define rd read()
#define rep(i,a,b) for( int i = (a); i <= (b); ++i)
#define per(i,a,b) for( int i = (a); i >= (b); --i)
using namespace std;
typedef long long ll;

const int N = 1e6;
const ll inf = 1LL << 61;

ll sum[N], f[N], q[N], d[N];
int n, T, k;

int read() {
    int X = 0, p = 1; char c = getchar();
    for(; c > '9' || c < '0'; c = getchar()) if( c == '-') p = -1;
    for(; c >= '0' && c <= '9'; c = getchar()) X = X * 10 + c - '0';
    return X * p;
}

double calc(int a, int b) {
    double y = f[b] + b * d[b + 1] - sum[b] - f[a] - a * d[a + 1] + sum[a];
    double x = d[b + 1] - d[a + 1];
    if(!x && !y) return inf;
    return y / x;
}

int main()
{
    T = rd;
    for(; T; T--) {
        n = rd; k = rd;
        rep(i, 1, n) d[i] = rd, sum[i] = sum[i - 1] + d[i];
        rep(i, k, n) f[i] = sum[i] - i * d[1];
        rep(i, 0, k - 1) f[i] = inf;
        int l = 1, r = 1;
        q[1] = k;
        rep(i, 2 * k, n) {
            while(l < r && calc(q[l], q[l + 1]) <= i) l++;
            f[i] = min(f[i], f[q[l]] + sum[i] - sum[q[l]] - (i - q[l]) * d[q[l] + 1]);
            while(l < r && calc(q[r - 1], q[r]) >= calc(q[r], i - k + 1)) r--;
            q[++r] = i - k + 1;
        }
        printf("%lld\n", f[n]);
    }
}