树状数组 1 [cdq分治]

树状数组 1
题目描述
如题,已知一个数列,你需要进行下面两种操作:

1.将某一个数加上x

2.求出某区间每一个数的和

输入格式
第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3个整数,表示一个操作,具体如下:

操作1: 格式:1 x k 含义:将第x个数加上k

操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和

输出格式
输出包含若干行整数,即为所有操作2的结果。

输入输出样例
输入 #1 复制
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4
输出 #1 复制
14
16
说明/提示
时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=8,M<=10

对于70%的数据:N<=10000,M<=10000

对于100%的数据:N<=500000,M<=500000

样例说明:
树状数组 1 [cdq分治]_第1张图片
故输出结果14、16

题意:就简单的树状数组操作,单点修改,区间查询
题解:将单点修改对应一个点,将区间查询改成两个前缀和相减,一个区间[x,y]对应两个点x-1,y,然后按照操作的顺序存储各个点,可以发现每个点有两维属性[time,pos],time是操作发生的时间,由于是按照操作顺序存储的各点,所以在时间这一维已经有序,每个操作只会对后面的操作产生影响,则使用cdq分治排序另一维pos(每个点的位置)并在这个过程计算前面操作对后面操作的影响即可

#include
using namespace std;
#define debug(x) cout<<#x<<" is "<
typedef long long ll;
const int maxn=1e6+5;
struct pot{
    int type;
    int pos;
    ll val;
}p[maxn<<1],tmp[maxn<<1];
ll ans[maxn];
void solve(int l,int r){
    if(l==r)return;
    int mid=(l+r)>>1;
    solve(l,mid);
    solve(mid+1,r);
    int x=l;
    int y=mid+1;
    ll sum=0;
    int s=l;
    while(x<=mid&&y<=r){
        if(p[x].pos<=p[y].pos){
            if(p[x].type==2){sum+=p[x].val;}
            tmp[s++]=p[x++];
        }
        else{
            if(p[y].type!=2){ans[p[y].val]+=p[y].type*sum;}
            tmp[s++]=p[y++];
        }
    }
    while(y<=r){if(p[y].type!=2)ans[p[y].val]+=p[y].type*sum;tmp[s++]=p[y++];}
    while(x<=mid){tmp[s++]=p[x++];}
    for(int i=l;i<=r;i++){
        p[i]=tmp[i];
    }
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){p[i].type=2;p[i].pos=i;scanf("%lld",&p[i].val);}
    int t1=n;
    int t2=0;
    while(m--){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        if(a==1){
            p[++t1].pos=b;
            p[t1].type=2;
            p[t1].val=c;
        }
        else{
            p[++t1].pos=b-1;
            p[t1].type=-1;
            p[t1].val=++t2;
            p[++t1].pos=c;
            p[t1].type=1;
            p[t1].val=t2;
        }
    }
    solve(1,t1);
    for(int i=1;i<=t2;i++){
        printf("%lld\n",ans[i]);
    }
    return 0;
}

你可能感兴趣的:(cdq分治)