Codeforces Round #835 (Div. 4) - E. Binary Inversions

E. Binary Inversions

贪心

        找到数组最左边的0的位置l与最右边的1的位置r。

        对数组的l位置取反，求满足条件的逆序对个数。

        对数组的r位置取反，求满足条件的逆序对个数。

        不执行取反操作，求满足条件的逆序对个数。

        对以上三者取最大值即为答案。

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for(int i = a;i<n;i++)
#define per(i,a,n) for(int i = n-1;i>=a;i--)
#define pb push_back
#define mp make_pair
#define eb emplace_back
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
#define yes cout<<"YES"<<'\n';
#define no cout<<"NO"<<'\n';
#define endl '\n';
typedef vector<int> VI;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef double db;
mt19937 mrand(random_device{}());
const ll mod=1000000007;
int rnd(int x) {return mrand() % x;}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;};
ll lcm(int a,int b){return a*b/gcd(a,b);}

const int N=2e5+10;
int T;
int n;
int a[N];

ll check(int pos){
	ll res=0,c=0;
	if(pos>=0) a[pos]^=1;
	rep(i,0,n){
		if(a[i]&1) c++;
		else res+=c;
	}
	if(pos>=0) a[pos]^=1;
	return res;
}

void solve(){
	cin>>n;
	int l=-1,r=-1,x=0,y=0;
	rep(i,0,n){
		cin>>a[i];
		if(a[i]&1) r=i;
		if(!a[i]&&l==-1) l=i;
	}
	ll ans=0;
	if(l!=-1) ans=max(ans,check(l));
	if(r!=-1) ans=max(ans,check(r));
	ans=max(ans,check(-1));
	cout<<ans<<endl;
}

int main(){
	ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>T;
	while(T--){
		solve();
	}
    return 0;
}