HDU 4262 Juggler

点我看题

初步想法是模拟，找到下一个位置并记录操作数，O(n^2)肯定会超时。

那么进行优化，会发现到下一位置的操作数就是两个位置之间存在的数的个数，于是就变成了计数问题。

不难想到用树状数组或线段树进行计数，时间复杂度O(nlogn)。

#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<string.h>
using namespace std;
const int Nmax=100005;
int n,now;
long long ans;
int num[Nmax],pos[Nmax];
int     c[Nmax];

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

int get_sum(int x)
{
    int ans=0;
    while(x)
        ans+=c[x],x-=lowbit(x);
    return ans;
}

int query(int l,int r)
{
    return get_sum(r)-get_sum(l-1);
}

void update(int p,int add)
{
    while(p<=n)
        c[p]+=add,p+=lowbit(p);
}

void build()
{
    for(int i=0;i<=n;i++)
        c[i]=0;
    for(int i=1;i<=n;i++)
        update(i,1);
}

int main()
{
    freopen("in.in","r",stdin);
    while(scanf("%d",&n) && n!=0)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
            pos[num[i]]=i;
        }
        build();
        now=1,ans=0;
        for(int i=1;i<=n;i++)
        {
            if(pos[i]>now)
                ans+=min(query(now,pos[i]),query(1,now)+query(pos[i],n));
            
            else if(pos[i]<now) 
                ans+=min(query(pos[i],now),query(now,n)+query(1,pos[i]));
            else
                ans++;

            now=pos[i],update(pos[i],-1),num[pos[i]]=0;
            if(i==n)
                break;
            while(!num[now]) 
                now=now%n+1;
        }
        printf("%lld
",ans);

    }
    return 0;
}