HDU 1166 线段树单点更新

简介:
第一行一个整数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条命令

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define N 50005
using namespace std;
int num[N];
struct Tree
{
    int l;  //左端点
    int r;  //右端点
    int sum; //总数
} tree[N*4]; // 总线段的长度为N,开数组的话一般开到N 的四倍
void build(int root,int l,int r) // root表示根节点 ,他的区间范围【l,r】
{
    tree[root].l=l;
    tree[root].r=r;
    if(tree[root].l==tree[root].r) // 当左右端点相等时就是叶子节点
    {
        tree[root].sum=num[l];        // 赋除值
        return;                    //  递归出口
    }
    int mid=(l+r)/2;
    build(root<<1,l,mid);             // k<<1相等于k*2 即是他的左孩子
    build(root<<1|1,mid+1,r);         // k<<1|1 相当于k*2+1 ,即是他的右孩子
    tree[root].sum = tree[root<<1].sum + tree[root<<1|1].sum; // 父亲的sum  = 左孩子的sum  + 右孩子的sum
}
void update(int root,int pos,int val) // root是根节点,pos,val表示:我们要跟新在pos点出的值更新为val
{
    if(tree[root].l==tree[root].r)    //  如果是叶子节点,即是pos对应的位置
    {
        tree[root].sum=val;           //  更新操作
        return;                       //  递归出口
    }
    int mid=(tree[root].l + tree[root].r)/2;
    if(pos<=mid)                      //  如果pos点是在root对应的左孩子的话,就调用update(k<<1,pos,val); 在左孩子里找
        update(root<<1,pos,val);
    else
        update(root<<1|1,pos,val);
    tree[root].sum = tree[root<<1].sum + tree[root<<1|1].sum; // 父亲的sum  = 左孩子的sum  + 右孩子的sum
}

int query(int root,int L,int R) // root表示根节点,[L,R]表示要查询的区间
{
    if(L<=tree[root].l&&R>=tree[root].r)   // [L,R]要查询的区间 包含root节点表示的区间 直接返回root节点的sum值
        return tree[root].sum;
    int mid=(tree[root].l + tree[root].r)/2,ret=0;
    if(L<=mid) ret+=query(root<<1,L,R);        //  查询root节点的左孩子
    if(R>mid)  ret+=query(root<<1|1,L,R);      //  查询root节点的右孩子
    return ret;                             //  返回
}
int main()
{
    int t,n,ca=0;
    char s[10];
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
            scanf("%d",&num[i]);
        build(1,1,n);
        printf("Case %d:\n",++ca);
        while(scanf("%s",s),s[0]!='E')
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(s[0]=='Q')
            {
                if(a>b)
                    swap(a,b);
                printf("%d\n",query(1,a,b));
            }
            if(s[0]=='A')
                num[a]+=b,update(1,a,num[a]);
            if(s[0]=='S')
                num[a]-=b,update(1,a,num[a]);
        }
    }
    return 0;
}


目录
相关文章
|
6月前
poj 1990 MooFest 树状数组
题意就是有N头牛,每头牛都有一个坐标和声调值(x, v),两头牛之间通讯要花费的能量是他们的距离乘以最大的一个音调值,现在要任意两头牛之间都相互通讯一次,求总共需要花费多少能量?
20 0
|
7月前
|
算法
数星星(树状数组模板题)
数星星(树状数组模板题)
27 0
|
人机交互
POJ-2524,Ubiquitous Religions(并查集模板题)
POJ-2524,Ubiquitous Religions(并查集模板题)
|
人工智能 BI 存储
|
人工智能 网络架构
|
人工智能
POJ 1804 Brainman(5种解法,好题,【暴力】,【归并排序】,【线段树单点更新】,【树状数组】,【平衡树】)
Brainman Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 10575   Accepted: 5489 Description BackgroundRaymond Babbitt drives his brother Charlie mad.
1236 0
|
Java
HDU 1754 I Hate It(线段树之单点更新,区间最值)
I Hate It Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 70863    Accepted Submission(s): 27424 Problem Description 很多学校流行一种比较的习惯。
1079 0
线段树-poj-2823
Sliding Window Description An array of size n ≤ 106 is given to you. There is a sliding window of size k which is moving from the very left of the array to the very right. You can only see the k
1057 0
|
C++ 消息中间件 定位技术
nyoj题目20吝啬的国度【深搜】
吝啬的国度 时间限制:1000 ms  |  内存限制:65535 KB 难度:3 描述 在一个吝啬的国度里有N个城市,这N个城市间只有N-1条路把这个N个城市连接起来。
755 0