运行时错误:如何在一个循环内缩短代码

时间:2019-08-03 11:36:33

标签: python python-3.x algorithm sorting data-structures

我正在解决一个运行时错误的问题,如果还有其他错误,我们可以缩短代码长度,请提出建议

T=int(input())

for i in range(T):
    N= int(input())
    P=[0]*N

    C=list(map(int,input().split()))
    H=list(map(int,input().split()))
    for i in range(N):
        if((i-C[i]>=0)and (i+C[i]<N)):
                for j in range(i-C[i],i+C[i]+1):
                    P[j]=P[j]+1
        elif((i-C[i]>=0)and (i+C[i]>N)):
                for j in range(i-C[i],N):
                    P[j]=P[j]+1




        elif((i+C[i]<N) and (i-C[i]<0)) :
                for j in range(0,i+C[i]+1):
                    P[j]=P[j]+1
        else:
                for j in range(0,N):
                    P[j]=P[j]+1
    H=sorted(H)
    P=sorted(P)
    if(H==P):
        print("YES")
    else:
        print("NO")

1 个答案:

答案 0 :(得分:0)

对于测试用例和数组,您使用相同的循环变量。 在测试用例循环中使用其他任何变量,则不会出现运行时错误。