美文网首页
【完全版】线段树初学--HDU 1166 敌兵布阵

【完全版】线段树初学--HDU 1166 敌兵布阵

作者: db5bacb5a79c | 来源:发表于2015-03-25 10:04 被阅读246次

    链接:click here~~ 

    题意:

    第一行一个整数T,表示有T组数据。

    每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。

    接下来每行有一条命令,命令有4种形式:

    (1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30)

    (2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30);

    (3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数;

    (4)End 表示结束,这条命令在每组数据最后出现;

    每组数据最多有40000条命令

    Output

    对第i组数据,首先输出“Case i:”和回车,

    对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

    Sample Input

    1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End

    Sample Output

    Case 1: 6 33 59

    【解题思路】单点更新:最最基础的线段树,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来

    代码:


    #include#include#include#include#include#include#include#include#include#includeusing namespace std;int dir[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};#define mem(a) memset(a,0,sizeof(a))#define LL long long#define Max(a,b) a>b?a:b;#define Min(a,b) a>1;

    build(lchild);

    build(rchild);

    Pushup(rt);

    }

    void update(int p,int add,int l,int r,int rt)  // 单点更新

    {

    if(l==r)

    {

    tree[rt]+=add;

    return ;

    }

    int m=(l+r)>>1;

    if(p<=m) update(p,add,lchild);

    else update(p,add,rchild);

    Pushup(rt);

    }

    int query(int L,int R,int l,int r,int rt)      //区间查询

    {

    if(L<=l&&r<=R)

    {

    return tree[rt];

    }

    int m=(l+r)>>1;

    int sum=0;

    if(L<=m) sum+=query(L,R,lchild);

    if(R>m) sum+=query(L,R,rchild);

    return sum;

    }

    int scan()

    {

    int res = 0, flag = 0;

    char ch;

    if((ch = getchar()) == '-') flag = 1;

    else if(ch >= '0' && ch <= '9') res = ch - '0';

    while((ch = getchar()) >= '0' && ch <= '9')

    res = res * 10 + (ch - '0');

    return flag ? -res : res;

    }

    int main()

    {

    int t,n,m,i,j,tot=1;

    scanf("%d",&t);

    while(t--)

    {

    printf("Case %d:\n",tot++);

    n=scan();

    build(1,n,1);

    char op[10];

    while(scanf("%s",op))

    {

    if(op[0]=='E') break;

    int a,b;

    a=scan();

    b=scan();

    if(op[0]=='Q') printf("%d\n",query(a,b,1,n,1));

    else if(op[0]=='S') update(a,-b,1,n,1);

    else update(a,b,1,n,1);

    }

    }

    return 0;

    }

    相关文章

      网友评论

          本文标题:【完全版】线段树初学--HDU 1166 敌兵布阵

          本文链接:https://www.haomeiwen.com/subject/chbgxttx.html