贪心:本来越靠前的数,就应该越靠前,如果不这样的话,就会一定会导致不是最优解。
分析:由于题目1s的时间限制,与1e9的计算次数,所以我们只能通过O(n)级别的复杂度进行求解,那么如果需要我们瞬间定位一个奇数或者是偶数是很难实现的。
做法:
首先:如果奇数和偶数的数量相差大于1的时候,是一定无解的,因为不可能实现奇偶相间。
可以记录下来所有的奇数和偶数的下标,然后我们这样思考:
我们可以让记录下来的所有数的下标定位到其应在的位置,比如现在我们想让偶数开头,那么结果就是偶数的下标按顺序对应到1,3,5,7…等的位置的距离相加。
- 在n为偶数的时候,第一个数字可以为奇数或者是偶数,那么我们可以求出两种情况下的交换次数,并且取最小值。
- 在n为奇数的时候,这时候一定会有奇数的数量大于偶数或者是偶数的数量大于奇数,则我们只需要让数量多的那一个放到1,3,5,7…的位置上就可以了。
注意结果的计算不要算重了,计算完偶数对应位置之后就不用再计算奇数对应位置,因为所有偶数都排好了那么奇数就一定都排好了。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int odd[N];
int even[N];
void solve(){
int n;cin >> n;
int cntodd = 0,cnteven = 0;
for(int i = 1;i <= n;i++){
int a;cin >> a;
if(a & 1){
odd[++cntodd] = i;
}
else {
even[++cnteven] = i;
}
}
if(abs(cntodd - cnteven) > 1){
cout << "-1\n";
return;
}
if(n & 1){
int res = 0;
if(cntodd > cnteven){
for(int i = 1;i <= cntodd;i++){
res += abs(odd[i] - (i * 2 - 1));
}
cout << res << endl;
}else{
for(int i = 1;i <= cnteven;i++){
res += abs(even[i] - (i * 2 - 1));
}
cout << res << endl;
}
}else{
int res1 = 0;//odd first
for(int i = 1;i <= cntodd;i++){
res1 += abs(odd[i] - (i * 2 - 1));
}
int res2 = 0;
for(int i = 1;i <= cnteven;i++){
res2 += abs(even[i] - (i * 2 - 1));
}
cout << min(res1,res2) << endl;
}
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
int T;cin >> T;
while(T--){
solve();
}
return 0;
}