PTA乙级题目1010（python3）

PTA乙级题目1010（python3）

题目信息:
设计函数求一元多项式的导数。（注：x
​n
​​ （n为整数）的一阶导数为nx
​n−1
​​ 。）

输入格式:
以指数递降方式输入多项式非零项系数和指数（绝对值均为不超过 1000 的整数）。数字间以空格分隔。

输出格式:
以与输入相同的格式输出导数多项式非零项的系数和指数。数字间以空格分隔，但结尾不能有多余空格。注意“零多项式”的指数和系数都是 0，但是表示为 0 0。

python3

list1 = list(map(int, input().strip().split()))

# 只有两个都不为0才有数值
def fun1(a, b):
    if a == 0 and b == 0:
        return None
    elif b == 0 and a != 0:
        return None
    else:
        return [a * b, b - 1]

result = ""

# 结果长度大于2，无需输出0 0项
if len(list1) > 2:
    for i in range(0, len(list1) - 2, 2):
        list2 = fun1(list1[i], list1[i+1])
        if list2 != None:
            result += str(list2[0]) + " " + str(list2[1]) + " "
        

list3 = fun1(list1[-2], list1[-1])

# 长度大于2且最后结果不为空
if list3 != None and result != "":   
    result += str(list3[0]) + " " + str(list3[1])
# 长度大于2但最后结果为空
elif list3 == None and result != "":
	# 去掉最后空格
    result = result.rstrip(" ")
# 长度小于2，输出0 0
elif list3 == None and result == "":
    result += "0 0"


print(result)