USACO 2019 US Open Contest, Gold Problem 3. Balancing Inversions —— 模拟

This way

题意：

给你两个长度为n的01序列，每个序列的值是其中逆序对的个数，每一次可以交换相邻的两个位置的值，并且第一个序列的尾与第二个序列的头是连在一起的，但是计算答案的时候分开计算，问你最少要交换几次可以让这两个序列的值相同。

题解：

有两种可能：两个序列内部交换，两个序列相互之间会交换。
那么我们知道每次交换一个10对的时候答案-1,01对答案+1，那么我们枚举序列间的交换次数，有两种可能：第一个序列的0与第二个序列的1交换，或者相反。其它交换情况没有意义。
如果我们交换10，那么必然是第一个数的末尾1与第二个数的首位0交换，那么交换这两个数的步数是n-pa+pb，也就是左边交换n-pa次，右边交换pb-1次，最后两个值交换一次。
那么10交换与01交换也是不相同的。
10交换的时候左边的值-（最后一个1右边0的个数）+（最后一个1左边1的个数），右边的值-（第一个0左边1的个数）+（第一个0右边0的个数）
01交换的时候左边的值-（最后一个0左边1的个数），右边的值-（第一个1右边0的个数）

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+5;
int a[N],b[N],rigb0[N],lefa1[N];
int main()
{
    int n;
    ll sa=0,sb=0,numa=0,numb=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]==0)
            sa+=numa;
        else
            numa++;
        lefa1[i]=lefa1[i-1]+a[i];
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&b[i]);
        if(b[i]==0)
            sb+=numb;
        else
            numb++;
    }
    for(int i=n;i>=1;i--)
        rigb0[i]=rigb0[i+1]+(b[i]==0);
    int pa=n,pb=1;
    if(sa==sb)
        return 0*printf("0\n");
    ll mi=sa-sb>=0?sa-sb:sb-sa;
    ll ssa=sa,ssb=sb,step=0;
    for(int i=1;;i++)
    {
        while(pa>=1&&a[pa]!=0)
            pa--;
        if(pa<=0)
            break;
        while(pb<=n&&b[pb]!=1)
            pb++;
        if(pb>n)
            break;
        step+=n-pa+pb;
        ssa-=lefa1[pa],ssb-=rigb0[pb];
        mi=min(mi,(ssa-ssb>=0?ssa-ssb:ssb-ssa)+step);
        pa--,pb++;
    }
    pa=n,pb=1,step=0;
    for(int i=1;;i++)
    {
        while(pa>=1&&a[pa]!=1)
            pa--;
        if(pa<=0)
            break;
        while(pb<=n&&b[pb]!=0)
            pb++;
        if(pb>n)
            break;
        step+=n-pa+pb;
        sa=sa-(n-pa)+lefa1[pa-1],sb=sb-(pb-1)+rigb0[pb+1];
        mi=min(mi,(sa-sb>=0?sa-sb:sb-sa)+step);
        pa--,pb++;
    }
    printf("%lld\n",mi);
    return 0;
}
/*
5
1 0 1 0 0 1 0 1 0 1

6
1 1 1 0 0 0 0 0 0 1 1 1
*/