美团 · 堆 · 算法编程题
美团 时限 2 秒 / 256 MB

题目描述

小美和小团所在公司的食堂有N张餐桌,从左到右摆成一排,每张餐桌有2张餐椅供至多2人用餐,公司职员排队进入食堂用餐。小美发现职员用餐的一个规律并告诉小团:当男职员进入食堂时,他会优先选择已经坐有1人的餐桌用餐,只有当每张餐桌要么空着要么坐满2人时,他才会考虑空着的餐桌;
当女职员进入食堂时,她会优先选择未坐人的餐桌用餐,只有当每张餐桌都坐有至少1人时,她才会考虑已经坐有1人的餐桌;
无论男女,当有多张餐桌供职员选择时,他会选择最靠左的餐桌用餐。现在食堂内已有若干人在用餐,另外M个人正排队进入食堂,小团会根据小美告诉他的规律预测排队的每个人分别会坐哪张餐桌。
进阶:时间复杂度 O(nlogn) ,空间复杂度 O(n)

输入输出

输入描述
第一行输入一个整数T(1每组数据占四行,第一行输入一个整数N(1第二行输入一个长度为N且仅包含数字0、1、2的字符串,第i个数字表示左起第i张餐桌已坐有的用餐人数;
第三行输入一个整数M(1第四行输入一个长度为M且仅包含字母M、F的字符串,若第i个字母为M,则排在第i的人为男性,否则其为女性。
输出描述
每组数据输出占M行,第i行输出一个整数j(1<=j<=N),表示排在第i的人将选择左起第j张餐桌用餐。

样例共 1 组

样例 1
输入
1
5
01102
6
MFMMFF
输出
2
1
1
3
4
4

算法解析依据充分

考点:堆

限制 2 秒 / 256MB | 标准输入输出

题目画像

  • 源站时限:2 秒(牛客口径,非本题专属门槛)
  • 题目显式声明:复杂度要求 O(nlogn)、O(n)

解题思路

推荐方向:堆

本题切入点

男职员优先选「已有 1 人」的桌、女职员优先选空桌,且都取最靠左;用两个有序结构(堆/集合)分别维护空桌和单人桌,实现 O(n log n)。

用堆在 O(log n) 内取极值,求 TopK 与动态中位数。

思路框架(堆 通法 · 非本题专属)

  1. 求最大 K 个:维护大小 K 的小根堆,超了就把堆顶弹出。
  2. 动态中位数:用「对顶堆」——小的一半放在大根堆,大的一半放在小根堆,两个堆顶就是中位数。
  3. 移动窗口中的中位数需要配合「延迟删除」处理过期元素。

实现要点:Python 的 heapq 是小根堆,要大根堆就存负数。

复杂度:时间 O(n log k) / O(n log n) | 空间 O(k)

该范式的通法易错点

  • 求第 K 大时堆类型选反(小根堆还是大根堆)。
  • 窗口滑出元素时忘记从堆里清理或标记过期。

样例

样例 1

  • 输入:1 / 5 / 01102 / 6 / MFMMFF
  • 输出:2 / 1 / 1 / 3 / 4 / 4

解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。

本题来源:美团2023校招笔试第10场编程题。

‹ 上一题 全部编程题 下一题 ›
编程算法题为只读内容:无需作答,直接看题与解析 · 本站不提供在线判题 · 解析由校招宝本地引擎整理,非官方题解