题目如下:
拍集体照时队形很重要,这里对给定的N个人K排的队形设计排队规则如下:
- 每排人数为N/K(向下取整),多出来的人全部站在最后一排;
- 后排所有人的个子都不比前排任何人矮;
- 每排中最高者站中间(中间位置为m/2+1,其中m为该排人数,除法向下取整);
- 每排其他人以中间人为轴,按身高非增序,先右后左交替入队站在中间人的两侧(例如5人身高为190、188、186、175、170,则队形为175、188、190、186、170。这里假设你面对拍照者,所以你的左边是中间人的右边);
- 若多人身高相同,则按名字的字典序升序排列。这里保证无重名。
现给定一组拍照人,请编写程序输出他们的队形。
输入格式:
每个输入包含1个测试用例。每个测试用例第1行给出两个正整数N(<=10000,总人数)和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
题目解析:
首先处理输入,名字和身高,作为一个列表;将输入信息列表按要求排序;计算每行人数(最后一行不同),江每行的人放入对应的一个列表中(person_eachrow)(按排序顺序,这一步十分方便);最后处理每行的人,按要求输出。这里再解释两处:
1> 排序函数: python3没有cmp了,只能使用key,只能传入一个参数的函数;对于名字string作为第二关键字的比较,鄙人暂时探索发现只能如下使用,不能加-号,所以只可以让x[0]这个int加-号表示逆序;如果比较方法再复杂一些,不知如何实现。
2> 输出控制:ret为人的拍照位置,先找到最中间的位置mid,然后依次像两边放置人,看代码理解一下吧~
n, k = map(int,input().split())
lst = []
for i in range(n):
name, height = input().split()
lst.append([int(height), name])
lst.sort(key=lambda x:(-x[0], x[1]))
n_per_row = n//k
n_lastrow = n - n_per_row * (k-1)
person_eachrow = [lst[:n_lastrow]] + [ lst[n_lastrow + n_per_row * i:n_lastrow + n_per_row * (i+1)] for i in range(k-1)]
# print(person_eachrow)
for pers in person_eachrow:
length = len(pers)
ret = ['' for i in range(length)]
mid = length//2
ret[mid] = pers[0][1]
cn = 1
for j in range(1,length,2):
if j == length - 1:
ret[mid - cn] = pers[j][1]
else:
ret[mid - cn] = pers[j][1]
ret[mid + cn] = pers[j+1][1]
cn += 1
print(" ".join(ret))