51Nod 1081 子段求和
1081 子段求和
基准时间限制:1 秒 空间限制:131072 KB 分值: 0 难度:基础题 收藏 关注
给出一个长度为N的数组,进行Q次查询,查询从第i个元素开始长度为l的子段所有元素之和。
例如,1 3 7 9 -1,查询第2个元素开始长度为3的子段和,1 {3 7 9} -1。3 + 7 + 9 = 19,输出19。
Input
第1行:一个数N,N为数组的长度(2 <= N <= 50000)。
第2 至 N + 1行:数组的N个元素。(-10^9 <= N[i] <= 10^9)
第N + 2行:1个数Q,Q为查询的数量。
第N + 3 至 N + Q + 2行:每行2个数,i,l(1 <= i <= N,i + l <= N)
Output
共Q行,对应Q次查询的计算结果。
Input示例
5
1
3
7
9
-1
4
1 2
2 2
3 2
1 5
Output示例
4
10
16
19
刚开始我是这样写的,我觉得线性的O(n)的复杂度已经不错了。谁知道还是爆。好气哦。我百度了一下,发现算是一个很简单的DP。唉,惆怅,有的题还能看得出来是,没想着这么简单的一个题,我却什么都没看出来。
#include"cstdio"
#include"algorithm"
#include"cstring"
#include"cmath"
using namespace std;
int main ()
{
int i, j, k, q;
int a, b, n;
scanf("%d",&n);
int c[n];
for(i=0;i<n;i++)
scanf("%d",&c[i]);
scanf("%d",&q);
for(i=0;i<q;i++)
{
long long int sum=0;
scanf("%d %d",&a,&b);
for(j=a-1;j<a+b-1;j++)
{
sum=sum+c[j];
}
printf("%I64d\n",sum);
}
return 0;
}
换成下面的这种写法,复杂度为O(1),很快。当然我也百度到了更厉害的方法。树状数组。还没看懂,我放链接吧。http://blog.csdn.net/int64ago/article/details/7429868
#include"cstdio"
#include"algorithm"
#include"cstring"
#include"cmath"
using namespace std;
int main ()
{
int i, j, k, q;
int a, b, n;
scanf("%d",&n);
int c[n+1];
long long int sum[n+1];
sum[0]=0;
for(i=1;i<=n;i++)
{
scanf("%d",&c[i]);
sum[i]=sum[i-1]+c[i];
}
scanf("%d",&q);
for(i=0;i<q;i++)
{
scanf("%d %d",&a,&b);
printf("%I64d\n",sum[a+b-1]-sum[a-1]);
}
return 0;
}
嗯就这样。
上一篇: 51nod 1081 子段求和
下一篇: HTML5视频自动循环播放