HDU - 1166 A - 敌兵布阵 单点更新+区间查询

题意:如题

线段树

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
ll arr[maxn];
char str[maxn];
/// Pushup 函数更新节点信息，这里是求和
struct SegTree
{
    ll sum[maxn<<2],cnt[maxn<<2],lz[maxn<<2];
    void init()     /// 初始化
    {
        memset(sum,0,sizeof(sum));
        memset(cnt,0,sizeof(cnt));
        memset(lz,0,sizeof(lz));
    }
    void push_up(int rt)    /// 向上更新(result)
    {
        sum[rt]=sum[rt<<1]+sum[rt<<1|1];
        cnt[rt]=cnt[rt<<1]+cnt[rt<<1|1];
    }
    void push_down(int rt)  /// 更下更新(laze)
    {
        if(!lz[rt])return;
        sum[rt<<1]+=cnt[rt<<1]*lz[rt];
        sum[rt<<1|1]+=cnt[rt<<1|1]*lz[rt];
        lz[rt<<1]+=lz[rt];
        lz[rt<<1|1]+=lz[rt];
        lz[rt]=0;
    }
    void build(ll arr[],int l,int r,int rt)                /// 数组建立线段树
    {
        if(l == r) {
            sum[rt]=arr[l];
            cnt[rt] ++;
            return;
        }
        push_down(rt);
        int mid = (l + r) >> 1;
        build(arr,l,mid,rt<<1);
        build(arr,mid+1,r,rt<<1|1);
        push_up(rt);
    }
    void activate(int pos,ll val,int l,int r,int rt)       /// 单点建立线段树
    {
        if(l==r)
        {
            sum[rt]+=val;
            cnt[rt]++;
            return;
        }
        push_down(rt);
        int mid=(l + r) >> 1;
        if(pos<=mid) activate(pos,val,l,mid,rt<<1);
        else activate(pos,val,mid+1,r,rt<<1|1);
        push_up(rt);
    }
    void update(int pos,ll val,int l,int r,int rt)         /// 单点更新
    {
        if(l == r) {
            sum[rt] += val;
            lz[rt] += val;
            return;
        }
        push_down(rt);
        int mid = (l + r) >> 1;
        if(pos <= mid) update(pos,val,l,mid,rt<<1);
        else update(pos,val,mid+1,r,rt<<1|1);
        push_up(rt);
    }
    void update(int ql,int qr,ll val,int l,int r,int rt)   /// 区间更新
    {
        if(ql==l && qr==r)
        {
            sum[rt]+=cnt[rt]*val;
            lz[rt]+=val;
            return;
        }
        push_down(rt);
        int mid=(l + r) >> 1;
        if(qr<=mid) update(ql,qr,val,l,mid,rt<<1);
        else if(ql>mid) update(ql,qr,val,mid+1,r,rt<<1|1);
        else
        {
            update(ql,mid,val,l,mid,rt<<1);
            update(mid+1,qr,val,mid+1,r,rt<<1|1);
        }
        push_up(rt);
    }
    ll query(int ql,int qr,int l,int r,int rt)              /// 区间查询
    {
        if(ql==l && qr==r)return sum[rt];
        push_down(rt);
        int mid=(l + r) >> 1;
        if(qr<=mid)return query(ql,qr,l,mid,rt<<1);
        if(ql>mid)return query(ql,qr,mid+1,r,rt<<1|1);
        return query(ql,mid,l,mid,rt<<1)+query(mid+1,qr,mid+1,r,rt<<1|1);
    }
}seg;
int main()
{
    int caset,t = 0;scanf("%d",&caset);
    while(caset--) {
        int n,l,r,pos;
        ll x;
        seg.init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%lld",&arr[i]);
        seg.build(arr,1,n,1);
        printf("Case %d:\n",++t);
        while(scanf("%s",str) && str[0] != 'E')
        {
            if(str[0] == 'A') {
                scanf("%d%lld",&pos,&x);
                seg.update(pos,x,1,n,1);
            }
            else if(str[0] == 'S') {
                scanf("%d%lld",&pos,&x);
                seg.update(pos,-x,1,n,1);
            }
            else if(str[0] == 'Q') {
                scanf("%d%d",&l,&r);
                printf("%lld\n",seg.query(l,r,1,n,1));
            }
        }
    }
    return 0;
}

我们知道单点修改+区间查询可以用树状数组来做

树状数组代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
ll arr[maxn];
char str[maxn];
struct BIT {
    ll c[maxn];
    ll lowbit(ll x) { return x&-x; }
    void init() {
        memset(c,0,sizeof(c));
    }
    void Update(int pos,ll x,int n) {
        for(int i=pos;i<=n;i += lowbit(i))
            c[i] += x;
    }
    ll Query(int pos) {
        ll ans = 0;
        for(int i=pos;i;i-=lowbit(i))
            ans += c[i];
        return ans;
    }

}bit;
int main()
{
    int caset,t = 0;scanf("%d",&caset);
    while(caset--) {
        printf("Case %d:\n",++t);
        bit.init();
        int n,l,r,pos;
        ll x;
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%lld",&arr[i]),bit.Update(i,arr[i],n);
        while(scanf("%s",str) && str[0] != 'E')
        {
            if(str[0] == 'A') {
                scanf("%d%lld",&pos,&x);
                bit.Update(pos,x,n);
            }
            else if(str[0] == 'S') {
                scanf("%d%lld",&pos,&x);
                bit.Update(pos,-x,n);
            }
            else if(str[0] == 'Q') {
                scanf("%d%d",&l,&r);
                printf("%lld\n",bit.Query(r) - bit.Query(l-1));
            }
        }
    }
    return 0;
}