您的位置:首页 > 其它

nyoj108——士兵杀敌(一)

2017-01-18 10:10 162 查看
描述
南将军手下有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 <iostream>
#include <cstdio>
#include <cstring>
#define maxn 1000005
using namespace std;
int n,m;
int c[maxn];
int lowbit(int x)
{
return x&(-x);
}
int Query(int x)//查询前x个数的和
{
int sum=0;
while(x>0)
{
sum+=c[x];
x-=lowbit(x);
}
return sum;
}
void add(int x,int val)
{
while(x<=n)
{
c[x]+=val;
x+=lowbit(x);
}
}
int main()
{
scanf("%d%d",&n,&m);
int val;
for(int i=1; i<=n; i++)
{
scanf("%d",&val);
add(i,val);
}
int s,e;
while(m--)
{
scanf("%d%d",&s,&e);
if(s>e){
swap(s,e);
}
int ans=Query(e)-Query(s-1);
printf("%d\n",ans);
}
return 0;
}

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