1. 问题描述:
农夫约翰的 N 头奶牛排成一排,每头奶牛都用其品种 ID 进行描述。如果两头相同品种的牛靠得太近,它们就会吵架。具体的说,如果同一品种的两头奶牛在队列中的位置相差不超过 K,我们就称这是一对拥挤的牛。请计算品种 ID 最大的拥挤奶牛对的品种 ID。
输入格式
第一行包含两个整数 N 和 K。接下来 N 行,每行包含一个整数表示队列中一头奶牛的品种 ID。
输出格式
输出品种 ID 最大的拥挤奶牛对的品种 ID。如果不存在拥挤奶牛对,则输出 −1。
数据范围
1 ≤ N ≤ 50000,
1 ≤ K < N,
品种 ID 范围 [0,10 ^ 6]。
输入样例:
6 3
7
3
4
2
3
4
输出样例:
4
样例解释
一对品种 ID 为 3 的奶牛以及一对品种 ID 为 4 的奶牛属于拥挤奶牛对。所以,最大拥挤奶牛对的品种 ID 为 4。
来源:https://www.acwing.com/problem/content/description/1971/
2. 思路分析:
首先看数据范围为50000,所以考虑枚举,枚举每一个当前的ID,因为是判断当前长度为k的区间内是否存在相同的ID,所以我们只需要看靠近当前位置长度为k的区间是否存在当前枚举的ID即可,可以使用队列来维护长度为k的区间元素,哈希表来记录当前长度为k的区间的元素出现次数,如果发现哈希表中存在当前枚举的ID那么更新一下答案即可。
3. 代码如下:
import collections
class Solution:
def process(self):
n, k = map(int, input().split())
count = collections.defaultdict(int)
q = collections.deque()
idx = -1
for i in range(n):
num = int(input())
q.append(num)
# 判断长度为k的窗口是否存在当前的num
if count[num] > 0: idx = max(idx, num)
count[num] += 1
# 窗口的长度大于了k那么需要弹出队首元素
if len(q) > k:
t = q.popleft()
count[t] -= 1
return idx
if __name__ == "__main__":
print(Solution().process())