请问大佬,这题用Python怎么做呀,最后的测试一直超时
查看原帖
请问大佬,这题用Python怎么做呀,最后的测试一直超时
785202
love_money楼主2024/1/26 09:30
n,m,a,b=map(int,input().split())
num=[]
for i in range(a):
    t=list(map(int,input().split()))
    num.append(t)
song=[]
for i in range(n):
    score=0
    for j in range(a):
        score+=num[j][i]
    song.append(score)

s=sorted(song,reverse=True)
x=num[b-1].index(max(num[b-1]))+1
choice=[]

for i in range(m):
    choice.append(song.index(s[i])+1)

if x in choice and choice.index(x)!=0:
    choice.remove(x)
    choice.insert(0,x)
elif x not in choice:
    choice[m-1]=x
for i in choice:
    print(i,end=" ")
2024/1/26 09:30
加载中...