HDU 1166 敌兵布阵(树状数组)

本文深入解析树状数组的基本原理及其实现方法,通过一个具体的实战案例,展示了如何使用树状数组进行区间修改和查询操作,适用于算法竞赛和数据结构初学者。

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

地址:http://acm.hdu.edu.cn/showproblem.php?pid=1166
树状数组基础题

include<stdio.h>
#include <iostream>
#include <stdio.h>
#include <cmath>
using namespace std;

int a[50045],node[50045];
int lowbit(int x) { return x & -x; }

void build(int a[], int node[], int size)
{
    //将数组a处理成前缀和的形式
    for (int i = 1; i <= size; i++)
        a[i] += a[i - 1];
    //根据每个节点覆盖的范围求出对应的值
    for (int i = 1; i <= size; i++)
        node[i] = a[i] - a[i - lowbit(i)];
}

void add(int pos, int val, int node[], int size)
{
    //递推求出每个父节点,逐一更新直至达到根节点
    for (int i = pos; i <= size; i += lowbit(i))
        node[i] += val;
}

int prefix_sum(int n, int node[])
{
    int sum = 0;
    //从当前节点开始,每次向前求lowbit(i)个数之和,直至到达第一个节点
    for (int i = n; i >= 1; i -= lowbit(i))
        sum += node[i];
    return sum;
}

int sum_between(int l, int r, int node[])
{
    //则区间[l,r]的和可表示为前r项的和减去前l-1项和
    return prefix_sum(r, node) - prefix_sum(l - 1, node);
}

int main()
{

    int t,n1;
    scanf("%d",&t);
    for(n1=1;n1<=t;n1++)
    {
        int n2,n,i,j,x;
        char s[10];
        printf("Case %d:\n",n1);
        scanf("%d",&n);
        for(n2=1;n2<=n;n2++)
        {
            scanf("%d",&a[n2]);
        }
        build(a,node,n);
       while(scanf("%s",s)!=EOF&&s[0]!='E')
       {
           scanf("%d%d",&i,&j);
           if(s[0]=='A')
           {
               add(i,j,node,n);
           }
           if(s[0]=='S')
           {
               add(i,j*(-1),node,n);
           }
           if(s[0]=='Q')
           {
               x=sum_between(i,j,node);
               printf("%d\n",x);
           }
       }
    }
    return 0;
}```

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值