小美和小团所在公司的食堂有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 5 01102 6 MFMMFF
2 1 1 3 4 4
考点:堆
限制 2 秒 / 256MB | 标准输入输出
推荐方向:堆
本题切入点
男职员优先选「已有 1 人」的桌、女职员优先选空桌,且都取最靠左;用两个有序结构(堆/集合)分别维护空桌和单人桌,实现 O(n log n)。
用堆在 O(log n) 内取极值,求 TopK 与动态中位数。
思路框架(堆 通法 · 非本题专属)
实现要点:Python 的 heapq 是小根堆,要大根堆就存负数。
复杂度:时间 O(n log k) / O(n log n) | 空间 O(k)
该范式的通法易错点
样例 1
1 / 5 / 01102 / 6 / MFMMFF2 / 1 / 1 / 3 / 4 / 4解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。
本题来源:美团2023校招笔试第10场编程题。