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

期末预测之最佳阈值

题目背景

题目描述

输入格式

输出格式

样例1输入

6
0 0
1 0
1 1
3 1
5 1
7 1

样例1输出

3

样例1解释

样例2输入

8
5 1
5 0
5 0
2 1
3 0
4 0
100000000 1
1 0

样例2输出

100000000

子任务

代码

#include <iostream>
#include<vector>
#include<map>
using namespace std;
int n;
map<int, pair<int,int>>m;//保存阈值与该阈值挂科的情况
int main()
{
	cin >> n;
	int w, s,num;
	num = 0;
	int zero=0;
	for (int i = 0; i < n; i++)
	{
		cin >> w >> s;
		if (s == 0) m[w].second++;//该阈值挂科次数加一
		else m[w].first++;//该阈值未挂科次数加一
	}
	int sum = 0;
	vector<pair<int,pair<int, int>>>ve(m.begin(),m.end());//用vector容器存储map中的数据，方便访问对应元素
	vector<int>vv;//vv是挂科的前缀数组
	vv.push_back(0);//第一个位置的前缀的挂科个数为0
	for (int i = 1; i < ve.size(); i++)
	{
		vv.push_back(ve[i-1].second.second+vv[i-1]);//计算对应位置的前缀数组
	}
	for (int i = ve.size()-2; i >= 0; i--)
	{
		ve[i].second.first += ve[i + 1].second.first;//计算表示未挂科的后缀数组的值
	}
	for (int i = ve.size() - 1; i >= 0; i--)
	{
		if (ve[i].second.first + vv[i] > sum)//计算当前前缀数组与后缀数组的值，即为该位置阈值对应的正确预测个数
		{
			sum = ve[i].second.first + vv[i];
			num = ve[i].first;
		}
	}
	cout << num << endl;
	return 0;
}

思路

题目中的给出的阈值所对应的挂科和未挂科的个数可能不止一个，所以在存储读取进来的数据的时候采用map是一个不错的方法，这样不会产生多余的空间浪费，同时也可以使得后面遍历的时候遍历的每个位置都有意义。map中的数据分为两部分，一部分是int数据一部分是pair数据，int存储的是阈值，pair存储的是阈值对应的未挂科个数和挂科个数。将map的数据存储到vector容器中，方便之后访问。之后使用循环得出前缀数组和后缀数组，前缀数组代表当前的阈值预测正确的挂科个数，后缀数组代表的是当前的阈值预测正确的未挂科个数，将二者相加就是当前阈值对应的预测正确的个数，之后再循环访问每个阈值进行比较得出最大的阈值即可。