055 集体照 (25 分)

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()

---

易错点：

• 左右交替地插值 ？？
• 按身高降序排序，身高相等的按照姓名字典序升序排序 ？？

---

总结：

• deque
• cmp_to_key
• deque
  

---