CodeForces.1174D.ArraySplitting.(后缀和数组 + 贪心)

  题目链接

参考代码:

#include <cstdio>
#include <algorithm>
using namespace std;

typedef long long ll;
const int maxn = 300000 + 5;
int n, k;
ll ans = 0, value[maxn];

bool cmp(const ll  &a, const ll  &b) {
    return a > b;
}

int main() {
    scanf("%d %d", &n, &k);
    for(int i = 1; i <= n; i ++) scanf("%I64d", &value[i]);
    for(int i = n; i; i --) value[i] += value[i + 1];//构造value后n项和的后缀和数组
    sort(value + 2, value + n + 1, cmp);//对后n - 1项进行排序,因为第后n项和是必须要加的
    for(int i = 1; i <= k; i ++) ans += value[i];//选取其中最大的n - 1项和第一项即可,就可以实现将所选的数字加k次
    printf("%I64d\n", ans);
    return 0;
}

 

posted @ 2019-06-06 20:37  Cruel_King  阅读(467)  评论(0编辑  收藏  举报