bzoj3110 K大数查询

最新推荐文章于 2018-10-21 16:58:17 发布

我不会额

最新推荐文章于 2018-10-21 16:58:17 发布

阅读量287

点赞数

分类专栏： CDQ分治

本文链接：https://blog.csdn.net/ACVector/article/details/78209882

版权

CDQ分治专栏收录该内容

8 篇文章 0 订阅

订阅专栏

题目：有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c

如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

N<=50000

M<=50000

思路：二分答案，要用到树状数组区间更新

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<ctime>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
//0x3f3f3f3f

const int maxn=5e4+50;
struct Node{
    int type,id,a,b;
    LL c;
}q[maxn],q1[maxn],q2[maxn];
struct BIT{
    int n;
    LL bit0[maxn],bit1[maxn];
    void init(int _n){
        n=_n;
        mm(bit0,0);
        mm(bit1,0);
    }
    void add(LL *b,int i,LL val){
        for(;i<=n;i+=i&(-i))
            b[i]+=val;
    }
    LL sum(LL *b,int i){
        LL ret=0;
        for(;i>0;i-=i&(-i))
            ret+=b[i];
        return ret;
    }
    void ADD(int l,int r,LL c){
        add(bit0,l,-c*(l-1));
        add(bit0,r+1,c*r);
        add(bit1,l,c);
        add(bit1,r+1,-c);
    }
    LL SUM(int l,int r){
        LL ret=sum(bit0,r)-sum(bit0,l-1);
        ret+=sum(bit1,r)*r-sum(bit1,l-1)*(l-1);
        return ret;
    }
}bit;
int ans[maxn],vis[maxn];
int n,m;
void cdq(int L,int R,int l,int r){
    if(L>R) return;
    if(l==r){
        for(int i=L;i<=R;i++)
            if(q[i].type==2)
                ans[q[i].id]=l;
        return;
    }
    int sz1=0,sz2=0,mid=(l+r)>>1;
    for(int i=L;i<=R;i++){
        if(q[i].type==1){
            if(q[i].c<=mid) q1[sz1++]=q[i];
            else{
                bit.ADD(q[i].a,q[i].b,1);
                q2[sz2++]=q[i];
            }
        }
        else{
            LL ret=bit.SUM(q[i].a,q[i].b);
            if(ret>=q[i].c){
                q2[sz2++]=q[i];
            }
            else{
                q[i].c-=ret;
                q1[sz1++]=q[i];
            }
        }
    }
    for(int i=0;i<sz2;i++)
        if(q2[i].type==1)
            bit.ADD(q2[i].a,q2[i].b,-1);
    memcpy(q+L,q1,sz1*sizeof(Node));
    memcpy(q+L+sz1,q2,sz2*sizeof(Node));
    cdq(L,L+sz1-1,l,mid);
    cdq(L+sz1,R,mid+1,r);
}
int main(){

//    freopen("D:\\input.txt","r",stdin);
//    freopen("D:\\output.txt","w",stdout);
    while(~scanf("%d%d",&n,&m)){
        bit.init(n);
        mm(vis,0);
        mm(ans,0);
        int mi=n,mx=-n;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d%lld",&q[i].type,&q[i].a,&q[i].b,&q[i].c);
            q[i].id=i;
            if(q[i].type==2)
                vis[i]=1;
            else{
                mi=min(mi,(int)q[i].c);
                mx=max(mx,(int)q[i].c);
            }
        }
        cdq(1,m,mi,mx);
        for(int i=1;i<=m;i++)
            if(vis[i])
                printf("%d\n",ans[i]);
    }
    return 0;
}