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

题目链接：题目链接

问题分析

通过分析题目我们可以发现问题可以转化为在y_i中寻找一个θ，使得满足y_i<θ时result_i为0的个数+满足y_i≥θ时result_i为1的个数最大。如果有多个θ，则选取最大的θ输出即可。

AC代码

#include<bits/stdc++.h>
using namespace std;
struct p
{
	int y;
	int result;
};
bool cmp(p& p1, p& p2)
{
	if(p1.y != p2.y)
		return p1.y < p2.y;
	else
		return p1.result < p2.result;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int dp0[100006] = {0},dp1[100005] = {0},count=0,m,max=-1,index=0;
	p input[100005];
	cin>>m;
	for(int i=0;i<m;i++)
	{
		cin>>input[i].y>>input[i].result;
	}
	sort(input,input+m,cmp);
	if(input[m-1].result == 1)
		dp1[m-1] = 1;
	if(input[0].result == 0)
		count = 1;
	for(int i=1;i<m;i++)
	{
		if(input[i].y != input[i-1].y)
		{
			dp0[i] = dp0[i-1] + count;
			count = 0;
		}
		else
			dp0[i] = dp0[i-1];
		if(input[i].result == 0)
			count++;
		dp1[m-i-1] = dp1[m-i] + (input[m-i-1].result==1?1:0);
	}
	for(int i=0;i<m;i++)
	{
		if(max<=dp0[i] + dp1[i])
		{
			max = dp0[i]+dp1[i];
			index = i;
		}
	}
	cout<<input[index].y;
	return 0;
}