代码思路:
本题若是直接暴力求解只能得70分!
- 读入学生成绩的数量 m 和每个学生的分数和是否及格(0 表示不及格,1 表示及格)。
- 将输入数据保存到 vector<pair<int, int>> 类型的数组 a 中,并按照第一个元素——阈值从小到大进行排序。
- 计算前缀和数组 prefix,prefix[i] 表示在当前阈值之前不及格的人数。可以通过遍历排序后的数组 a 来获得。如果当前遍历到的学生分数对应的是否及格为 0,则说明该学生在当前阈值下不及格,此时只需将 prefix 数组中的前缀和增加 1 即可。
- 计算后缀和数组 suffix,suffix[i] 表示在当前阈值之后及格的人数。同样可以通过遍历排序后的数组 a 来获得。如果当前遍历到的学生分数对应的是否及格为 1,则说明该学生在当前阈值下及格,此时只需将 suffix 数组中的后缀和增加 1 即可。
- 计算下标映射数组 p,p[i] 表示排完序后第 i 个学生对应的原始序列中的下标。如果排序后的数组 a 中第 i 个学生和第 i-1 个学生的分数相同,则将 p[i] 映射到 p[i-1] 所对应的下标;否则,更新 pos 并将 p[i] 映射到当前位置。这样就可以避免重复计算相同分数所对应的不及格人数和及格人数。
- 遍历排序后的数组 a,在每个阈值处计算在该阈值之前不及格的人数和在该阈值之后及格的人数,并记录在 cur 变量中。如果 cur 大于等于前面出现过的最大值 mx,则更新 mx 和最佳分数阈值 ans。
- 输出最佳的阈值 ans 即可。
代码实现:
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int M = 100000;
pair<int, int> a[M + 1]; // 用 pair 来存储学生成绩和是否及格
int prefix[M + 2], suffix[M + 2], p[M + 2]; // 前缀和数组、后缀和数组、下标映射数组
int main()
{
int m, i;
cin >> m;
for (i = 1; i <= m; i++)
cin>>a[i].first>>a[i].second; // 读入学生成绩和是否及格
sort(a + 1, a + 1 + m); // 按照阈值从小到大排序
// 计算前缀和(不及格人数)
prefix[0] = 0;
for (i = 1; i <= m; i++)
prefix[i] = prefix[i - 1] + (a[i].second == 0 ? 1 : 0);
// 计算后缀和(及格人数)
suffix[m + 1] = 0;
for (i = m; i >= 1; i--)
suffix[i] = suffix[i + 1] + (a[i].second == 1 ? 1 : 0);
int pos = 1; // 当前处理的阈值对应的下标
p[1] = 1; // 排完序后第1个学生(即阈值最小)对应的原下标为1
for (i = 1; i <= m; i++)
if (a[i].first == a[i - 1].first) // 如果当前学生和前一个学生的分数一样
p[i] = pos; // 将当前学生映射到上一个学生对应的下标
else // 如果当前学生和前一个学生的分数不一样
p[i] = (pos = i); // 更新 pos 并将当前学生映射到当前位置
int ans = 0, mx = 0; // 最佳阈值和对应的及格人数
for (i = 1; i <= m; i++) { // 遍历所有阈值
int cur = prefix[p[i] - 1] + suffix[i]; // 计算在当前阈值下的及格人数
if (cur >= mx) // 如果当前及格人数大于等于前面出现过的最大值
mx = cur, ans = a[i].first; // 更新最佳阈值和对应的及格人数
}
cout<< ans; // 输出最佳阈值
return 0;
}