P3332 [ZJOI2013]K大数查询

https://www.luogu.org/problemnew/show/P3332

题目描述

有N个位置，M个操作。操作有两种，每次操作如果是：

1 a b c：表示在第a个位置到第b个位置，每个位置加上一个数c

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

输入输出格式

输入格式：

 

第一行N，M接下来M行，每行形如1 a b c或2 a b c

 

输出格式：

 

输出每个询问的结果

 

输入输出样例

输入样例

2 5
1 1 2 1
1 1 2 2
2 1 1 2
2 1 1 1
2 1 2 3

输出样例

1
2
1

说明

样例说明

第一个操作后位置 1 的数只有 1 ， 位置 2 的数也只有 1 。
第二个操作 后位置 1 的数有 1 、 2 ，位置 2 的数也有 1 、 2 。
第三次询问 位置 1 到位置 1 第 2 大的数是1 。
第四次询问 位置 1 到位置 1 第 1 大的数是 2 。 第五次询问 位置 1 到位置 2 第 3大的数是 1 。

N,M<=50000,N,M<=50000
a<=b<=N
1操作中abs(c)<=N
2操作中c<=long long

　　

#include<bits/stdc++.h>
using namespace std;
int n,m,tot;
int ans[100050];
struct node
{
    int opt,l,r,pos;
    long long val;
}num[100050],tl[100050],tr[100050];
long long c1[50050],c2[50050];
void add(int u,int v)
{
    for(int i=u;i<=n;i+=i&-i)
        c1[i]+=v,c2[i]+=v*u;
}
long long ask(int u)
{
    long long tmp=0;
    for(int i=u;i;i-=i&-i)
        tmp+=(u+1)*c1[i]-c2[i];
    return tmp;
}
void calc_add(int l,int r,int v)
{
    add(l,v);    add(r+1,-v);
}
long long calc_ask(int l,int r)
{
    return ask(r)-ask(l-1);
}
void calc(int L,int R,int l,int r)
{
    if(l==r)
    {
        for(int i=L;i<=R;++i)    ans[num[i].pos]=l;
        return ;
    }
    int mid=(l+r)>>1;
    bool fl=0,fr=0;
    int topl=0,topr=0;
    for(int i=L;i<=R;++i)
        if(num[i].opt==1)
        {
            if(num[i].val<=mid)
                tl[++topl]=num[i];
            else
            {
                calc_add(num[i].l,num[i].r,1);
                tr[++topr]=num[i];
            }
        }
        else
        {
            long long tmp=calc_ask(num[i].l,num[i].r);
            if(num[i].val>tmp)
            {
                num[i].val-=tmp;
                fl=true;
                tl[++topl]=num[i];
            }
            else
            {
                fr=true;
                tr[++topr]=num[i];
            }
        }
    for(int i=L;i<=R;++i)
        if(num[i].opt==1&&num[i].val>mid)
            calc_add(num[i].l,num[i].r,-1);
    for(int i=1;i<=topl;++i)    num[L+i-1]=tl[i];
    for(int i=1;i<=topr;++i)    num[L+topl+i-1]=tr[i];
    if(fl)    calc(L,L+topl-1,l,mid);
    if(fr)    calc(L+topl,R,mid+1,r);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;++i)
    {
        scanf("%d%d%d%lld",&num[i].opt,&num[i].l,&num[i].r,&num[i].val);
        if(num[i].opt==2)    num[i].pos=++tot;
    }
    calc(1,m,-n,n);
    for(int i=1;i<=tot;++i)    printf("%d\n",ans[i]);
    return 0;
}