3110: [Zjoi2013]K大数查询

Description

有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c
如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

Input

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

Output

输出每个询问的结果

Sample Input

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

Sample Output

1
2

1

思路：线段树套线段树，外层线段树表示权值，内层线段树维护出现的位置及次数

PS：空间动态开。这里懒惰标记可以不PushDown，感觉PushDown会访问一些本来可以不访问也就是不用开辟的节点，省空间。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxn 50080
#define maxm 30058000
int root[maxn<<2];
int sum[maxm],lazy[maxm],lson[maxm],rson[maxm];
int n,m,cnt;
void scanf_f(int & x)
{
    x = 0;
    char c = getchar();
    while(!(c >= '0' && c <= '9'))
        c = getchar();
    while(c >= '0' && c <= '9')
    {
        x = x*10 + c - '0';
        c = getchar();
    }
}
/*
void PushDown(int node,int l,int r)
{
    if(lazy[node])
    {
        int mid = (l+r) >> 1;
        if(lson[node] == 0) lson[node] = ++cnt;
        if(rson[node] == 0) rson[node] = ++cnt;
        sum[lson[node]] += lazy[node]*(mid-l+1);
        lazy[lson[node]] += lazy[node];
        sum[rson[node]] += lazy[node]*(r-mid);
        lazy[rson[node]] += lazy[node];
    }
    lazy[node] = 0;
}
*/
void PushUp(int node,int l,int r)
{
    sum[node] = sum[lson[node]] + sum[rson[node]] + lazy[node]*(r-l+1);
}
void add(int & node,int L,int R,int l,int r)
{
    if(!node)   node = ++cnt;
    if(L == l && R == r)
    {
        sum[node] += r-l+1;
        lazy[node]++;
        return;
    }
    //PushDown(node,L,R);
    int mid = (L+R) >> 1;
    if(r <= mid)    add(lson[node],L,mid,l,r);
    else if(l > mid)    add(rson[node],mid+1,R,l,r);
    else add(lson[node],L,mid,l,mid),add(rson[node],mid+1,R,mid+1,r);
    PushUp(node,L,R);
}
void Add(int l,int r,int k)
{
    int node = 1;
    int L = 1,R = n;
    while(L < R)
    {
        int mid = (L+R) >> 1;
        add(root[node],1,n,l,r);
        if(k > mid)
        {
            node = node << 1 | 1;
            L = mid+1;
        }
        else
        {
            node = node << 1;
            R = mid;
        }
    }
    add(root[node],1,n,l,r);
}
int query(int node,int L,int R,int l,int r)
{
    if(!node)   return 0;
    if(L == l && R == r)    return sum[node];
    //PushDown(node,L,R);
    int ans = lazy[node]*(r-l+1);
    int mid = (L+R) >> 1;
    if(mid >= r)   return query(lson[node],L,mid,l,r)+ans;
    else if(mid < l)    return query(rson[node],mid+1,R,l,r)+ans;
    else return query(lson[node],L,mid,l,mid) + query(rson[node],mid+1,R,mid+1,r) + ans;
}

int Query(int l,int r,int k)
{
    int node = 1;
    int L = 1,R = n;
    while(L < R)
    {
        int mid = (L+R) >> 1;
        int temp = query(root[node<<1|1],1,n,l,r);
        if(temp >= k)
        {
            L = mid+1;  node = node << 1 | 1;
        }
        else
        {
            R = mid;    node = node << 1;
            k -= temp;
        }
    }
    return L;
}

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= m;i++)
    {
        int opt,l,r,c;
        //scanf("%d%d%d%d",&opt,&l,&r,&c);
        scanf_f(opt);
        scanf_f(l);
        scanf_f(r);
        scanf_f(c);
        if(opt == 1)
        {
            Add(l,r,c);
        }
        else
        {
            printf("%d\n",Query(l,r,c));
        }
    }
    return 0;
}