055 集体照 (25 分)
题意描述:
拍集体照时队形很重要,这里对给定的 N 个人 K 排的队形设计排队规则如下:
-
每排人数为 N/K(向下取整),多出来的人全部站在最后一排;
-
后排所有人的个子都不比前排任何人矮;
-
每排中最高者站中间(中间位置为 m/2+1,其中 m 为该排人数,除法向下取整);
-
每排其他人以中间人为轴,按身高非增序,先右后左交替入队站在中间人的两侧(例如5人身高为190、188、186、175、170,则队形为175、188、190、186、170。这里假设你面对拍照者,所以你的左边是中间人的右边);
-
若多人身高相同,则按名字的字典序升序排列。这里保证无重名。
现给定一组拍照人,请编写程序输出他们的队形。
输入格式:
每个输入包含 1 个测试用例。每个测试用例第 1 行给出两个正整数 N(≤10^4 ,总人数)和 K(≤10,总排数)。随后 N 行,每行给出一个人的名字(不包含空格、长度不超过 8 个英文字母)和身高([30, 300] 区间内的整数)。
输出格式:
输出拍照的队形。即K排人名,其间以空格分隔,行末不得有多余空格。注意:假设你面对拍照者,后排的人输出在上方,前排输出在下方。
输入样例:
10 3
Tom 188
Mike 170
Eva 168
Tim 160
Joe 190
Ann 168
Bob 175
Nick 186
Amy 160
John 159
输出样例:
Bob Tom Joe Nick
Ann Mike Eva
Tim Amy John
解题思路:
Alice: emm, 这也是道模拟题吧,逻辑倒是很清楚,就是不太好写。
Bob: emmm, 是有点小麻烦,先要对字典排序,然后每排还要左右交替地排序。list只能往后面插值,需要一个既能在左边插值,又能在右边插值的数据结构。
Alice: 队列??Python中应该有吧⊙(・◇・)?
Bob: 不是队列,队列是一个一边插值,另一边取值。我们是要两边都能插值,应该是叫双向队列,collections
里面有个deque
的模块就是这个了。
Alice: 对字典按值排序,还是用cmp_to_key
吗 ?
Bob: emmm, 这是一种可行的方法,目前我还就只会这一个,┐(:´ゞ`)┌
Alice: - _ - | |
Bob: - _ - | |
代码:
- Alice’s Python Version:
from functools import cmp_to_key
# 使用 cmp_to_key,对字典排序时要用到
from collections import deque
# deque是一个双向队列,既可以在左边插值,也可以在右边插值
def main():
N, K = (int(z) for z in input().split())
# N 是 总人数, K 是排数
global data
data = {}
# 将 data 声明为全局变量,下面的排序函数要用。
for x in range(N):
name, height = (z for z in input().split())
# 依次接收每个人的姓名和身高
height = int(height)
data[name] = height
# 存进到字典data中
sorted_indexs = sorted(data, key=cmp_to_key(compare), reverse=True)
# 对字典按照 值 排序,sorted_indexs 为排序后的 键 的列表
m = int(N / K)
# 前 K - 1 排每排的人数
last_line = N - m * (K - 1)
# 最后一排的人数
answer = []
# 从最后一排开始,对每排内的人排序
answer.extend([sort_in_line(sorted_indexs[:last_line])])
# sorted_indexs[:last_line] 获取最后一排的人名列表
# sort_in_line(sorted_indexs[:last_line]) 对这一排的人排序
start = last_line
for x in range(K - 1):
# 一共 K 排,对最后一排排序后,对剩余的 K-1 排排序
answer.extend(
[sort_in_line(sorted_indexs[start + m * x:start + m * (x + 1)])])
# sorted_indexs[start + m * x : start + m * (x + 1)] 获取这一排的人名
# 排序后,添加到 answer 中
for x in answer:
print(' '.join(x))
# 依次输出每排排序后的结果
def sort_in_line(names):
# 根据身高,右左交替地对一排中的人排序
left = 1
temp = deque()
for name in names:
if left == 1:
temp.append(name)
else:
temp.appendleft(name)
left *= -1
# 右左交替
return list(temp)
# 将deque类型转换为list类型后返回之。
def compare(a, b):
# 对字典排序,按身高降序排列,若身高相等,按姓名字典序升序排列
if data[a] != data[b]:
return data[a] - data[b]
else:
if a > b:
return -1
return 1
if __name__ == '__main__':
main()
易错点:
- 左右交替地插值 ??
- 按身高降序排序,身高相等的按照姓名字典序升序排序 ??
总结: