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

题目链接

#include <iostream>
#include <bits/stdc++.h>
#define lowbit(x) (x&(-x))
using namespace std;
const long long int maxx=50005;
int a[maxx];
int c[maxx];
int n;
void update(int i,int k)
{
    while(i<=n)
    {
        c[i]+=k;
        i+=lowbit(i);
    }

}
int  getsum(int i)
{
    int ans=0;
    while(i>0)
    {

        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{  ios::sync_with_stdio(false);
    int t;
    cin>>t;
    char s[20];
    for(int cnt=1; cnt<=t; cnt++)
    {    memset(c,0,sizeof(c));

        cout<<"Case "<<cnt<<":"<<endl;
        cin>>n;
        for(int i=1; i<=n; i++)
        {
            cin>>a[i];
            update(i,a[i]);
        }

        while(cin>>s)
        {

            if(s[0]=='Q')
            {
                int x,y;
                cin>>x>>y;
                int sum=getsum(y)-getsum(x-1);
                cout<<sum<<endl;
            }
            else if(s[0]=='A')
            {
                int x,y;
                cin>>x>>y;
                update(x,y);
            }
            else if(s[0]=='S')
            {
                int x,y;
                cin>>x>>y;
                update(x,-y);
            }
            else if(s[0]=='E')
                break ;
        }
    }
    return 0;
}

树状数组的模板

三、树状数组的几种变式(区间更新,区间查询)

上面介绍的是最普通的单点更新,区间查询,但如果有些时候是区间更新,单点求和怎么半,又或是区间更新,区间求和怎么办。这里将介绍各种情况该怎么写。

如果上面的单点更新,区间查询还没看懂,建议再思考再往下看。

#1.单点更新、单点查询

传统数组可做

2.单点更新、区间查询

已讲解,详细看上面

3.区间更新、单点查询

这就是第一个问题,如果题目是让你把x-y区间内的所有值全部加上k或者减去k,然后查询操作是问某个点的值,这种时候该怎么做呢。如果是像上面的树状数组来说,就必须把x-y区间内每个值都更新,这样的复杂度肯定是不行的,这个时候,就不能再用数据的值建树了,这里我们引入差分,利用差分建树。

假设我们规定A[0] = 0;

则有 A[i] = Σij = 1D[j];(D[j] = A[j] - A[j-1]),即前面i项的差值和,这个有什么用呢?例如对于下面这个数组

A[] = 1 2 3 5 6 9
D[] = 1 1 1 2 1 3
如果我们把[2,5]区间内值加上2,则变成了

A[] = 1 4 5 7 8 9
D[] = 1 3 1 2 1 1
发现了没有,当某个区间[x,y]值改变了,区间内的差值是不变的,只有D[x]和D[y+1]的值发生改变,至于为什么我想我就不用解释了吧。

所以我们就可以利用这个性质对D[]数组建立树状数组,代码为:

#include <iostream>
#define lowbit(x) x&(-x)
using namespace std;
int a[50005]= {0},c[50005];
int n;
int k;
void update(int i,int k){    //在i位置加上k
while(i<=n){
    c[i]+=k;
    i+=lowbit(i);
}
}
int getsum(int i){   //求D[1 - i]的和,即A[i]值
   int ans=0;
   while(i>0)
    {ans+=c[i];
     i-=lowbit(i);
    }
    return ans;
}
int main()
{  ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1; i<=n; i++)
    {

        cin>>a[i];     //输入初值的时候,也相当于更新了值
        update(i,a[i]-a[i-1]);
    }
    int x,y;
    cin>>x>>y>>k;
     //[x,y]区间内加上k
    update(x,k); //A[x] - A[x-1]增加k
    update(y+1,-k);   //A[y+1] - A[y]减少k
    int i;
       //查询i位置的值
    cin>>i;
    int sum=getsum(i);
    cout<<sum;


    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值