UCF Local Programming Contest 2017—— Rotating Cards(树状数组+思维)

题目链接：https://nanti.jisuanke.com/t/44825

此题需要每次去掉top位置的纸牌后实时更新前缀和，于是可以考虑树状数组，树状数组的时间复杂度是log级别。

思路：设刚去掉的纸牌原来的位序是pre，现在的位序是now，剩余纸牌的数字和是n，比如从top到bottom依次为3 5 1 4 2 ；设pre=3，now=5，分别对应的是1和2号纸牌； 接下来要去掉2号牌(1号已经去掉)，需要比较两个区间的和，一个是(pre,now)开区间，设和为x，另一个区间不需要指出，只需要直接求出其和n-x;所以取min(n-x,x)；其中x=sum[now-1]-sum[pre-1];当然上述情况为now>pre的情况，now<pre的情况同理可以推理出。

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#include <cstdio>
#include <string>
#include <stack>
#include <set>
#define IOS ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
typedef long long ll;
ll id[100010],pre;
ll num[100010],c;
ll d[100010];
ll lowbit(ll pos){
    return pos&(-pos);
}
void add(ll pos,ll val){
    while(pos<=c){
        d[pos]+=val;
        pos+=lowbit(pos);
    }
    return ;
}
ll getsum(ll pos){
    ll sum=0;
    while(pos>0){
        sum+=d[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
int main()
{
    IOS;
    ll t;
    cin>>t;
    while(t--){
        memset(id,0,sizeof(id));
        memset(num,0,sizeof(num));
        memset(d,0,sizeof(d));
        cin>>c;
        for(int i=1;i<=c;i++){
            cin>>num[i];
            add(i,num[i]);
            id[num[i]]=i;//建立索引
        }
        ll n=c*(c+1)/2;
        ll cost=0;
        pre=0;
        for(int i=1;i<=c;i++){
            ll p1=getsum(id[i]-1),p2=getsum(pre-1);//先处理两个前缀和
            //接下来分两种情况取最小值
            if(id[i]>pre){
                cost+=min(n-(p1-p2),p1-p2);
                add(id[i],-i);
                n-=i;
                pre=id[i];
            }
            else{
                cost+=min(n-(p2-p1),p2-p1);
                add(id[i],-i);
                n-=i;
                pre=id[i];
            }
        }
        cout<<cost<<endl;
    }
    getchar();
    getchar();
    return 0;
}