HDU 4262 Juggler 树状数组

最新推荐文章于 2018-06-25 08:49:41 发布

aegt43853

最新推荐文章于 2018-06-25 08:49:41 发布

阅读量80

点赞数

原文链接：http://www.cnblogs.com/GBRgbr/p/3297745.html

版权

将每个球按输入顺序编号，建立它第几个被扔掉->编号的映射关系。

记录当前在手里的球的编号，按扔掉的顺序查找这个球的编号，看看这个球是逆时针转到手里更近还是顺时针转到手里更近，即当前扔掉球的编号与当前手里球的编号之间有几个球。

树状数组C[i]记录编号i的球是否还在。

球是环形排列的，特殊处理一下。

对于扔掉一个球之后下一个落在手里的球的编号，二分判定，找顺时针方向第一个有球的位置

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>

#define LL long long int

using namespace std;

const int MAXN = 100100;

int C[MAXN];
int N;
int PosToID[MAXN];
int IDToPos[MAXN];

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

int query( int x )
{
    if ( !x ) return 0;
    int res = 0;
    while ( x > 0 )
    {
        res += C[x];
        x -= lowbit(x);
    }
    return res;
}

void update( int x, int val )
{
    while ( x <= N )
    {
        C[x] += val;
        x += lowbit(x);
    }
    return;
}

int BiSearch( int l, int r )
{
    int pre = l;
    if ( query(r) - query(l) == 0 )
    {
        r = l;
        l = 1;
        pre = 0;
    }

    int ans = 0;
    while ( l <= r )
    {
        int mid = ( l + r ) >> 1;
        if ( query(mid) - query(pre) > 0 )
        {
            ans = mid;
            r = mid - 1;
        }
        else l = mid + 1;
    }
    return ans;
}

int main()
{
    while ( scanf( "%d", &N ), N )
    {
        memset( C, 0, sizeof(C) );
        for ( int i = 1; i <= N; ++i )
        {
            scanf("%d", &IDToPos[i] );
            PosToID[ IDToPos[i] ] = i;
            update( i, 1 );
        }

        int cur = 1;   //当前手里的
        LL ans = 0;
        for ( int i = 1; i <= N; ++i )
        {
            int id = PosToID[i];

            if ( id < cur )
            {
                int tmp = query(cur) - query(id);
                ans += min( tmp, query(N) - query(cur) + query(id) );
            }
            else if ( id > cur )
            {
                int tmp = query(id) - query(cur);
                ans += min( tmp, query(N) - query(id) + query(cur) );
            }
            update( id, -1 );
            ++ans;
            cur = BiSearch( id, N );
        }
        printf( "%I64d\n", ans );
    }
    return 0;
}