码迷,mamicode.com
首页 > 编程语言 > 详细

Libre OJ 130、131、132 (树状数组 单点修改、区间查询 -> 区间修改,单点查询 -> 区间修改,区间查询)

时间:2019-03-14 01:08:10      阅读:223      评论:0      收藏:0      [点我收藏+]

标签:psu   模板题   col   ups   cst   head   stream   font   inner   

#130. 树状数组 1 :单点修改,区间查询

题目链接:https://loj.ac/problem/130

题目描述

这是一道模板题。

给定数列 a[1], a[2], \dots, a[n]a[1],a[2],,a[n],你需要依次进行 qq 个操作,操作有两类:

  • 1 i x:给定 i,xi,x,将 a[i]a[i] 加上 xx;
  • 2 l r:给定 l,rl,r,求 \sum_{i=l}^ra[i]i=lr?a[i] 的值(换言之,求 a[l]+a[l+1]+\dots+a[r]a[l]+a[l+1]+?+a[r] 的值)

输入格式

第一行包含 22 个正整数 n,qn,q,表示数列长度和询问个数。保证 1\le n,q\le 10^61n,q106。
第二行 nn 个整数 a[1], a[2], \dots, a[n]a[1],a[2],,a[n],表示初始数列。保证 |a[i]|\le 10^6a[i]106。
接下来 qq 行,每行一个操作,为以下两种之一:

  • 1 i x:给定 i,xi,x,将 a[i]a[i] 加上 xx;
  • 2 l r:给定 l,rl,r,求 \sum_{i=l}^ra[i]i=lr?a[i] 的值。

保证 1\le l\le r\le n,1lrn|x|\le 10^6x106。

输出格式

对于每个 2 l r 操作输出一行,每行有一个整数,表示所求的结果。

样例

样例输入

3 2
1 2 3
1 2 0
2 1 3

样例输出

6

数据范围与提示

对于所有数据,1\le n,q\le 10^6,1n,q106|a[i]|\le 10^6a[i]106, 1\le l\le r\le n,1lrn|x|\le 10^6x106。

思路:板子

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6;
int n,m;
ll sum[maxn];
int lowbit(int x){return x&(-x);}
void add(int x,ll val){
    while(x<=maxn){
        sum[x]+=val;
        x+=lowbit(x);
    }
}
ll getsum(int x){
    ll res=0;
    while(x){
        res+=sum[x];
        x-=lowbit(x);
    }
    return res;
}
int main(){
    scanf("%d%d",&n,&m);
    ll c;
    for(int i=1;i<=n;i++){
        scanf("%lld",&c);
        add(i,c);
    }
    while(m--){
        int id,x,y;
        scanf("%d%d%d",&id,&x,&y);
        if(id==1)add(x,y);
        else
            cout<<getsum(y)-getsum(x-1)<<endl;
    }
    return 0;
}

 

#131. 树状数组 2 :区间修改,单点查询

题目链接:https://loj.ac/problem/131

题目描述

这是一道模板题。

给定数列 a[1], a[2], \dots, a[n]a[1],a[2],,a[n],你需要依次进行 qq 个操作,操作有两类:

  • 1 l r x:给定 l,r,xl,r,x,对于所有 i\in[l,r]i[l,r],将 a[i]a[i] 加上 xx(换言之,将 a[l], a[l+1], \dots, a[r]a[l],a[l+1],,a[r] 分别加上 xx);
  • 2 i:给定 ii,求 a[i]a[i] 的值。

输入格式

第一行包含 22 个正整数 n,qn,q,表示数列长度和询问个数。保证 1\le n,q\le 10^61n,q106。
第二行 nn 个整数 a[1], a[2], \dots, a[n]a[1],a[2],,a[n],表示初始数列。保证 |a[i]|\le 10^6a[i]106。
接下来 qq 行,每行一个操作,为以下两种之一:

  • 1 l r x:对于所有 i\in[l,r]i[l,r],将 a[i]a[i] 加上 xx;
  • 2 i:给定 ii,求 a[i]a[i] 的值。

保证 1\le l\le r\le n,1lrn|x|\le 10^6x106。

输出格式

对于每个 2 i 操作,输出一行,每行有一个整数,表示所求的结果。

样例

样例输入

3 2
1 2 3
1 1 3 0
2 2

样例输出

2

数据范围与提示

对于所有数据,1\le n,q\le 10^6,1n,q106|a[i]|\le 10^6a[i]106, 1\le l\le r\le n,1lrn|x|\le 10^6x106。

思路:这里需要用到差分数组,我们定义sum【i】为第i个数与第i-1个数的差,即sum【i】=a【i】-a【i-1】,这就使得a【i】=sum【1】+sum【2】+……sum【i】,就是sum数组的前缀和了,我们用数组数组维护sum数组的前缀和就好了。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll maxn=1e6+10;
ll sum[maxn];
int n,q;
int lowbit(int x){return x&(-x);}
void add(int x,int val){
    while(x<=n){
        sum[x]+=val;
        x+=lowbit(x);
    }
}
ll ask(int x){
    ll res=0;
    while(x){
        res+=sum[x];
        x-=lowbit(x);
    }
    return res;
}
int main(){
    cin>>n>>q;
    ll tmp=0;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        add(i,x-tmp);
        tmp=x;
    }
    while(q--){
        int id,l,r,x;
        cin>>id;
        if(id==1){
            cin>>l>>r>>x;
            add(l,x); add(r+1,-x);
        }
        else{
            cin>>x;
            cout<<ask(x)<<endl;
        }
    }
    return 0;
}

 

#132. 树状数组 3 :区间修改,区间查询

题目链接:https://loj.ac/problem/132

题目描述

这是一道模板题。

给定数列 a[1], a[2], \dots, a[n]a[1],a[2],,a[n],你需要依次进行 qq 个操作,操作有两类:

  • 1 l r x:给定 l,r,xl,r,x,对于所有 i\in[l,r]i[l,r],将 a[i]a[i] 加上 xx(换言之,将 a[l], a[l+1], \dots, a[r]a[l],a[l+1],,a[r] 分别加上 xx);
  • 2 l r:给定 l,rl,r,求 \sum_{i=l}^ra[i]i=lr?a[i] 的值(换言之,求 a[l]+a[l+1]+\dots+a[r]a[l]+a[l+1]+?+a[r] 的值)。

输入格式

第一行包含 22 个正整数 n,qn,q,表示数列长度和询问个数。保证 1\le n,q\le 10^61n,q106。
第二行 nn 个整数 a[1],a[2],\dots,a[n]a[1],a[2],,a[n],表示初始数列。保证 |a[i]|\le 10^6a[i]106。
接下来 qq 行,每行一个操作,为以下两种之一:

  • 1 l r x:对于所有 i\in[l,r]i[l,r],将 a[i]a[i] 加上 xx;
  • 2 l r:输出 \sum_{i=l}^ra[i]i=lr?a[i] 的值。

保证 1\le l\le r\le n,1lrn|x|\le 10^6x106。

输出格式

对于每个 2 l r 操作,输出一行,每行有一个整数,表示所求的结果。

样例

样例输入

5 10
2 6 6 1 1
2 1 4
1 2 5 10
2 1 3
2 2 3
1 2 2 8
1 2 3 7
1 4 4 10
2 1 2
1 4 5 6
2 3 4

样例输出

15
34
32
33
50

数据范围与提示

对于所有数据,1\le n,q\le 10^6,1n,q106|a[i]|\le 10^6a[i]106, 1\le l\le r\le n,1lrn|x|\le 10^6x106。

思路:与上一题差不多,我们继续用一个数组sum1【i】存第i个数与第i-1个数的差,即

sum1【x】=a【i】-a【x-1】,a【x】=sum1【1】+sum1【2】+……sum1【x】

我们也可以很容易得出:

a【1】+a【2】+……a【x】=sum1【1】+(sum1【1】+sum1【2】)+……(sum1【1】+sum1【2】+……sum1【x】)

                                            =x*sum1【1】+(x-1)*sum1【2】+……sum1【x】

                                            =x*(sum1【1】+sum1【2】+……sum1【x】)-∑(i=1-x)(i-1)*sum1[i]

,所以我们就多建立一个数组sum2用来维护(x-1)sum1【x】就好了。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll maxn=1e6+10;
ll sum1[maxn],sum2[maxn],a[maxn];
int n,q;
int lowbit(int x){return x&(-x);}
void add(int x,ll val){
    for(int i=x;i<=n;i+=lowbit(i)){
        sum1[i]+=val;
        sum2[i]+=(x-1)*val;
    }
}
ll ask(int x){
    ll res=0;
    for(int i=x;i;i-=lowbit(i)){
        res+=x*sum1[i]-sum2[i];
    }
    return res;
}
int main(){
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        add(i,a[i]-a[i-1]);
    }
    while(q--){
        int id,l,r,x;
        scanf("%d",&id);
        if(id==1){
            scanf("%d%d%d",&l,&r,&x);
            add(l,x); add(r+1,-x);
        }
        else{
            scanf("%d%d",&l,&r);
            printf("%lld\n",ask(r)-ask(l-1));
        }
    }
    return 0;
}

 

Libre OJ 130、131、132 (树状数组 单点修改、区间查询 -> 区间修改,单点查询 -> 区间修改,区间查询)

标签:psu   模板题   col   ups   cst   head   stream   font   inner   

原文地址:https://www.cnblogs.com/zjl192628928/p/10527577.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!