CSP 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
子任务:
解题思路:
首先用pair数组stu存取输入的数据,进行排序。
由于子任务中有部分数据m最大达到10的5次方,如果用两重循环会超时,将不能得到满分。
这里用到了前缀和。数组cnt0和cnt1分别记录到第i个位置当中0和1的数量,后面就能够很方便地按照题意求出答案了。
100分代码:
#include <iostream>
#include <cstdio>
#include <set>
#include <algorithm>
using namespace std;
int cnt0[100005],cnt1[100005];
pair<int,int> stu[100005];
set<int> check;
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int m,ans,maxcnt=0;
cin>>m;
for(int i=1;i<=m;i++)
cin>>stu[i].first>>stu[i].second;
sort(stu+1,stu+m+1);
for(int i=1;i<=m;i++)
{
cnt1[i]=cnt1[i-1]+stu[i].second;
cnt0[i]=i-cnt1[i];
}//前缀和,记录到第i个位置当中0和1的数量
for(int i=1;i<=m;i++)
{
int num=stu[i].first;
if(check.count(num)) continue;
check.insert(num);
int temp=cnt0[i-1]+cnt1[m]-cnt1[i-1];
if(temp>=maxcnt)
{
maxcnt=temp;
ans=stu[i].first;
}
}
cout<<ans<<endl;
return 0;
}