HDU 1166 敌兵布阵

最新推荐文章于 2025-08-17 17:21:07 发布

原创最新推荐文章于 2025-08-17 17:21:07 发布 · 389 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #线段树

线段树专栏收录该内容

12 篇文章

订阅专栏

本文详细介绍了一道线段树入门题目，包括线段树的建立、更新和查询操作，并提供完整的C++代码实现。适合初学者理解线段树的基本原理及应用。

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

简单的线段树入门题。

AC_code:

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>

using namespace std;

struct node{
    int l,r,sum;
    int mid(){
        return (l + r) >> 1;
    }
}tree[50000 << 2];

int ans;
void buildtree(int l, int r, int rt){
    tree[rt].l = l;
    tree[rt].r = r;
    if(l == r){
        scanf("%d",&tree[rt].sum);
        return;///一段递归结束
    }
    int mid = tree[rt].mid();
    buildtree(l, mid, rt << 1);
    buildtree(mid + 1, r, rt << 1 | 1);
    tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum;
}

void update(int rt, int i, int j){
    if(tree[rt].l == i && tree[rt].r == i){
        tree[rt].sum += j;
        return;
    }
    int mid = tree[rt].mid();
    if(i <= mid)    update(rt << 1, i, j);
    else    update(rt << 1 | 1, i, j);
    tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum;
}

void query(int l, int r, int rt){
    if(tree[rt].l >= l && tree[rt].r <= r){
        ans += tree[rt].sum;
        return;
    }
    int mid = tree[rt].mid();
    if(r <= mid)    query(l, r, rt << 1);
    else if(l > mid)    query(l, r, rt << 1 | 1);
    else{
        query(l, mid, rt << 1), query(mid + 1, r, rt << 1 | 1);
    }
}

int main(){
//    freopen("in.txt", "r", stdin);
    int cas, t = 1;
    scanf("%d",&cas);
    while(cas--){
        int N;
        scanf("%d",&N);
        buildtree(1, N, 1);
        printf("Case %d:\n", t++);
        char str[11];
        while(scanf("%s",str)){
            if(str[0]=='E')
                break;
            int a,b;
            scanf("%d%d",&a,&b);
            if(str[0]=='Q')
            {
                ans=0;
                query(a,b,1);
                printf("%d\n",ans);
            }
            else if(str[0]=='A')
                update(1,a,b);
            else
                update(1,a,-b);
        }
    }
    return 0;
}