BZOJ 1901: Zju2112 Dynamic Rankings( BIT 套 BST )
BIT 套 splay
其实也是不难...每个 BIT 的结点保存一颗 splay , 询问就二分答案然后判断rank...
---------------------------------------------------------------
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<iostream>
#define rep( i , n ) for( int i = 0 ; i < n ; ++i )
#define clr( x , c ) memset( x , c , sizeof( x ) )
#define Rep( i , n ) for( int i = 1 ; i <= n ; ++i )
#define lowbit( x ) ( ( x ) & ( -x ) )
using namespace std;
const int maxn = 10000 + 5;
const int maxnode = 5e6;
const int inf = 2e9;
int n , mn , mx;
int seq[ maxn ];
struct Node* null;
queue< Node* > Q;
struct Node {
Node *ch[ 2 ] , *p;
int v , s;
Node( int _ = inf ) : v( _ ) , s( 1 ) {
ch[ 0 ] = ch[ 1 ] = p = null;
}
inline bool d() {
return this == p -> ch[ 1 ];
}
inline void setc( Node* c , int d ) {
ch[ d ] = c;
c -> p = this;
}
inline void upd() {
s = ch[ 0 ] -> s + ch[ 1 ] -> s + 1;
}
void* operator new( size_t ) {
Node* t = Q.front(); Q.pop();
return t;
}
};
Node pool[ maxnode ];
struct SPLAY {
Node* root;
void rot( Node* t ) {
Node* p = t -> p;
int d = t -> d();
p -> p -> setc( t , p -> d() );
p -> setc( t -> ch[ d ^ 1 ] , d );
t -> setc( p , d ^ 1 );
p -> upd();
if( p == root )
root = t;
}
void splay( Node* t , Node* f = null ) {
for( Node* p = t -> p ; p != f ; p = t -> p ) {
if( p -> p != f )
p -> d() != t -> d() ? rot( t ) : rot( p );
rot( t );
}
t -> upd();
}
Node* select( int k ) {
for( Node* t = root ; ; ) {
int s = t -> ch[ 0 ] -> s;
if( k == s ) return t;
if( k > s )
k -= s + 1 , t = t -> ch[ 1 ];
else
t = t -> ch[ 0 ];
}
}
int rank( int v ) {
int ans = 0;
for( Node* t = root ; t != null ; )
if( t -> v < v )
ans += t -> ch[ 0 ] -> s + 1 , t = t -> ch[ 1 ];
else
t = t -> ch[ 0 ];
return ans;
}
void insert( int v ) {
int k = rank( v );
Node *L = select( k - 1 ) , *R = select( k );
splay( L ) , splay( R , L );
R -> setc( new Node( v ) , 0 );
R -> upd();
root -> upd();
}
void del( int v ) {
int k = rank( v );
Node *L = select( k - 1 ) , *R = select( k + 1 );
splay( L );
splay( R , L );
Q.push( R -> ch[ 0 ] );
R -> setc( null , 0 );
R -> upd();
L -> upd();
}
};
SPLAY h[ maxn ];
void init() {
rep( i , maxnode ) Q.push( pool + i );
null = new Node();
null -> s = 0;
Rep( i , n ) {
h[ i ].root = new Node();
h[ i ].root -> setc( new Node( -inf ) , 0 );
h[ i ].root -> upd();
}
}
inline void del( int x , int v ) {
for( ; x <= n ; x += lowbit( x ) )
h[ x ].del( v );
}
inline void insert( int x , int v ) {
for( ; x <= n ; x += lowbit( x ) )
h[ x ].insert( v );
}
inline int rank( int l , int r , int v ) {
int ans = 0;
for( l-- ; l ; l -= lowbit( l ) )
ans -= h[ l ].rank( v ) - 1;
for( ; r ; r -= lowbit( r ) )
ans += h[ r ].rank( v ) - 1;
return ans + 1;
}
int query( int l , int r , int k ) {
int L = mn , R = mx , ans;
while( L <= R ) {
int m = ( L + R ) >> 1;
if( rank( l , r , m ) > k )
R = m - 1;
else
L = ( ans = m ) + 1;
}
return ans;
}
void modify( int pos , int v ) {
del( pos , seq[ pos ] );
insert( pos , seq[ pos ] = v );
}
int main() {
freopen( "test.in" , "r" , stdin );
freopen( "test.out" , "w" , stdout );
int q;
cin >> n >> q;
init();
Rep( i , n ) {
scanf( "%d" , seq + i );
insert( i , seq[ i ] );
}
mn = mx = seq[ 1 ];
for( int i = 2 ; i <= n ; ++i ) {
mx = max( seq[ i ] , mx );
mn = min( seq[ i ] , mn );
}
char c;
while( q-- ) {
scanf( " %c" , &c );
if( c == 'Q' ) {
int l , r , k;
scanf( "%d%d%d" , &l , &r , &k );
int t = query( l , r , k ) , p = *min_element( seq + l , seq + r + 1 );
for( int i = l ; i <= r ; ++i )
if( seq[ i ] <= t && seq[ i ] > p )
p = seq[ i ];
printf( "%d\n" , p );
} else {
int pos , v;
scanf( "%d%d" , &pos , &v );
mn = min( v , mn );
mx = max( v , mx );
modify( pos , v );
}
}
return 0;
}
---------------------------------------------------------------
1901: Zju2112 Dynamic Rankings
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 5363 Solved: 2243
[Submit][Status][Discuss]
Description
给定一个含有n个数的序列a[1],a[2],a[3]……a[n]，程序必须回答这样的询问：对于给定的i,j,k，在a[i],a[i+1],a[i+2]……a[j]中第k小的数是多少(1≤k≤j-i+1)，并且，你可以改变一些a[i]的值，改变后，程序还能针对改变后的a继续回答上面的问题。你需要编一个这样的程序，从输入文件中读入序列a，然后读入一系列的指令，包括询问指令和修改指令。对于每一个询问指令，你必须输出正确的回答。 第一行有两个正整数n(1≤n≤10000)，m(1≤m≤10000)。分别表示序列的长度和指令的个数。第二行有n个数，表示a[1],a[2]……a[n]，这些数都小于10^9。接下来的m行描述每条指令，每行的格式是下面两种格式中的一种。 Q i j k 或者 C i t Q i j k （i,j,k是数字，1≤i≤j≤n, 1≤k≤j-i+1）表示询问指令，询问a[i]，a[i+1]……a[j]中第k小的数。C i t (1≤i≤n，0≤t≤10^9)表示把a[i]改变成为t。
Input
对于每一次询问，你都需要输出他的答案，每一个输出占单独的一行。
Output
Sample Input
5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3
Sample Output
3
6
6
HINT
20%的数据中，m,n≤100; 40%的数据中，m,n≤1000; 100%的数据中，m,n≤10000。
