zoukankan      html  css  js  c++  java
  • 【codevs1080】线段树练习1

    题目描述 Description

    一行N个方格,开始每个格子里都有一个整数。现在动态地提出一些问题和修改:提问的形式是求某一个特定的子区间[a,b]中所有元素的和;修改的规则是指定某一个格子x,加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1≤N<100000,,提问和修改的总数m<10000条。

    输入描述 Input Description

    输入文件第一行为一个整数N,接下来是n行n个整数,表示格子中原来的整数。接下一个正整数m,再接下来有m行,表示m个询问,第一个整数表示询问代号,询问代号1表示增加,后面的两个数x和A表示给位置X上的数值增加A,询问代号2表示区间求和,后面两个整数表示a和b,表示要求[a,b]之间的区间和。

    输出描述 Output Description

    共m行,每个整数

    样例输入 Sample Input

    6

    3

    4

    1 3 5

    2 1 4

    1 1 9

    2 2 6

    样例输出 Sample Output

    22

    22

    数据范围及提示 Data Size & Hint

    1≤N≤100000, m≤10000 。

    分析

    这是一道"线段树/树状数组"的模板题。

    代码

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<algorithm>
    using namespace std;
    const int maxn=100000+5;
    inline int read()
    {
        int x=0,f=1; char ch=getchar();
        while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
        while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();}
        return x*f;
    }
    int n,m,len;
    int a[maxn];
    struct node
    {
        int l,r,lc,rc,c;
    }tr[maxn<<1];
    inline void bt(int x,int y)
    {
        len++; int now=len;
        tr[now].l=x;tr[now].r=y;tr[now].c=0;
        if(x==y) tr[now].c=a[x];
        else
        {
            int mid=(x+y)>>1;
            tr[now].lc=len+1; bt(x,mid);
            tr[now].rc=len+1; bt(mid+1,y);
            tr[now].c=tr[tr[now].lc].c+tr[tr[now].rc].c;
        }
    }
    inline void update(int now,int x,int k)
    {
        int lc=tr[now].lc,rc=tr[now].rc;
        if(tr[now].l==tr[now].r) tr[now].c+=k;
        else
        {
            int mid=(tr[now].l+tr[now].r)>>1;
            if(x<=mid) update(lc,x,k);
            else if(x>=mid+1) update(rc,x,k);
            tr[now].c=tr[lc].c+tr[rc].c;
        }
    }
    inline int query(int now,int x,int y)
    {
        int lc=tr[now].lc,rc=tr[now].rc;
        if(tr[now].l==x&&tr[now].r==y) return tr[now].c;
        else
        {
            int mid=(tr[now].l+tr[now].r)>>1;
            if(y<=mid) return query(lc,x,y);
            else if(x>=mid+1) return query(rc,x,y);
            else return query(lc,x,mid)+query(rc,mid+1,y);
        }
    }    
    int main()
    {
        n=read();
        for(int i=1;i<=n;i++) a[i]=read();
        bt(1,n);
        m=read();
        for(int i=1;i<=m;i++)
        {
            int p,x,y;
            p=read();x=read();y=read();
            if(p==1) update(1,x,y);
            else printf("%d
    ",query(1,x,y));
        }
        return 0;
    }    

     

  • 相关阅读:
    compose 函数实现
    垂直居中的实现方式
    数组去重方法总结
    前端性能优化
    简简单单的几个正则表达式
    ES6
    Vue不兼容IE8原因以及Object.defineProperty详解
    Vuex(二)——关于store
    Vuex(一)——vuejs的状态管理模式
    关于REST的浅显了解
  • 原文地址:https://www.cnblogs.com/bahl/p/7374197.html
Copyright © 2011-2022 走看看