士兵杀敌( 一) 南阳理工 108

title: 士兵杀敌( 一) 南阳理工 108
tags: [树状数组]

题目链接

描述

南将军手下有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<stack>
#include<string.h>
#include<stdlib.h>
#include<queue>
using namespace std;
int N,M;
int a[1000009]= {0};///每个士兵的杀敌数目
int e[1000009]= {0};///树状数组
int zhuan(int n)
{
    return (n&(-n));
}

void UpDate(int biao,int num)///加入当前的士兵的话,就要把含有当前士兵的区间全部更新
{
    while(biao<=N)
    {
        e[biao]+=num;
        biao+=zhuan(biao);
    }
}

int SUM(int n)///求1~n这个区间的和
{
    int sum=0;
    while(n>0)
    {
        sum+=e[n];
        n-=zhuan(n);
    }
    return sum;
}
int main()
{
    scanf("%d%d",&N,&M);
    for(int i=1; i<=N; i++)
    {
        scanf("%d",&a[i]);
        UpDate(i,a[i]);///每次输入一个数后,都要把当前的树状数组更新
    }
    int n1,n2;
    for(int i=1; i<=M; i++)
    {

        scanf("%d%d",&n1,&n2);
        printf("%d
",SUM(n2)-SUM(n1-1));
    }

    return 0;
}
原文地址:https://www.cnblogs.com/cmmdc/p/6728699.html