HUD 1166 敵兵佈陣(樹狀數組或者線段樹)【樹狀數組】

#include<iostream>
#include<cstring>
using namespace std;
int n;
int a[50001];
int b[40001];
int lowbit(int i)
{
    return i&(-i);
}
void update(int x,int val)
{
    while(x<=n)
    {
        a[x]+=val;
        x=x+lowbit(x);
    }
}
int sum(int m,int y)
{
    int sum1=0;
    int sum2=0;
    int x=m-1;
    while(x>0)
    {
        sum1+=a[x];
        x=x-lowbit(x);
    }
    while(y>0)
    {
        sum2+=a[y];
        y=y-lowbit(y);
    }
    return sum2-sum1;
}
int main()
{
    int t;
    cin>>t;
    int val;
    int num=0;
    while(t--)
    {
        num++;
        cin>>n;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i=1; i<=n; i++)
        {
            cin>>val;
            update(i,val);
        }
//        for(int i=1;i<=n;i++)
//            cout<<a[i]<<endl;
        char hh[10];
        int p,q;
        int i=0;

        while(cin>>hh)
        {
            if(hh[0]=='Q')
            {
                cin>>p>>q;
                b[i]=sum(p,q);
                i++;
//                cout<<b[i-1]<<endl;
                continue;
            }
            if(hh[0]=='A')
            {
                cin>>p>>q;
                update(p,q);
                continue;
            }
            if(hh[0]=='S')
            {
                cin>>p>>q;
                update(p,-q);
                continue;
            }
            if(hh[0]=='E')
                break;
        }
        cout<<"Case "<<num<<":"<<endl;
        for(int l=0;l<i;l++)
            cout<<b[l]<<endl;

    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章