NYOJ-108-士兵杀敌(一)

士兵杀敌(一)
时间限制:1000 ms | 内存限制:65535 KB
难度:3
描述
南将军手下有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

很经典的树状数组水题,寒假做的,当时也没开博客,现在把题解补上。
看数据也不大不知道普通方法能不能过
树状数组略解http://blog.csdn.net/qq_32680617/article/details/50867705

代码

#include<stdio.h>
#include<string.h>
#include<string>
#include<stack>
#include<queue>
#include<math.h>
#include<limits.h>
#include<iostream>
#include<algorithm>
using namespace std;
int c[1000005];//树状数组
int N;//士兵个数
int M;//询问次数
int x,y;//询问区间
int lowbit(int x)
{
    return x&(-x);
}
void add(int i,int x)
{
    while(i<=N)
    {
        c[i]=c[i]+x;
        i=i+lowbit(i);
    }
}
int sum(int i)
{
    int sum=0;
    while(i>0)
    {
        sum=sum+c[i];
        i=i-lowbit(i);
    }
    return sum;
}
int main()
{
    while(scanf("%d%d",&N,&M)!=EOF)
    {
        memset(c,0,sizeof(c));
        int temp;//临时数据
        for(int i=1; i<=N; i++)
        {
            scanf("%d",&temp);
            add(i,temp);
        }
        for(int i=1; i<=M; i++)
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",sum(y)-sum(x-1));
        }
    }
    return 0;
}

你可能感兴趣的:(树状数组)