在Codechef中提交解决方案时出现NZEC错误

Question

这是我对codechef的特定问题的代码：

n=int(input())
for _ in range(n):
    a,b =input().split()
    a=int(a)
    b=int(b)
    x=[int(q) for q in input().split()]
    x.sort(reverse=True)
    new_x=[item for pos,item in enumerate(x) if x.index(item)==pos]
    last=new_x[b]
    i=x.index(last)
    ans= i - 1 + x.count(last)
    print(ans)

当我在自己的环境中尝试它时，它会通过所有测试用例，但是在提交解决方案时会显示运行时（NZEC）。 我在互联网上进行了大量搜索，但无法找出问题所在。 请帮帮我。