使用线段树解决士兵杀敌问题-优快云博客

本文链接：https://blog.youkuaiyun.com/xzerui/article/details/106743743

该博客介绍了一道编程题目，涉及到利用线段树数据结构来处理士兵杀敌数目的查询和更新问题。南将军手下有N个士兵，军师需要快速回答关于指定士兵区间杀敌总数的询问，并能处理士兵杀敌数的动态增加。博客内容包括题目的描述、输入输出格式、样例以及解题思路，特别强调了线段树的建树、增加和查询操作，以及处理过程中需要注意的细节，如数组大小和变量类型的选择。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

南将军手下有N个士兵，分别编号1到N，这些士兵的杀敌数都是已知的。
小工是南将军手下的军师，南将军经常想知道第m号到第n号士兵的总杀敌数，请你帮助小工来回答南将军吧。
南将军的某次询问之后士兵i可能又杀敌q人，之后南将军再询问的时候，需要考虑到新增的杀敌数。

输入

只有一组测试数据
第一行是两个整数N,M，其中N表示士兵的个数(1<N<1000000)，M表示指令的条数。(1<M<100000)
随后的一行是N个整数，ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行是一条指令，这条指令包含了一个字符串和两个整数，首先是一个字符串，如果是字符串QUERY则表示南将军进行了查询操作，后面的两个整数m,n，表示查询的起始与终止士兵编号；如果是字符串ADD则后面跟的两个整数I,A(1<=I<=N,1<=A<=100),表示第I个士兵新增杀敌数为A.

输出

对于每次查询，输出一个整数R表示第m号士兵到第n号士兵的总杀敌数，每组输出占一行

样例输入

5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5

样例输出

解题思路：

使用算法：模板线段树
线段树函数模块：建树(build)、增加(add)，查询(quary)
注意点：
1.数组开两倍大小
2.变量开long long

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define N  2000005
struct Treenode
{
    LL val;
    int l, r;
}T[N];
int a[N];
char op[10];
void build(int beg,int end,int root)
{
    T[root].l=beg,T[root].r=end;
    if(beg==end)
        T[root].val=a[beg];
    else
    {
        int mid=(beg+end)/2;
        build(beg,mid,root*2);
        build(mid+1,end,root*2+1);
        T[root].val=T[root*2].val+T[root*2+1].val;
    }
    return;
}
int query(int i,int l,int r)
{
    if (T[i].l==l&&T[i].r==r)
        return T[i].val;
    int mid=(T[i].l+T[i].r)/2;
    if (r<=mid)
        return query(i*2,l,r);
    else if (l > mid)
        return query(i*2+1,l,r);
    else
    {
        return query(i*2,l,mid)+query(i*2+1,mid+1,r);
    }
}
void add(int i,int num,int pos)
{
    if (T[pos].l==i&&T[pos].r==i)
    {
        T[pos].val+=num;
        return;
    }
    int mid=(T[pos].l+T[pos].r)/2;
    if (i<=mid)
        add(i,num,pos*2);
    else
        add(i,num,pos*2+1);
    T[pos].val=T[pos * 2].val+T[pos * 2 + 1].val;
}
int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    build(1,n,1);
    int l,r;
    while(q--)
    {
        scanf("%s%d%d",op,&l,&r);
        if (op[0]=='Q')
            printf("%d\n",query(1, l, r));
        else
            add(l, r, 1);
    }
    return 0;
}