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=;
 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=;
     while(x)
         ans+=c[x],x-=lowbit(x);
     return ans;
 }

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

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

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

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

             else if(pos[i]<now)
                 ans+=min(query(pos[i],now),query(now,n)+query(,pos[i]));
             else
                 ans++;

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

     }
     return ;
 }