PTA:Python解答1055 集体照

1055 集体照 (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

解题思路:

这道题主要考察的是排序问题:按身高排序、身高相同按名字排序,用好sorted(n, key=lambda , reverse)就能快速解决问题。

完整代码:

a, b = [int(x) for x in input().split()]
l = []
for i in range(b):
    if i == 0:
        l.append(a // b + a % b)
    else:
        l.append(a // b)
n = []
for i in range(a):
    s = input().split()
    n.append(s)
n = sorted(n, key=lambda x: x[0])
n = sorted(n, key=lambda x: int(x[1]), reverse=True)
for i in l:
    ls = []
    if i % 2 == 0:
        for j in range(i):
            if j % 2 == 0:
                ls.append(n.pop(0)[0])
            else:
                ls.insert(0, n.pop(0)[0])
    else:
        ls.append(n.pop(0)[0])
        for j in range(i - 1):
            if j % 2 == 0:
                ls.insert(0, n.pop(0)[0])
            else:
                ls.append(n.pop(0)[0])
    print(' '.join(ls))
点赞

发表评论

电子邮件地址不会被公开。必填项已用 * 标注