BZOJ4415: [Shoi2013]发牌 树状数组+二分

Description

 

假设一开始，荷官拿出了一副新牌，这副牌有N张不同的牌，编号依次为1到N。由于是新牌，所以牌是按照顺序排好的，从牌库顶开始，依次为1, 2,……直到N，N号牌在牌库底。为了发完所有的牌，荷官会进行N次发牌操作，在第i次发牌之前，他会连续进行R_i次销牌操作，R_i由输入给定。请问最后玩家拿到这副牌的顺序是什么样的？

举个例子，假设N = 4，则一开始的时候，牌库中牌的构成顺序为{1, 2, 3, 4}。

假设R1=2，则荷官应该连销两次牌，将1和2放入牌库底，再将3发给玩家。目前牌库中的牌顺序为{4, 1, 2}。

假设R2=0，荷官不需要销牌，直接将4发给玩家，目前牌库中的牌顺序为{1,2}。

假设R3=3，则荷官依次销去了1, 2, 1，再将2发给了玩家。目前牌库仅剩下一张牌1。

假设R4=2，荷官在重复销去两次1之后，还是将1发给了玩家，这是因为1是牌库中唯一的一张牌。

 

Input

第1行，一个整数N，表示牌的数量。第2行到第N + 1行，在第i + 1行，有一个整数R_i， 0≤R_i<N

Output

第1行到第N行：第i行只有一个整数，表示玩家收到的第i张牌的编号。

Sample Input

4
2
0
3
2

Sample Output

3
4
2
1

HINT

N<=70万

Solution

提供一个跑的贼慢的做法

在树状数组上二分

复杂度$O(nlognlogn)$

和权值线段树差不多的思路，但是好写啊

#include <bits/stdc++.h>

using namespace std ;

inline int lowbit( int x ) {
    return x & -x ;
} 

#define N 700010

int n ;
int c[ N ] ;

void add( int x , int val ) {
    for( int i = x ; i <= n ; i += lowbit( i ) ) 
        c[ i ] += val ;
}

int query( int x ) {
    int ans = 0 ;
    for( int i = x ; i ; i -= lowbit( i ) ) ans += c[ i ] ;
    return ans ;
}

int find( int x ) {
    int l = 1 , len = n ;
    while( len ) {
        int mid = l +( len >> 1 ) ;
        if( query( mid ) < x ) {
            l = mid + 1 ;
            len = len - ( len >> 1 | 1 ) ;
        } else len = len >> 1 ;
    }
    return l ;
}

int main() {
    int now = 0 ;
    scanf( "%d" , &n ) ;
    for( int i = 1 ; i <= n ; i ++ ) add( i , 1 ) ;
    for( int i = 1 ; i <= n ; i ++ ) {
        int t ;
        scanf( "%d" , &t ) ;
        t = t % ( n - i + 1 ) ;
        now = ( now + t ) % ( n - i + 1 ) ;
        t = find( now + 1 ) ;
        printf( "%d
" , t ) ;
        add( t , -1 ) ;
    }
    return 0 ;
}