NSWOJ1204 士兵杀敌(一)(树状数组)

77 阅读1分钟

题目:

1204 - 士兵杀敌(一)

时间限制:1秒 内存限制:128兆\

4 次提交 1 次通过

    • 题目描述

    • 南将军手下有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 <string.h>
int N;//士兵个数
int M;//询问次数
int x,y;//询问区间
int c[1000005];
int lowbit(int x)
{
    return x&-x;
}//树状数组的关键
void add(int i,int k)
{
    while(i<=N)
    {
        c[i]+=k;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int sum=0;
    while(i>0)
    {
        sum+=c[i];
        i-=lowbit(i);
    }
    return sum;
}
int main()
{
    memset(c,0,sizeof(c));
    int q;
    scanf("%d%d",&N,&M);
    for(int i=1; i<=N; i++)
    {
        scanf("%d",&q);
        add(i,q);
    }
    for(int i=1; i<=M; i++)
    {
        scanf("%d %d",&x,&y);
        printf("%d\n",sum(y)-sum(x-1));//利用树状数组

    }
    return 0;
}

关于树状数组:

引用学长的:点这里--->树状数组略解

百度百科:--->树状数组

关键代码:

lowbit(x),这是一个自定义的函数,函数名是约定成俗的,作用就是返回x的二进制表示中最后一位1的权值 :

int lowbit(int x)//位运算,利用计算机补码特性
{
    return x&-x;
}

求数组a的前n位和 :

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

\

\

再附上一篇树状数组的博客:blog.csdn.net/acmmmm/arti…

PS:树状数组大大减少了求和所用的时间,所以算法真神奇~


\