【洛谷 P8598】[蓝桥杯 2013 省 AB] 错误票据 题解（模拟+优先队列）

[蓝桥杯 2013 省 AB] 错误票据

题目背景

某涉密单位下发了某种票据，并要在年终全部收回。

题目描述

每张票据有唯一的 ID 号，全年所有票据的 ID 号是连续的，但 ID 的开始数码是随机选定的。因为工作人员疏忽，在录入 ID 号的时候发生了一处错误，造成了某个 ID 断号，另外一个 ID 重号。

你的任务是通过编程，找出断号的 ID 和重号的 ID。

数据保证断号不可能发生在最大和最小号。

输入格式

一个整数 $N(N<100)$ 表示后面数据行数，接着读入 $N$ 行数据，每行数据长度不等，是用空格分开的若干个（不大于 $100$ 个）正整数（不大于 $10^5$），每个整数代表一个 ID 号。

输出格式

要求程序首先输入要求程序输出 $1$ 行，含两个整数 $m$，$n$，用空格分隔，其中，$m$ 表示断号 ID，$n$ 表示重号 ID。

样例 #1

样例输入 #1

2
5 6 8 11 9
10 12 9

样例输出 #1

7 9

样例 #2

样例输入 #2

6
164 178 108 109 180 155 141 159 104 182 179 118 137 184 115 124 125 129 168 196
172 189 127 107 112 192 103 131 133 169 158
128 102 110 148 139 157 140 195 197
185 152 135 106 123 173 122 136 174 191 145 116 151 143 175 120 161 134 162 190
149 138 142 146 199 126 165 156 153 193 144 166 170 121 171 132 101 194 187 188
113 130 176 154 177 120 117 150 114 183 186 181 100 163 160 167 147 198 111 119

样例输出 #2

105 120

---

思路

首先，将每个测试用例的 ID 号存入优先队列（堆）。这里的优先队列是一个最大堆，也就是说，堆顶元素总是最大的。

接下来，将堆顶元素赋值给 prev，并将其从堆中移除。然后，只要堆中还有元素，就持续进行以下操作：取出堆顶元素，并将其从堆中移除，然后判断这个元素是否等于 prev。如果等于，说明找到了重号的 ID，将其赋值给 b。如果 prev 和这个元素的差为2，说明找到了断号的 ID，将其赋值给 a。每次循环结束时，都将当前元素赋值给 prev，为下一次循环做准备。

最后，打印出断号的 ID（a）和重号的 ID（b）。

---

AC代码

#include <algorithm>
#include <iostream>
#include <queue>
#define AUTHOR "HEX9CF"
using namespace std;

int n;
int a, b;
priority_queue<int> hmax;

int main() {
	scanf("%d", &n);
	for (int t; scanf("%d", &t) != EOF;) {
		hmax.push(t);
	}

	int prev = hmax.top();
	hmax.pop();
	while (hmax.size()) {
		int t = hmax.top();
		hmax.pop();
		if (t == prev) {
			b = t;
		} else if (prev - t == 2) {
			a = t + 1;
		}
		prev = t;
	}

	printf("%d %d", a, b);
	return 0;
}