Codeforces Round #609 (Div. 2) E. K Integers

K Integers
题意:给n个数的一个全排列,每次交换相邻两个数
求使 数字1-k相邻且递增的最小步骤,k(1,n);
思路:
当1-k已经相邻时,ans即为1-k的逆序数。
那么只需逐步算出使1-k这些数相邻所需的最小步骤
容易看出 使这k个数 向中间靠拢时花费步数最小
并容易推出如下公式

本题难点在于如何计算向中间靠拢时的步数 自己一直在考虑怎样正向算出答案, 并且没有想到利用优先队列来维护这个步数,而是推出来另一个看似可解的办法导致浪费很多时间,最后也没有完成

逆向思维 容斥 两个优先队列维护左右边

#include<bits/stdc++.h>
using namespace std;
typedef long long L;
const int maxn=2e5+5;
const int oo=1e9+7;
int a[maxn],c[maxn],n;
int pos[maxn];
L ans[maxn];
void update(int x){for(int i=x;i<=n&&i;i+=i&(-i)) c[i]++;}
L    query (int x){L ans=0;while(x)ans=ans+1ll*c[x],x-=x&(-x);return ans;}
priority_queue<int>q1,q2;

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),pos[a[i]]=i;
    //ans[1]=0;ans[2]=abs(pos[2]-pos[1])-1+(pos[2]>pos[1]?0:1);
    //update(pos[1]);update(pos[2]);

    L a1=0,ln=0,rn=0;
    for(L i=1;i<=n;i++){
        update(pos[i]);
        a1=a1+i-query(pos[i]);

        if(q1.size()&&pos[i]<q1.top()){
            ln+=pos[i];q1.push(pos[i]);
        } else {
            rn+=pos[i];q2.push(-pos[i]);
        }

        while(q1.size()<q2.size()){
            L c=-q2.top();

            ln+=c;rn-=c;

            q2.pop();q1.push(c);
        }

        while(q1.size()>q2.size()+1){

            L c=q1.top();
            ln-=c;rn+=c;

            q1.pop();q2.push(-c);
        }

        L top=q1.top();
        ans[i]=a1+1ll*top*q1.size()-ln-1ll*q1.size()*(q1.size()-1)/2
                 +rn-top*q2.size()-1ll*q2.size()*(q2.size()+1)/2;
        ;
    }
    for(int i=1;i<=n;i++){
        if(i!=1) printf(" ");
        printf("%lld",ans[i]);
    }
}
/*
10
5 1 6 2 8 3 4 10 9 7
*/

发布了61 篇原创文章 · 获赞 7 · 访问量 3196

猜你喜欢

转载自blog.csdn.net/qq_43914084/article/details/103663710