CF1760E

题目

洛谷题面链接中文题面

题解

题目的意思就是最多对一个数取反,然后看逆序对最大是多少,我们分析一下,将一个数字取反后,

我们肯定是要让逆序对的数量增加的越多越好.

这样我们就要思考,关于一个数,对于逆序对个数的贡献是怎样的.

如果是1那么总对数就是在它右边的0的个数;

如果是0那么总对数就是在它左边的1的个数.

所以说如果我们把一个位置上的1变成了0,那么我们就会失去这个位置上1的贡献,得到0的贡献.

如果把一个位置上的0变成了1,就反过来.

所以我们只要用前缀和的思想来记录0,1个数,然后遍历,看那个位置上可以增加的越多,就选择那个位置取反.

代码

#include<bits/stdc++.h>
using namespace std;
using LL = long long;
const int N = 1e6+9;
int n,a[N],pre[N];
void solve(){
    cin >> n;
    for(int i=1;i<=n;i++) {cin >>a[i];pre[i]=pre[i-1]+a[i];}
    int idx=0,jia=0;//idx是取反位置 jia 最大增加量
    for(int i=1;i<=n;i++){
        if(a[i]){ //如果是1的话,将这个位置变为0 ,就会失去这个位置作为1的贡献,得到作为0的贡献
            if(pre[i-1]-((n-i)-pre[n]+pre[i])>jia){
                jia=pre[i-1]-((n-i)-pre[n]+pre[i]);
                 idx=i;
            }  
        }
        else {//如果是0,跟上面相反
            if(((n-i)-pre[n]+pre[i])-pre[i-1]>jia){
                jia=((n-i)-pre[n]+pre[i])-pre[i-1];
                idx=i;
            }
        }
    }
    a[idx]^=1; //取反
    LL ans=0; 
    //cout << idx << '\n';
    for(int i=1;i<=n;i++) pre[i]=pre[i-1]+a[i]; //再次前缀和
    for(int i=1;i<n;i++) if(a[i])ans+=(n-i)-pre[n]+pre[i]; //计算对数
    cout << ans << '\n';
}
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int _ =1;
    cin >> _;
    while(_--)solve();
}