C - 総和
解説
- 累積和の配列を作る
- 0番目の要素は0になるようにする
- (k + i までの累積和) - (i までの累積和)
実装例(C++)
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n, k;
cin >> n >> k;
vector<long> a(n);
for (int i = 0; i < n; i++) {
cin >> a[i];
}
vector<long> sum(n+1);
for (int i = 0; i < n; i++) {
sum[i + 1] = sum[i] + a[i];
}
long total = 0;
for (int i = 0; i < n - k + 1; i++) {
total += sum[k + i] - sum[i];
}
cout << total <<endl;
}