Codeforces 940E Cashback

Cashback

题意：一共给你N个数， 然后将他们分成连续的子集， 每个集合可以删除 元素个数/c  个最小元素， 然后求每个集合删除元素后的总和。

题解：我们将集合分为1个元素或者K个元素，每K个元素查询一下最小值，然后找到最小值， 然后 dp一下就可以找到最后的结果

emmm， 当时CF的时候题意读错了，没有注意到连续，也没有注意到每K个只删除一个（我以为是删除所有的最小元素）， 然后 今天补题的时候发现了要连续， WA了一发看了数据， 才发现只删除一个，不是删除全部最小。。。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define max3(a,b,c) max(a,max(b,c))
const int INF = 0x3f3f3f3f;
const LL mod = 1e9+7;
typedef pair<int,int> pll;
const int N = 1e5+10;
LL sum[N<<2], Min[N<<2];
LL dp[N], a[N];
void PushUp(int rt){
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
    Min[rt] = min(Min[rt<<1], Min[rt<<1|1]);
}
void Build(int l, int r, int rt){
    if(l == r){
        sum[rt] = a[l];
        Min[rt] = sum[rt];
        return ;
    }
    int m = l+r >> 1;
    Build(lson);
    Build(rson);
    PushUp(rt);
}
LL Minn = 0;
LL Query(int L, int R, int l, int r, int rt){
    if(L <= l && r <= R){
        if(Minn > Min[rt]) Minn = Min[rt];
        return sum[rt];
    }
    LL ans = 0;
    int m = l+r >> 1;
    if(L <= m) ans += Query(L,R,lson);
    if(m < R)  ans += Query(L,R,rson);
    return ans;
}
int main(){
    int n, c;
    scanf("%d%d",&n,&c);
    for(int i = 1; i <= n; i++) scanf("%I64d", &a[i]);
    Build(1,n,1);
    for(int i = 1; i <= n; i++){
        if(i >= c){
            Minn = 1e10;
            LL tmp = Query(i-c+1,i,1,n,1);
            dp[i] = min(dp[i-1]+a[i], dp[i-c] + tmp - Minn);
        }
        else dp[i] = dp[i-1] + a[i];
    }
    printf("%I64d\n", dp[n]);
    return 0;
}