PAT~乙级~1055 集体照 ~python

骑猪看日落 2024-04-19 12:22 25阅读 0赞

题目描述：

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

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

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

### 输入格式： ###

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

\----------------------------------------------------------答题区----------------------------------------------------------

people_num,rank_num=list(map(int,input().split()))
    res = []
    for i in range(people_num):
        name,height = input().split()
        height = int(height)
        res.append((name,height))
    res.sort(key=lambda s:(-s[1], s[0]),reverse=False)
    last_rank = people_num//rank_num + people_num%rank_num
    other_rank = people_num//rank_num
    last_lst=['']*last_rank
    other_lst=['']*other_rank
    r_l_flag=1
    mid_cur=last_rank//2
    right_cur=mid_cur
    left_cur=mid_cur
    for i in res[:last_rank]:
        if(last_lst[mid_cur]==''):
            last_lst[mid_cur]=i[0]
        if(right_cur<=last_rank-1 and last_lst[right_cur]==''):
            last_lst[right_cur] =i[0]
        if (left_cur>=0 and last_lst[left_cur] == ''):
            last_lst[left_cur] = i[0]
        if(r_l_flag==0):
            right_cur+=1
            r_l_flag=1
        else:
            left_cur-=1
            r_l_flag=0
    print(' '.join(last_lst))
    
    num_flag=0
    mid_cur = other_rank // 2
    right_cur = mid_cur
    left_cur = mid_cur
    r_l_flag=1
    
    for i in res[last_rank:]:
        if (other_lst[mid_cur] == ''):
            other_lst[mid_cur] = i[0]
        if (right_cur <= other_rank - 1 and other_lst[right_cur] == ''):
            other_lst[right_cur] = i[0]
        if (left_cur >= 0 and other_lst[left_cur] == ''):
            other_lst[left_cur] = i[0]
        if (r_l_flag == 0):
            right_cur += 1
            r_l_flag = 1
        else:
            left_cur -= 1
            r_l_flag = 0
        num_flag+=1
        if (num_flag == other_rank):
            print(' '.join(other_lst))
            mid_cur = other_rank // 2
            right_cur = mid_cur
            left_cur = mid_cur
            r_l_flag = 1
            other_lst = [''] * other_rank
            num_flag = 0

维护了一个PAT测试点问题的共享文档，欢迎大家把遇到的问题写在上面。

一起写文档：https://yiqixie.com/d/home/fcACe6xZ2aroPxFQrQZ05Cw32  
github链接：https://github.com/Makerdd/PAT-yiji