题目
洛谷题面链接中文题面
题解
题目的意思就是最多对一个数取反,然后看逆序对最大是多少,我们分析一下,将一个数字取反后,
我们肯定是要让逆序对的数量增加的越多越好.
这样我们就要思考,关于一个数,对于逆序对个数的贡献是怎样的.
如果是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();
}