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

来源

[张云聪]原创

上传者

张云聪

这题自己用线段树做的，呵呵。果断超时

 超时代码
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
int bb;
struct node
{
    int l;
    int r;
    int num;
};
node a[3000001];
void build (int l,int r,int k)
{
    a[k].l=l;
    a[k].r=r;
    a[k].num=0;
    if(a[k].l==a[k].r)
    {
        //a[k].num=0;
        return ;
    }
    else
    {
        int mid=(l+r)/2;
        build(l,mid,2*k);
        build(mid+1,r,2*k+1);
    }
   // return ;
}
void add(int l,int r,int c,int k)
{
    if(a[k].l==l&&a[k].r==r&&l==r)
    {
        a[k].num+=c;
        return ;
    }
    else
    {
        a[k].num+=(r-l+1)*c;
        int mid=(a[k].r+a[k].l)/2;
        if(l>mid)
        {
            add(l,r,c,2*k+1);
        }
        else
        {
            add(l,r,c,2*k);
        }
    }

}
int query(int l,int r,int k)
{
    if(a[k].l==l&&a[k].r==r)
    {
        return a[k].num;
    }
    else
    {
        int mid=(a[k].r+a[k].l)/2;
        if(l>mid)
        {
            query(l,r,2*k+1);
        }
        else if(r<=mid)
        {
            query(l,r,2*k);
        }
        else
        {
           return query(l,mid,2*k)+query(mid+1,r,2*k+1);
        }

    }

}
int main()
{
    //int t;
    //cin>>t;
    //while(t--)
    //{
        int n,m,a,b;
        cin>>n>>m;
         build(1,n,1);
        for(int i=1;i<=n;i++)
      {
            cin>>bb;
            add(i,i,bb,1);
       }
        for(int i=1;i<=m;i++)
        {
            cin>>a>>b;
            cout<<query(a,b,1)<<endl;
        }
    //}
    return 0;
}
        

看了大牛代码发现，直接用一个数组记录下总和不就行了吗！！！ 

ac代码

 
 
#include <cstdio>
#include <cstring>


using namespace std;
const int maxn = 1e6;


int b[maxn + 5];
int a[maxn + 5];


int main()


{
    int n,m;
    scanf("%d%d",&n,&m);
    //printf("%d %d\n",n,m);
    for(int i = 1;i <= n;++i){
        scanf("%d",&a[i]);
    }
    for(int i = 1;i <= n;++i)
    {
        b[i] = b[i - 1] + a[i];
    }
    while(m--)
    {
        int c,d;
        scanf("%d%d",&c,&d);
        if(c > d)
        {
            int t = c;
            c = d;
            d = t;
        }
        printf("%d\n",b[d] - b[c] + a[c]);
    }
    return 0;
}//用时184ms

ac代码

 
#include <cstdio>
#include <cstring>

using namespace std;
const int maxn = 1e6;

int b[maxn + 5];
int a[maxn + 5];

int main()

{
    int n,m;
    scanf("%d%d",&n,&m);
    //printf("%d %d\n",n,m);
    for(int i = 1;i <= n;++i){
        scanf("%d",&a[i]);
    }
    for(int i = 1;i <= n;++i)
    {
        b[i] = b[i - 1] + a[i];
    }
    while(m--)
    {
        int c,d;
        scanf("%d%d",&c,&d);
        if(c > d)
        {
            int t = c;
            c = d;
            d = t;
        }
        printf("%d\n",b[d] - b[c] + a[c]);
    }
    return 0;
}//用时384。
//为何我觉得这两位大牛代码第二个更快事实却相反？





用树状数组做


 
#include <cstdio>

const int maxn = 1e6;

int a[maxn + 5];
int c[maxn + 5];
int n,m;
int lowbit(int n)
{
    return n&(-n);
}
void add(int k,int num)

{
    while(k <= n)
    {
        c[k] += num;
        k += lowbit(k);
    }
}

int query(int k)

{
    int sum = 0;
    while(k)
    {
        sum += c[k];
        k -= lowbit(k);
    }
    return sum;
}
int main()

{
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;++i){
        scanf("%d",&a[i]);
        add(i,a[i]);
    }
    while(m--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        if(x > y)
        {
            int t = x;
            x = y;
            y = t;
        }
        printf("%d\n",query(y) - query(x - 1));
    }
    return 0;
}//用时604