[AT_past202104_d] K項足し算 题解

$\text{题目传送门}$

题意

有一个长度为 $n$ 的数列，数列中第 $i$ 个数被记为 $a_i$ （ $1\le i\le n$ ）。

给定一个 $k$ ，请按照 $i=1,2,...,n-k+1$ 的顺序计算 $a_i+a_{i+1}+...+a_{i+k-1}$ 的结果并输出。

分析

这道题最直观的做法是直接按题目模拟，用双重循环来做，主要代码见下：

llrd(n);llrd(k);
for(int i=1;i<=n;i++) llrd(a[i]);
for(int i=1;i<=n-k+1;i++)
{
	ll ans=0;
	for(int j=i;j<=i+k-1;j++) ans+=a[j];
	llwt(ans);
}

可是我们发现数据范围是 $1\le K\le N\le 5\times 10^5$，所以双重循环会超时，不能用双重循环来做。

---

正解：前缀和做法。

不知道前缀和的同学可以看一下这篇文章。

我们用 $su{m}_i$ 来表示从第 $1$ 项到第 $i$ 项的和，所以第 $a$ 项到第 $b$ 项的和就是 $su{m}_b-su{m}_{a-1}$。

枚举每个 $i$，输出 $su{m}_{i+k-1}-su{m}_{i-1}$ 即可，记得换行。

完整代码见下。

代码

#include<bits/stdc++.h>
#define rd(n) scanf("%d",&n)
#define llrd(n) scanf("%lld",&n)
#define wt(n) printf("%d",n)
#define llwt(n) printf("%lld\n",n)
#define FOR(i,a,b) for(int i=a;i<=b;i++)
#define ll long long
using namespace std;
ll n,k;
ll a[500100],sum[500100];
int main()
{
	llrd(n);llrd(k);
	for(int i=1;i<=n;i++)
	{
		llrd(a[i]);
		sum[i]=sum[i-1]+a[i];
	}
	for(int i=1;i<=n-k+1;i++) llwt(sum[i+k-1]-sum[i-1]);
   	return 0;
}