`
暴风雪
  • 浏览: 375010 次
  • 性别: Icon_minigender_2
  • 来自: 杭州
社区版块
存档分类
最新评论

[线段树]hdoj 1166:敌兵布阵

阅读更多

大致题意:
    给出一个n长度的数组,在线的实现改变某个值,查询一个区间和的功能。

 

大致思路:
    基础线段树。。G++死活过不去,郁闷,用c++水过去了

 

 

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int Max = 50005;

struct data{
    int l, r, num;
}node[3*Max];     //  线段树节点的数据结构。
int num[Max];

void BuildTree(int left, int right, int u){      //  建树。
    node[u].l = left;
    node[u].r = right;
    if(left == right)
        node[u].num = num[left];
    else{
        BuildTree(left, (left+right)/2, 2*u);
        BuildTree((left+right)/2+1, right, 2*u+1);
        node[u].num = node[2*u].num + node[2*u+1].num;
    }
}

int updata(int camp, int w, int u){              //  修改。
    node[u].num += w;
    if(node[u].l == node[u].r) return 0;
    else if(camp <= node[2*u].r) updata(camp, w, 2*u);
    else updata(camp, w, 2*u+1);
    return 0;
}

int query(int left, int right, int u){           //  查询。
    if(node[u].l == left && node[u].r == right)
        return node[u].num;
    if(right <= node[2*u].r)
        return query(left, right, 2*u);
    if(left >= node[2*u+1].l)
        return query(left, right, 2*u+1);
    int a = query(left, (node[u].l+node[u].r)/2, 2*u);
    int b = query((node[u].l+node[u].r)/2+1, right, 2*u+1);
    return a + b;
}

int main()
{
    char str[10];
    int cas,i,j,a,b,c,n,t=0;
    cin>>cas;
    while(cas--)
    {
        cin>>n;
        for(i=1;i<=n;i++)
        {
            //cin>>num[i];
            scanf("%d",&num[i]);
        }
        printf("Case %d:\n",++t);
        BuildTree(1,n,1);
        while(cin>>str&&str[0]!='E')
        {
            if(str[0]=='Q')
            {
                cin>>a>>b;
                //cout<<query(a,b,1)<<endl;
                a=query(a,b,1);
                printf("%d\n",a);
            }
            if(str[0]=='A')
            {
                //cin>>a>>b;
                scanf("%d%d",&a,&b);
                updata(a,b,1);
            }
            if(str[0]=='S')
            {
                scanf("%d%d",&a,&b);
                updata(a,-b,1);
            }
        }
    }
    return 0;
}
0
0
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics