题目链接: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;
}