敌兵布阵(线段树或树状数组)

题目传送门

一个既可以用树状数组又可以用线段树写的题。(不过树状数组好像耗时更多,但是写起来比线段树香)
题目就是给一个原数组,要么对一个区间内一起增加或者减少一个数值,要么就询问该段区间的和。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+5;
const int mod=1e9+7;
const int INF=0x7fffffff;
const ll LLINF=0x7fffffffffffffff;
const double EPS=1e-10;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define pii pair<int,int>
#define pll pair<ll,ll>
#define debug cout<<"debug"<<endl;
//#define int long long
int a[N];
struct tree
{
    int l,r,pre,add;
}t[N<<2];
void build(int p,int l,int r)
{
    t[p].l=l;t[p].r=r;
    if(l==r)
    {
        t[p].pre=a[l];
        return ;
    }
    int mid=t[p].l+t[p].r>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
void spread(int p)
{
    if(t[p].add)
    {
        t[p<<1].pre+=t[p].add*(t[p<<1].r-t[p<<1].l+1);
        t[p<<1|1].pre+=t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1);
        t[p<<1].add+=t[p].add;
        t[p<<1|1].add+=t[p].add;
        t[p].add=0;
    }
}
void change(int p,int x,int y,int z)
{
    if(x<=t[p].l&&t[p].r<=y)
    {
        t[p].pre+=z*(t[p].r-t[p].l+1);
        t[p].add+=z;
        return ;
    }
    spread(p);
    int mid=t[p].r+t[p].l>>1;
    if(x<=mid)
        change(p<<1,x,y,z);
    if(y>mid)
        change(p<<1|1,x,y,z);
    t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
int ask(int p,int x,int y)
{
    if(x<=t[p].l&&t[p].r<=y)
    {
        return t[p].pre;
    }
    spread(p);
    int mid=t[p].l+t[p].r>>1;
    int ans=0;
    if(x<=mid)
        ans+=ask(p<<1,x,y);
    if(y>mid)
        ans+=ask(p<<1|1,x,y);
    return ans;
}
signed main()
{
    IOS;
    //freopen("","r",stdin);
    //freopen("","w",stdout);
    int T;
    cin>>T;
    int cnt=0;
    while(T--)
    {
        memset(t,0,sizeof t);
        cnt++;
        cout<<"Case "<<cnt<<":"<<endl;
        int n;
        cin>>n;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
        }
        build(1,1,n);
        string s;
        while(cin>>s)
        {
            if(s=="End")
                break;
            if(s=="Add")
            {
                int i,j;
                cin>>i>>j;
                change(1,i,i,j);
            }
            if(s=="Sub")
            {
                int i,j;
                cin>>i>>j;
                change(1,i,i,-j);
            }
            if(s=="Query")
            {
                int i,j;
                cin>>i>>j;
                int res=ask(1,i,j);
                cout<<res<<endl;
            }
        }
    }
}
发布了93 篇原创文章 · 获赞 9 · 访问量 4204

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/104432833