我见没人写这篇文章题解,刚好我刚学会,我就想写写题解。
题目描述:
Background
大黄猫刚学习了数学中的加法运算,请你帮他用程序计算一下这个简单的加法问题……
Description
大黄猫刚刚学会了加法但还不是很熟练,所以喜欢做一些重复性的加法运算。为了考一考他,我们给出一个长度为n的正整数序列,并要求大黄猫对整个数列中所有长度为m的连续部分求和,并用程序计算出最后的答案从而判断他的答案是否正确,请你帮助设计这样一个程序进行简单的加法运算。
Input
第一行有两个正整数n, m分别代表数列的长度、要求的连续部分的长度。
第二行是整个正整数数列,包含n个正整数,用空格隔开。
Output
一行一个整数sum,表示最终求得的和。
Sample 1
Inputcopy | Outputcopy |
---|---|
5 2 3 7 1 5 9 | 38 |
Sample 2
Inputcopy | Outputcopy |
---|---|
5 6 3 7 1 5 9 | 0 |
Hint
对于n,m有: 1≤n,m≤1061≤n,m≤106,
对于数列中的任何一个数ei有: 0<=ei<=1e8。
若整个数列中没有长度为m的部分,输出0即可;显然对于其他情况整个数列中有n - m + 1个满足要求的部分,请对他们分别求和再将结果相加输出。
这道题按正常思路走会超时,所以它肯定不是按惯性思维写的,它是考察一维前缀和一维差分的
#include"stdio.h"
const int N=1e6+10;
long long a[N],b[N];
int main()
{
long long n,m;
long long sum=0;
scanf("%lld %lld",&n,&m);
if(m>n)
printf("0");
else
{
b[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
b[i]=b[i-1]+a[i];
}
for(int i=1;i<=n-m+1;i++)
sum+=b[i+m-1]-b[i-1];
printf("%lld",sum);
}
return 0;
}