Python基础练习 1055 集体照 (25 分)

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

问题描述

　　拍集体照时队形很重要，这里对给定的 N 个人 K 排的队形设计排队规则如下：

• 每排人数为 N/K（向下取整），多出来的人全部站在最后一排；
• 后排所有人的个子都不比前排任何人矮；
• 每排中最高者站中间（中间位置为 m/2+1，其中 m 为该排人数，除法向下取整）；
• 每排其他人以中间人为轴，按身高非增序，先右后左交替入队站在中间人的两侧（例如5人身高为190、188、186、175、170，则队形为175、188、190、186、170。这里假设你面对拍照者，所以你的左边是中间人的右边）；

　　若多人身高相同，则按名字的字典序升序排列。这里保证无重名。

　　现给定一组拍照人，请编写程序输出他们的队形。

输入输出

• 输入格式

　　每个输入包含 1 个测试用例。每个测试用例第 1 行给出两个正整数 N（≤100000​​ ，总人数）和 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

Python版本 代码如下

n, k = map(int, input().split())

m = n//k
r = n%k

L=[] #存储输入的数据
for i in range(n):
    x = input().split()
    L.append([x[0], int(x[1])])
    
#先按照姓名，再按照身高排序，因为题目要求姓名排序时按照升序，必须分别排序
L.sort(key=lambda x:(x[0]), reverse=True)
L.sort(key=lambda x:(x[1]))

ans=[] #初步按照每排身高由高到低站好
for i in range(k):
    rank=[]; t=(i+1)*m-1
    for j in range(m):
        rank.append(L[t][0])
        t-=1
    if i==k-1 and i!=0 and r!=0:
        for e in L[-r:]:
            rank.insert(0, e[0])
    ans.append(rank)

#按照以中间人为轴，按身高非增排序
for i in range(k)[::-1]:
    name=[]
    name.append(ans[i][0]) #这里列表可以按照append和insert来模拟题目要求的左右排序
    for j in range(1,len(ans[i])):
        if j%2==1:
            name.insert(0, ans[i][j])
        else:
            name.append(ans[i][j])
    print(" ".join(name))