CSP202012-2 期末预测之最佳阈值

题目

题目可以在CSP的官网查看到哟！

算法思想

这道题要做的就是寻找最佳的阈值，使得预测正确的数量最多。看过最后的数据范围，发现2<=m<=100000，这道题很容易想到O(n * n)的算法，但是肯定会超时，因此需要一个时间复杂度较低的算法。接下来，小编介绍的算法时间复杂度是O(n*logn)。
首先对阈值进行排序，按照阈值从小到大进行排序，可以很容易地想到，在选择的阈值之前的预测为0的是预测正确的，在阈值之后的预测为1的是预测正确的，因此提前生成两个数组，一个数组记录该点 之前预测为0的数量，另一个数组记录该点之后（包括该点）预测为1的数量，而对于该点，两个数组相加就是以该点数值为阈值的预测正确的数量。
还有一个问题，就是如果存在多个相同的安全指数，该怎么办呢，小编手动模拟了出现相同多个安全指数的情况，发现：相同的安全指数，预测为1的排在前面，预测为0的排在后面，就可以得到正确的答案。因此在利用sort快速排序的时候，先按照安全指数从小到大排序，再按照预测值从1到0排序即可。
排序之后，利用前缀和和后缀和的思想，分别统计某一点之前预测为0的数量以及某一点之后（包括该点）预测为1的数量，之后寻找两者相加最大值所对应的最大阈值即可。

完整代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#pragma warning (disable:4996)
using namespace std;
int m;
struct Person
{
	int y, result;
}person[200000];   
int sum0[200000], sum1[200000];   
bool cmp(Person a, Person b)
{
	if (a.y != b.y) return a.y < b.y;
	else return a.result > b.result;
}
int main()
{
	freopen("in.in", "r", stdin);
	freopen("out.out", "w", stdout);
	int i, j;
	cin >> m;
	for (i = 1; i <= m; i++)
		cin >> person[i].y >> person[i].result;
	sort(person + 1, person + m + 1, cmp);
	sum0[0] = 0;
	sum0[1] = 0;
	for (i = 2; i <= m; i++)
	{
		if (person[i - 1].result == 0)
			sum0[i] = sum0[i - 1] + 1;
		else sum0[i] = sum0[i - 1];
	}
	sum1[m + 1] = 0;
	for (i = m; i >= 1; i--)
	{
		if (person[i].result == 1)
			sum1[i] = sum1[i + 1] + 1;
		else sum1[i] = sum1[i + 1];
	}
	int max_y = 0, max_num = 0;
	for (i = 1; i <= m; i++)
	{
		int num = sum0[i] + sum1[i];
		if (num >= max_num)
		{
			max_y = person[i].y;
			max_num = num;
		}
	}
	cout << max_y << endl;
	return 0;
}