【PTA】【数据结构与算法题目集】(Python实现)7-2 一元多项式的乘法与加法运算

这一题卡在第四个测试点上好久。。
“输入有零多项式和常数多项式”，就是最后添加的几个if…else。。就通过了

class ListNode:
    def __init__(self,coef,exp):
        self.coef = coef
        self.exp = exp
        self.next = None

def list2LinkList(lst):
    if lst[0]==0:
        return 0
    head = ListNode(0,0)
    tail = head
    for i in range(1,len(lst),2):
        node = ListNode(lst[i],lst[i+1])
        # print(node.coef,node.expon)
        tail.next = node
        tail = node
    return head.next

def add(lst1,lst2):
    h1 = lst1
    h2 = lst2
    addRes = []
    while h1 and h2:
        e1 = h1.exp
        e2 = h2.exp
        if e1<e2:
            if h2.coef!=0:
                addRes.append(h2