BZOJ3110: [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

HINT

【样例说明】

第一个操作 后位置 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<=Maxlongint

Source

第一次写树套树
本题有整体二分写法，但强行当成树套树题练习
注意50000*50000会爆int，改成unsigned int即可

#include <bits/stdc++.h>

using namespace std;

typedef unsigned int Int;

inline Int read()
{
Int sc = 0, f = 1; char ch = getchar();
while( ch < '0' || ch > '9' ) { if( ch == '-') f = -1; ch = getchar(); }
while( ch >= '0' && ch <= '9' ) sc = sc * 10 + ch - '0', ch = getchar();
return sc * f;
}

const Int MAXN = 50050;
const Int MAXM = MAXN << 8;

Int n, m, L, R, c, tag[MAXM], root[MAXN << 2], cnt, sum[MAXM], ls[MAXM], rs[MAXM];

inline void modify(Int &x, Int l, Int r, Int ql, Int qr, Int v)
{
if( !x ) x = ++cnt;
if( l == ql && r == qr ) { sum[ x ] += r - l + 1; tag[ x ]++; return ; }
Int mid = l + r >> 1;
if( qr <= mid ) modify( ls[ x ], l, mid, ql, qr, v );
else if( ql > mid ) modify( rs[ x ], mid + 1, r, ql, qr, v );
else modify( ls[ x ], l, mid, ql, mid, v ), modify( rs[ x ], mid + 1, r, mid + 1, qr, v );
sum[ x ] = sum[ ls[ x ] ] + sum[ rs[ x ] ] + tag[ x ] * ( r - l + 1 );
}

inline void update()
{
Int l = 1, r = n, x = 1;
while( l < r )
{
modify( root[ x ], 1, n, L, R, c );
Int mid = l + r >> 1;
if( c <= mid ) r = mid, x <<= 1;
else l = mid + 1, ( x <<= 1 ) |= 1;
}
modify( root[ x ], 1, n, L, R, c );
}

inline Int get(Int x, Int l, Int r, Int ql, Int qr)
{
if( l == ql && r == qr ) return sum[ x ];
Int mid = l + r >> 1, ans = ( min( qr, r ) - max( ql, l ) + 1 ) * tag[ x ];
if( qr <= mid ) return get( ls[ x ], l, mid, ql, qr ) + ans;
if( ql > mid ) return get( rs[ x ], mid + 1, r, ql, qr ) + ans;
return get( ls[ x ], l, mid, ql, mid ) + get( rs[ x ], mid + 1, r, mid + 1, qr ) + ans;
}

inline Int query()
{
Int l = 1, r = n, x = 1;
while( l < r )
{
int mid = l + r >> 1, tmp;
if( ( tmp = get( root[ x << 1 ], 1, n, L, R ) ) >= c ) r = mid, x <<= 1;
else l = mid + 1, ( x <<= 1 ) |= 1, c -= tmp;
}
return l;
}

int main()
{
n = read(), m = read();
while( m-- )
{
Int opt = read();
L = read(), R = read(), c = read();
if( opt == 1 ) c = n - c + 1, update();
else printf( "%d\n", n - query() + 1 );
}
}