杭电 HOJ 1166 敌兵布阵 解题报告

    直接求超时。用线段树。简单做法直接用2N数组,以1为root,2*i为左节点,2*i+1为右节点。

    求区间[a,b]和时,如果a为右节点,则减去左节点的值。如果b为左节点,则减去右节点的值。a,b迭代为自己的父母。当a,b相等时,加上当前节点的值,即为区间和。

    我也是第一次接触线段树的概念。刚开始也想自己建立一棵树。在网上找到了张昆玮的线段树的讲稿,才发现可以直接用数组。分享下:http://wenku.baidu.com/view/f27db60ee87101f69e319544.html

#include <iostream>
using namespace std;

int s[131072];

int main()
{
    char str[10];
    int cas,cas2=1;
    int i,n,t,m,num,sta,end,sum;

    scanf("%d",&cas);
    while(cas--)
    {
        printf("Case %d:\n",cas2++);
        scanf("%d",&n);

        t=n-1;
        sta=0;
        while(t)
        {
            t>>=1;
            sta++;
        }
        sta=1<<sta;
        end=sta+n;

        for(i=sta;i<end;i++)
            cin>>s[i];

        end=sta*2;
        for(;i<end;i++)
            s[i]=0;

        sta--;
        for(i=sta;i;i--)
            s[i]=s[t=2*i]+s[t+1];

        while(scanf("%s",str) && str[0]!='E')
        {
            scanf("%d%d",&m,&num);
            if(str[0]=='Q')
            {
                t=num;
                sum=0;
                for(m+=sta,t+=sta;m!=t;t>>=1,m>>=1)
                {
                    if(m&1)
                        sum-=s[m^1];
                    if(~t&1)
                        sum-=s[t^1];
                }
                printf("%d\n",sum+s[m]);
            }
            else
            {
                if(str[0]=='S')
                    num=-num;
                t=sta+m;
                while(t)
                {
                    s[t]+=num;
                    t>>=1;
                }
            }
        }
    }
}

    个人感觉代码不是很好看。还有就是不能用cin和cout,会超时。

posted @ 2013-02-27 19:17  SF-_-  阅读(348)  评论(0编辑  收藏  举报