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

 

算法分析：这题应该算是很明显的线段树，我看完题就马上向线段树方向着手，

其实这题可以有简单的算法，用个 S[]  数组来存前n项的和 如 S[n]=S[1]+S[2]+……+S[n];

S[a.b]=S[b]-S[a],这样更简单！！

不过我还是用的线段树这也是我第一个线段树，呵呵……

起初用cin 、cout 贡献了一次 TLE 后来改了 scanf 、printf 终于 AC 了

#include<iostream>
#include<stdio.h>
#include<string>
#include<algorithm>

using namespace std;

typedef struct SegTree
{  
    int left,right;
    int amount;
    struct SegTree *Lchild, *Rchild;  
}SegTreeNode;

SegTree* BuildSegTree(int L,int R)
{
    SegTree *node;
    node=(SegTree *)malloc(sizeof(SegTreeNode));
    node->left=L;
    node->right=R;
    node->amount=0;
    int M=(L+R)/2;
    if(L==R) node->Rchild=node->Lchild=NULL;
/*    
    如果不考虑L+1==R就是死循环，
    原因是如果L==1；R==2；那么M==1；于是两个子函数(1,1)(1,2)恒成立。
    这是二分法经常犯的错误。*/    
    else if(L+1==R)
    {
        node->Lchild=BuildSegTree(L,L);
        node->Rchild=BuildSegTree(R,R);
    }
    else
    {
        node->Lchild=BuildSegTree(L,M);        
        node->Rchild=BuildSegTree(M+1,R);
    }
    return node;
}

void InitSegTree(SegTree *root,int initnum,int initamount)
{
    int L=root->left,R=root->right;
    if(L<=initnum&&initnum<=R)
    {
        root->amount+=initamount;
        if(L==R)  return ;
        int M=(R+L)/2;
        if(initnum<=M)  InitSegTree(root->Lchild,initnum,initamount);
        else     InitSegTree(root->Rchild,initnum,initamount);
    }
}

int find(SegTree *root, int a,int b)
{
    int L=root->left,R=root->right;
    if(a==L&&b==R) return root->amount;
    int M=(L+R)/2;
    if(b<=M) return find(root->Lchild,a,b);
    if(M<a)  return find(root->Rchild,a,b);
    if(a<=M&&M<b) return find(root->Lchild,a,M)+find(root->Rchild,M+1,b);
}

int main()
{
    int n,m,i,e;
    scanf("%d%d",&n,&m);
    SegTree *root;
    root=BuildSegTree(1,n);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&e);
        InitSegTree(root,i,e);
    }
    int a,b;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&a,&b);
        printf("%d\n",find(root,a,b));
    }
    return 0;
}