Codeforces Round #609 (Div. 2) E K Integers(树状数组+逆序对)

题目链接：https://codeforces.com/contest/1269/problem/E

 

题目大意：

  给一个n个元素的数组，每次可以交换相邻两个数字，问形成1~k顺序的子串需要的最少次数，k取1~n

 

题目思路：

  qsc题解视频：传送门
  分成两个部分，一个是计算逆序对，一个是计算将需要的数字合并到一起的最小次数。
  可以知道。只要并在了一起，那想要顺序的最小次数就是逆序对数量。逆序对的计算就是对于每个数字来说，后面有多少个比它小的。因为从小到大插入，所以前面的数字贡献就不用算了，因为后面不可能有比它小的插入了。然后就每次查询i这个位置的树状数组结果，得到的是这个位置前有多少个小于等于当前数的数字，i-这个数字就是后面有多少个比它小的数，这就是逆序对结果要加的数字。
  那么如何让他们并到一起呢？就是把无关人等赶出去，然后大家都往最中间的人靠，一定能让移动次数最少。所以先用二分确定最中间的位置在哪儿，然后计算需要移动的数量。最终我们希望的是以二分出来的地方为中间的一个1~k顺序子串。所以先用查逆序对的那个query，得到二分出来的地方以及之前一共有多少个数字，设为x，就表示pos，pos-1,pos-2…pos-k+1是1~k的前半部分，用等差数列求和公式能得到这个，然后第二棵树状数组记录了前面的数字的位置和，用刚才的等差公式得到的和减去位置和，就是想形成前半部分需要的最小次数，后半部分同理。

 

以下是代码：

#include<bits/stdc++.h>

using namespace std;
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(ll i=a;i>=b;i--)
#define ll long long
const int MAXN = 2e5+5;
ll sum1[MAXN],sum2[MAXN],a[MAXN],pos[MAXN];
void add1(ll x,ll y){
    if(!x)return;
    for(;x<MAXN;x+=x&-x){
        sum1[x]+=y;
    }
}
ll query1(ll x){
    ll ans=0;
    for(;x;x-=x&-x){
        ans+=sum1[x];
    }
    return ans;
}
void add2(ll x,ll y){
    if(!x)return;
    for(;x<MAXN;x+=x&-x){
        sum2[x]+=y;
    }
}
ll query2(ll x){
    ll ans=0;
    for(;x;x-=x&-x){
        ans+=sum2[x];
    }
    return ans;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    while(cin>>n){
        memset(sum1,0,sizeof(sum1));
        memset(sum2,0,sizeof(sum2));
        ll ans=0;
        rep(i,1,n)cin>>a[i],pos[a[i]]=i;
        rep(i,1,n){
            int p=pos[i];
            add1(p,1);
            add2(p,p);
            ans+=i-query1(p);
            ll pp=ans;
            ll l=1,r=n,temp;
            while(l<=r){
                int mid=(l+r)>>1;
                if(query1(mid)*2>i){
                    temp=mid;
                    r=mid-1;
                }
                else l=mid+1;
            }
            ll x=query1(temp);
            pp+=(temp-x+1+temp)*x/2-query2(temp);
            x=i-x;
            pp+=query2(n)-query2(temp)-((temp+1+temp+x)*x/2);
            cout<<pp<<" ";
        }cout<<endl;
    }
    return 0;
}