T2. 排队选人 - 小米前端笔试编程题解

考试平台: 赛码

题目类型: 20道选择 + 2道编程题

考试时间: 2024-03-23 (两小时)

题目描述

小D是一名老师,他想选出一些同学参加一个团体比赛。

总共有n个同学,每个同学有一个能力值x和一个合作值y,小D将这些同学排成一排。

选人的规则如下:

  • 小D将选择一排中连续的k个同学参加比赛,而且为了保证比赛成绩,这k个同学的能力值不影小于a,合作值不能小于b。

小D想知道总共有多少种可能的选人方案。

输入描述

第一行四个正整数n,k,a,b。

第二行n个通过空格隔开的正整数表示一排n个同学的能力值。

第三行n个通过空格隔开的正整数表示一排n个同学的合作值。

输出描述

输出一个非负整数表示选人方案数。

样例输入

复制代码
10 2 2 4
2 2 9 1 8 1 6 1 7 7
4 8 5 1 9 4 1 3 9 4

样例输出

复制代码
3

提示

有选择第1,2两个人,第2,3两个人,第9,10两个人三种方案。

n <= 80000,保证k<=n月x,y,a,b<=100000

题解

这道题属于数组和滑动窗口类型的算法题。解题思路是使用滑动窗口来统计符合条件的连续k个同学的选人方案数。

代码大致描述:

  1. 首先读取输入的n、k、a、b以及每个同学的能力值和合作值。
  2. 使用两个数组ab和co分别存储每个同学的能力值和合作值。
  3. 设置一个变量cnt来统计当前滑动窗口内符合条件的同学数量,初始化为0。
  4. 遍历滑动窗口,从第一个同学开始,依次判断每个同学是否符合条件:
    • 如果符合条件,则cnt加1。
    • 如果窗口大小超过k,需要将窗口左边界的同学移除,同时更新cnt。
    • 如果cnt等于k,说明当前窗口内的同学符合条件,将结果计数器result加1。
  5. 最终输出结果result。

时间复杂度:该算法的时间复杂度为O(n),其中n为同学数量。

空间复杂度:该算法的空间复杂度为O(n),需要两个数组来存储同学的能力值和合作值。

python 复制代码
n,k,a,b = map(int,input().split())

ab = list(map(int,input().split())) # 能力值
co = list(map(int, input().split())) # 合作值

cnt = 0 # 统计当前滑动窗口内符合条件的同学数量
for i in range(k-1):
    if ab[i] >= a and co[i] >= b:
        cnt += 1

result = 0
for i in range(k-1, n):
    if ab[i] >= a and co[i] >= b:
        cnt += 1
    if i >= k:
        if ab[i - k] >= a and co[i - k] >= b:
            cnt -= 1
    if cnt == k:
        result += 1
print(result)

🙏整理题解不易, 如果有帮助到您,请给点个赞 ‍❤️‍ 和收藏 ⭐,让更多的人看到。🙏🙏🙏

相关推荐
chao_78918 分钟前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
烛阴27 分钟前
Python装饰器解除:如何让被装饰的函数重获自由?
前端·python
千鼎数字孪生-可视化41 分钟前
Web技术栈重塑HMI开发:HTML5+WebGL的轻量化实践路径
前端·html5·webgl
凌辰揽月42 分钟前
7月10号总结 (1)
前端·css·css3
noravinsc1 小时前
django 一个表中包括id和parentid,如何通过parentid找到全部父爷id
python·django·sqlite
lifallen1 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法
天天扭码1 小时前
很全面的前端面试——CSS篇(上)
前端·css·面试
lixzest1 小时前
C++ Lambda 表达式详解
服务器·开发语言·c++·算法
ajassi20001 小时前
开源 python 应用 开发(三)python语法介绍
linux·python·开源·自动化
EndingCoder1 小时前
搜索算法在前端的实践
前端·算法·性能优化·状态模式·搜索算法