python 链式A+B

题目描述

有两个用链表表示的整数，每个结点包含一个数位。这些数位是反向存放的，也就是个位排在链表的首部。编写函数对这两个整数求和，并用链表形式返回结果。

给定两个链表ListNode* A，ListNode* B，请返回A+B的结果(ListNode*)。

测试样例：

{1,2,3},{3,2,1}

返回：{4,4,4}

注意此处代码用的是python2.7

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
import math
class Plus:
    def plusAB(self, a, b):
        newHead = ListNode(0)
        tmp = newHead
        suma = 0
        sumb = 0
        count = 0
        while a :
            suma += a.val * math.pow(10,count)
            a = a.next
            count += 1
        count1 = 0
        while b :
            sumb += b.val * math.pow(10,count1)
            b = b.next
            count1 += 1
        Sum = int(suma + sumb)
        while Sum :
            tt = ListNode(0)
            tt.val = Sum%10
            tmp.next = tt
            tmp = tmp.next
            Sum /= 10
        tmp.next = None
        #tt = ListNode(0)
        #tmp.next = tt
        return newHead.next