期末预测之最佳阈值
题目背景
题目描述
输入格式
输出格式
样例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容器中,方便之后访问。之后使用循环得出前缀数组和后缀数组,前缀数组代表当前的阈值预测正确的挂科个数,后缀数组代表的是当前的阈值预测正确的未挂科个数,将二者相加就是当前阈值对应的预测正确的个数,之后再循环访问每个阈值进行比较得出最大的阈值即可。