Codeforces Round #466 (Div. 2) E. Cashback

Codeforces Round #466 (Div. 2) E. Cashback(dp + 贪心)

题意：

给一个长度为\(n\)的序列\(a_i\),给出一个整数\(c\)

定义序列中一段长度为k的区间的贡献为区间和减去前\(\lfloor \frac{k}{c} \rfloor\)小数的和

现在要给序列\(a_i\)做一个划分，使得贡献的总和最小。

思路:

容易想到最裸的dp的思路

\(dp[i] = min(dp[j] + cost(j,i)) , j < i\), \(cost(j,i)\)就是区间[j,i]的贡献

这样暴力枚举复杂度是\(O(n ^ {2} log n)\)的

观察发现 划分成一段\(k * c\)的区间不会优于划分成\(k\)段长度为\(c\)的区间

同理分成非\(c\)的整数倍长度的区间是不会优于拆分成\(c\)的整数倍和一段长度小于\(c\)的区间

由于长度小于\(c\)的区间是没有贡献的，所以和拆成长度为1的区间是等价的。

所以最优的划分方式就是分成长度为1或者分成长度为c，再来做dp就可以了。

#include<bits/stdc++.h>
#define LL long long
using namespace std;

const int N = 1e5 + 10;
const int inf = 0x3f3f3f3f;
int a[N];
int n, c;
int mi[N << 2];
LL dp[N],sum[N];
void build(int l,int r,int rt){
    if(l == r){
        scanf("%d",a + l);
        mi[rt] = a[l];
        return ;
    }
    int m = l + r >> 1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    mi[rt] = min(mi[rt<<1],mi[rt<<1|1]);
}
int querymi(int L,int R,int l,int r,int rt){
    if(L <= l && R >= r) return mi[rt];
    int ans = inf;
    int m = l + r >> 1;
    if(L <= m) ans = min(ans, querymi(L,R,l,m,rt<<1));
    if(R > m) ans = min(ans,querymi(L,R,m+1,r,rt<<1|1));
    return ans;
}
int main(){

    scanf("%d%d",&n,&c);
    build(1,n,1);
    for(int i = 1;i <= n;i++) sum[i] = sum[i - 1] + a[i];
    for(int i = 1;i <= n;i++){
        dp[i] = dp[i - 1] + a[i];
        if(i >= c){
            dp[i] = min(dp[i], dp[i - c] + sum[i] - sum[i - c] - querymi(i - c + 1,i,1,n,1));
        }
    }
    cout<<dp[n]<<endl;
    return 0;
}