Python算法输出故障排除

Question

我附上了一个python 2.7脚本来回答以下链接中的问题2：http://labs.spotify.com/puzzles/

我的解决尝试目前正在回复“错误答案”，但我的代码成功地用于网站上的示例输入。我已经尝试修改它以返回甚至打印前m个歌曲的列表，而不是单独打印出来，但这也不起作用。任何帮助或想法都会很棒。提前致谢

import sys

def main():

    line1 = sys.stdin.readline().split()
    total_songs = int(line1[0])-1
    num_songs_return = int(line1[1])

    data = sys.stdin.read().split()

    while(total_songs >= 0):
        data[2*total_songs]= float(data[2*total_songs]) * (total_songs+1)
        total_songs-=1

    answers = [(data[a], data[a+1]) for a in range(0,len(data),2)]
    answers.sort(reverse=True)
    for n in range(num_songs_return):
        print answers[n][1]

main()