PAT 甲级 1056 Mice and Rice

1056 Mice and Rice

题目大意：给出老鼠的数量和每组的人数，将老鼠分成若干组（多下来不够每组人数的算作一组），第二行给出每只老鼠最终的重量，第三行给出分组顺序

思路：记录每轮获胜的选手，淘汰掉的人的排名就是当前所在的层数，将获胜的选手继续下一轮排名。注意这里分组需要考虑总人数不满一组的情况！

n,m=map(int,input().split())
weight=list(map(int,input().split()))   #每只老鼠的重量
order=list(map(int,input().split()))    #按照order顺序分组
new_weight=[]
for i in range(n):
    new_weight.append(weight[order[i]])
win=order  #存放每轮获胜选手的下标
group=int(n/m)+1 if n%m!=0 else int(n/m)
trank=group+1  #每组老鼠的排名
rank=[trank]*n
index=0
while len(win)!=1:
    group=int(len(win)/m)+1 if len(win)%m!=0 else int(len(win)/m)   #这里取整要注意不满一组的情况
    trank=group+1
    for i in win:
        rank[i]=trank
    win=[]
    index=0
    for i in range(group):
        max_weight=max(new_weight[index:index+m])
        win_index=weight.index(max_weight)
        win.append(win_index)
        index+=m
    new_weight=[]
    for i in win:
        new_weight.append(weight[i])
rank[win[0]]=1
print(*rank)