ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

cdq实现树状数组

2021-09-16 20:05:28  阅读:153  来源: 互联网

标签:cout 树状 int mid cdq 数组 pos1 define


题目链接

思路

毒瘤的一批的东西

也算是cdq的一个应用吧

代码

#include<bits/stdc++.h>
#define fi first
#define se second
#define debug cout<<"I AM HERE"<<endl;
using namespace std;
typedef long long ll;
const int maxn=2e6+5,inf=0x3f3f3f3f,mod=1e9+7;
const double eps=1e-6;
int n,m,cnt;
struct node{
    int opt,x,y;
}a[maxn],b[maxn];
int ans[maxn];

void cdq(int l,int r){
    if(l>=r){
        return ;
    }
    int mid=(l+r)/2,sum=0;
    cdq(l,mid);
    cdq(mid+1,r);
    int pos1=l,pos2=mid+1,id=l;
    while(pos1<=mid&&pos2<=r){
        if(a[pos1].x<=a[pos2].x){
            if(a[pos1].opt==1) sum+=a[pos1].y;
            b[id++]=a[pos1++];
        }else{
            if(a[pos2].opt==2) ans[a[pos2].y]-=sum;
            if(a[pos2].opt==3) ans[a[pos2].y]+=sum;
            b[id++]=a[pos2++];
        }
    }
    while(pos1<=mid) b[id++]=a[pos1++];
    while(pos2<=r){
        if(a[pos2].opt==2) ans[a[pos2].y]-=sum;
        if(a[pos2].opt==3) ans[a[pos2].y]+=sum;
        b[id++]=a[pos2++];
    }
    for(int i=l;i<=r;i++){
        a[i]=b[i];
    }
}

signed main(){
    scanf("%d%d",&n,&m);
    for(int i=1,x;i<=n;i++){
        scanf("%d",&x);
        a[++cnt]={1,i,x};
    }
    int tot=0;
    for(int i=1,opt,x,y;i<=m;i++){
        scanf("%d%d%d",&opt,&x,&y);
        if(opt==1){
            a[++cnt]={1,x,y};
        }else{
            ++tot;
            a[++cnt]={2,x-1,tot};
            a[++cnt]={3,y,tot};
        }
    }
    cdq(1,cnt);
    for(int i=1;i<=tot;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}


标签:cout,树状,int,mid,cdq,数组,pos1,define
来源: https://www.cnblogs.com/hunxuewangzi/p/15299441.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有