codeforces 940E 线性DP

codeforces 940E

---

题意：

$给定含n个数的序列，将序列划分为若干段长度不等的子序列，记每段长度为k，$
$则每段子序列需要删除[\frac{k}{c}]个最小元素。$
$问所有子序列元素之和的最小值。$

---

题解：

$dp[i]表示前i个元素组成的序列的最优解，用sum[i]维护区间[1,i]的前缀和。$
$最优方案为每段子序列长度不超过c。$

• $母序列第i个元素是否与前c-1个元素构成长度为c的区间，dp[i]=min(dp[i-1]+a[i],dp[i-c]+sum[i]-sum[i-c]-min(i-c+1,i))$
• $维护c个元素的最小值，\ast mul.begin()代替上式min(i-c+1,i)，可得dp[i]=min(dp[i-1]+a[i],dp[i-c]+sum[i]-sum[i-c]-\ast mul.begin())$

---

#include <bits\stdc++.h>
using namespace std;
typedef long long ll;
const int N = 100001;
ll a[N], sum[N];
ll dp[N];

int main() {
    int n, c;
    cin >> n >> c;
    for(int i = 1 ; i <= n ; i++){
        cin >> a[i];
        dp[i] = sum[i] = sum[i-1]+a[i];
    }
    multiset <ll> mul;
    for(int i = 1 ; i < c ; i++){
        mul.insert(a[i]);
    }
    for(int i = c ; i <= n ; i++){
        mul.insert(a[i]);
        dp[i] = min(dp[i-1]+a[i], dp[i-c]+sum[i]-sum[i-c]-*mul.begin());
        mul.erase(mul.find(a[i-c+1]));
    }
    cout << dp[n] << endl;
    return 0; 
}