PAT 1010(多项式求导) python代码

num = [int(x) for x in input().split()]

for i in range(0,len(num)-1,2):
    num[i] = num[i] * num[i+1]
    num[i + 1] -= 1
    if num[i] == 0:
        num[i],num[i+1] = 'x','x'

while 'x' in num:
    num.remove('x')
if num == []:
    print('0 0',end='')
else:
    for i in range(0,len(num)):
        if i == len(num)-1:
            print(num[i],end='')
        else:
            print(num[i],end=' ')

注意当求导后的列表为空时要输出：0 0