您的位置:首页 > 其它

nyoj 士兵杀敌(一)(树状数组)

2016-11-09 21:37 295 查看
士兵杀敌(一)

描述

南将军手下有N个士兵,分别编号1到N,这些士兵的杀敌数都是已知的。

小工是南将军手下的军师,南将军现在想知道第m号到第n号士兵的总杀敌数,请你帮助小工来回答南将军吧。

注意,南将军可能会问很多次问题。

输入
只有一组测试数据

第一行是两个整数N,M,其中N表示士兵的个数(1<N<1000000),M表示南将军询问的次数(1<M<100000)

随后的一行是N个整数,ai表示第i号士兵杀敌数目。(0<=ai<=100)

随后的M行每行有两个整数m,n,表示南将军想知道第m号到第n号士兵的总杀敌数(1<=m,n<=N)。

输出
对于每一个询问,输出总杀敌数

每个输出占一行
样例输入
5 2
1 2 3 4 5
1 3
2 4


样例输出
6
9


代码示例:
#include<stdio.h>
#include<string.h>
int a[1000005];
int n,m;
int lowbit(int x)
{
return x&(-x);
}
void add(int i,int t)
{
while(i<=n)
{
a[i]=a[i]+t;
i=i+lowbit(i);
}
}
int sum(int i)
{
int sum=0;
while(i>0)
{
sum+=a[i];
i=i-lowbit(i);//求前i项的和
}
return sum;
}
int main()
{
scanf("%d%d",&n,&m);
int i,t,j,x,y;
for(i=1; i<=n; i++)
{
scanf("%d",&t);
add(i,t);
}
for(j=1; j<=m; j++)
{
scanf("%d%d",&x,&y);
printf("%d\n",sum(y)-sum(x-1));
}
return 0;
}

ps:树状数组就是快。。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: