知识点:前缀和
前缀和的题好久没做了,其实昨晚大体知道这题怎么写了,但是不知道为啥最近懒得想题,愁人,还是要多做做题(多想)。解题思路:
- 首先我们可以发现规律,就是往前移动,那么当前位置的数(要移动的数)的前 k 个数都会自增一倍
- 然后当前移动的数,他减小了 k 倍
- 所以我们最初处理下前缀和和一开始的总值即可
这里需要注意的是:他是必须移动的,就是不能把最初的值作为一个初始值
代码:
#include <iostream>
#include <cstdio>
using namespace std;
const int N = 100010;
long long a[N];
long long b[N];
int main(){
int t;
scanf("%d",&t);
while(t--){
int n, k;
long long ans = 0;
scanf("%d%d",&n,&k);
for (int i = 1; i <= n; i++){
scanf("%lld",&a[i]);
ans += a[i] * i;
b[i] = b[i - 1] + a[i] ;
}
long long res = -1e18;
for (int i = k + 1; i <= n; i ++){
int j = i - k;
res = max(res, b[i - 1] - b[j - 1] - a[i] * k );
}
printf("%lld\n",res + ans);
}
return 0;
}