【生活没有希望】hdu1166敌兵布阵 线段树

线段树水题刷刷，生活没有希望

最近看到代码跟树状数组差不多短的非递归线段树，常数也很小——zkw线段树

于是拿道水题练练手

短到让人身无可恋

void add(int pos,int x){    for(pos+=M+1;pos;pos/=2)    a[pos]+=x;}
int que(int l,int r){    for(l+=M,r+=M+2,ans=0;l^r^1;l>>=1,r>>=1)ans+=((l&1)?0:a[l|1])+((r&1)?a[r-1]:0);return ans;}

单点修改和区间查询（通过前缀和可以变成区间修改单点查询，再加特技标记可以变成区间修区间查）

        M=1;n+=2;
        while(M<n)
            M*=2;
        M--;
        for(int i=1;i<=M+n+1;i++)
            a[i]=0;

初始化（赋初值0是因为这道题多组数据，不然还不用）

#include <cstdio>
int t,n,M,p,l,r,ans,a[131073];char ch;
void add(int pos,int x){    for(pos+=M+1;pos;pos/=2)    a[pos]+=x;}
int que(int l,int r){    for(l+=M,r+=M+2,ans=0;l^r^1;l>>=1,r>>=1)ans+=((l&1)?0:a[l|1])+((r&1)?a[r-1]:0);return ans;}
int main()
{
    scanf("%d",&t);
    for(int T=1;T<=t;T++)
    {
        printf("Case %d:
",T);
        scanf("%d",&n);
        M=1;n+=2;
        while(M<n)
            M*=2;
        M--;
        for(int i=1;i<=M+n+1;i++)
            a[i]=0;
        for(int i=1;i<=n;i++)
            scanf("%d",&p),add(i,p);
        for(ch=getchar();ch!='E' && ch!='Q' && ch!='A' && ch!='S';ch=getchar());
        while(ch!='E')
        {
            switch(ch)
            {
                case'Q':
                    getchar();getchar();getchar();getchar();
                    scanf("%d%d",&l,&r);printf("%d
",que(l,r));
                    break;
                case 'A':
                    getchar();getchar();
                    scanf("%d%d",&l,&r);add(l,r);
                    break;
                case 'S':
                    getchar();getchar();
                    scanf("%d%d",&l,&r);add(l,-r);
                    break;
            }
            for(ch=getchar();ch!='E' && ch!='Q' && ch!='A' && ch!='S';ch=getchar());
        }
        getchar();getchar();
    }
    return 0;
}

这道水题理论上不需要常数那么小，但是代码短所以还是这么打了

主程序基本都是在输入输出，，，