参考代码:
#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 ", ans); return 0; }