1969 品种邻近(枚举 + 滑动窗口)

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())
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值