codeforces 题目 Binary Inversions

目录

题目：

题目描述：

思路：

AC代码：

---

题目：

题目描述：

给你长度为 n 的 01数组，最多可以进行一次操作，将其中一位进行反转（ 0 变成 1 ， 1 变成 0 ）问最终数组最多可以拥有多少个逆序对（i < j ,有 ai > aj ）

思路：

我们可以发现，总逆序对数，就是对数组中每个 1 后面的 0 的个数的加和。

题目所说可以最多进行一次操作，所以我们首先可以分成两种类型来考虑。

如果我们没进行操作：

总逆序对数，就是对数组中每个 1 后面的 0 的个数的加和。

而当我们进行了一次操作，反转了一位之后，这个总和会发生什么变化呢？我们可以进行如下分类讨论：

①其中一个 1 改 0 

总和减少：这个位置后面 0 的个数

总和增加：这个位置前面 1 的个数

所以为了让总和增加的最多，我们让最后一个出现的 1 改成 0 就能获得最大收益

②其中一个 0 改 1 

总和减少：这个位置前面 1 的个数

总和增加：这个位置后面 0 的个数

所以为了让总和增加的最多，我们让第一个出现的 0 改成 1 就能获得最大收益

具体操作，我们可以用前缀和的思想：

先开个 pre 数组（前缀：prefix）用来存数组中每个位置前面的 1 的个数。

再开一个 post 数组（后缀：postfix）存放每个位置后面的 0 的个数

然后对以上三种情况分别求值然后取最大值

思路有了，具体操作请看AC代码：

AC代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

const int N = 2e5 + 5;

int a[N];
int post[N];//后面0个数的后缀和
int pre[N];//前面1个数的前缀和

int main()
{
	std::ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	int t;
	cin >> t;

	while (t--)
	{
		memset(post, 0, sizeof(post));
		memset(pre, 0, sizeof(pre));

		int n;
		cin >> n;

		for (int i = 1; i <= n; i++)
			cin >> a[i];

		for (int i = n; i >= 1; i--)
		    post[i] = post[i + 1] + (a[i] == 0 ? 1 : 0);

		for (int i = 1; i <= n; i++)
			pre[i] = pre[i - 1] + (a[i] == 1 ? 1 : 0);

		ll sum1 = 0;//不变
		ll sum2 = 0;//第一个0改成1
		ll sum3 = 0;//最后一个1改成0

		int first0 = 0;//防止没有0情况，所以初始设为0
		int last1 = n + 1;//防止没有1情况，所以初始设为n+1

		for (int i = 1; i <= n; i++)
		{
			if (!first0 && a[i] == 0)first0 = i;

			if (a[i] == 1)
			{
				last1 = i;

				sum1 += post[i + 1];
			}
		}
		sum2 = sum1 - pre[first0 - 1] + post[first0 + 1];
		sum3 = sum1 + pre[last1 - 1] - post[last1 + 1];

		cout << max(sum1, max(sum2, sum3)) << '\n';
	}
	return 0;
}