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;
}

猜你喜欢

转载自blog.csdn.net/toohandsomeIeaseId/article/details/104138248