首页 > 代码库 > NYOJ108 士兵杀敌(一)【树状数组】

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;
}



NYOJ108 士兵杀敌(一)【树状数组】