具体题目内容链接如下:
CCF&CSP 202012-2 期末预测之最佳阈值
理解:前缀和求解问题,找一个数值,使得前缀和最大。即下图所示样例中的3
#include<iostream>
#include<algorithm>
using namespace std;
pair<int,int> pii[100005]; //pair数组储存信息,每个pair存储一个同学的y和result
int pre0[100005]; //记录该位置前面的result为0的个数(前缀和)
int rear1[100005]; //记录该位置及后面的result为1的个数(后缀和)
int k = -1; //k用来记录最佳阈值
int ma = 0; //ma用来存储最佳阈值对应的预测成功的次数
int main(){
int m;
cin>>m; //输入m
pii[0] = pair<int,int>(-1,-1);
for(int i = 1;i <= m;++i) //初始化pii数组
cin>>pii[i].first>>pii[i].second;
sort(pii + 1,pii + 1 + m); //将所有学生信息按照阈值从小到大排序,方便后续前缀后缀和的操作
for(int i = 1;i <= m;++i) //记录前缀0个数
if(pii[i-1].second == 0)
pre0[i] = pre0[i - 1] + 1; //pre0[0]=1
else
pre0[i] = pre0[i - 1];
for(int i = m;i >= 1;--i) //记录后缀1个数
if(pii[i].second == 1)
rear1[i] = rear1[i + 1] + 1; //rear1[m+1]=0
else
rear1[i] = rear1[i + 1];
for(int i = 1;i <= m;++i){ //最终处理
if(pii[i].first == pii[i - 1].first)
continue; //如果有阈值相同的情况,在出现的第一个位置统计了,直接跳过
if(ma <= pre0[i] + rear1[i])//更新k和ma
ma = pre0[i] + rear1[i],k = pii[i].first;
}
cout<<k;
return 0;
}
参考博客:期末预测之最佳阈值
说明 主要是最开始看了上面这个博客的代码,不过刚开始的时候,我有点不理解。所以修改了计算前缀0部分的代码,原博主是判断当前位置i
是否为0,最终计算:预测成功次数=i-1
位置上的前缀0 + i
位置上的后缀1相加。
这里我改成了,判断当前位置i
的前一个位置,也就是i-1
位置上是否为0,最终计算:预测成功次数=i
位置上的前缀0 + i
位置上的后缀1相加。