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 了

View Code

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

转载于:https://www.cnblogs.com/zhourongqing/archive/2011/10/14/2212671.html