NYOJ108 士兵杀敌（一）【树状数组】

士兵杀敌（一）

时间限制：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

题目大意：给你一个区间[1,N]，每个点上有一个值，然后是M个询问。

每次询问区间[X,Y]的和是多少。

思路：虽然递推直接求前N项和，但是还是想用树状数组的方法做。

注：query();询问前N项和

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;

int C[1000010],lowbit[1000010],num[1000010],N,M;

int query(int p)
{
    int sum = 0;
    while(p > 0)
    {
        sum += C[p];
        p -= lowbit[p];
    }
    return sum;
}

int main()
{
    scanf("%d%d",&N,&M);
    for(int i = 1; i <= N; i++)
        lowbit[i] = i&(i^(i-1));
    for(int i = 1; i <= N; i++)
    {
        scanf("%d",&num[i]);
        num[i] += num[i-1];
        C[i] = num[i] - num[i-lowbit[i]];
    }
    while(M--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",query(b)-query(a-1));
    }
    return 0;
}