python两数相加代码_Leetcode2. 两数相加 Python实现

题目要求：

思路：

遍历第一个与第二个链表，把相同位置的相加

设置一个flag，用来标记相加后大于10的部分，下一位再相加时，就要加上flag的值

遍历至一个链表结束

如果此时flag不为0，把未结束的链表当前的第一位加上flag

falg记录相加大于10的部分

直至当前列表也遍历完

如果遍历完，flag仍然不为0，再补一位1

核心代码：

遍历两个链表：

# 把链表1 2 的头赋给h1，h2

h1 = l1

h2 = l2

# 创建新链表，用来保存计算完的数

l3 = ListNode()

h3 = l3

# flag标志两数加完大于9的部分

flag = 0

# 遍历两个链表，h3当前的值为h1+h2去掉十位的部分，flag保存进位，运算完h1，h2，h3移向下一位

while h1 and h2 :

h3.next = ListNode((h1.val + h2.val + flag) % 10)

flag = (h1.val + h2.val + flag) // 10

h1 = h1.next

h2 = h2.next

h3 = h3.next

- 如果链表1或2未遍历完:

#原理如上

if h1 :

while h1 :

h3.next = ListNode((h1.val + flag) % 10)

flag = (h1.val + flag) // 10

h1 = h1.next

h3 = h3.next

# 用elif，如果是h1有未遍历的，遍历结束后就不遍历h2了

elif h2 :

while h2 :

h3.next = ListNode((h2.val + flag) % 10)

flag = (h2.val + flag) // 10

h2 = h2.next

h3 = h3.next

- 如果最后一位相加后，仍需进位：

if flag == 1:

h3.next = ListNode(1)

完整代码：加上判断l1和l2一开始是否为空，如果l1为空，返回l2，如果l2为空，返回l1

# Definition for singly-linked list.

# class ListNode(object):

# def __init__(self, x):

# self.val = x

# self.next = None

class Solution(object):

def addTwoNumbers(self, l1, l2):

"""

:type l1: ListNode

:type l2: ListNode

:rtype: ListNode

"""

if l1 is None:

return l2

if l2 is None:

return l1

h1 = l1

h2 = l2

l3 = ListNode()

h3 = l3

flag = 0

while h1 and h2 :

h3.next = ListNode((h1.val + h2.val + flag) % 10)

flag = (h1.val + h2.val + flag) // 10

h1 = h1.next

h2 = h2.next

h3 = h3.next

if h1 :

while h1 :

h3.next = ListNode((h1.val + flag) % 10)

flag = (h1.val + flag) // 10

h1 = h1.next

h3 = h3.next

elif h2 :

while h2 :

h3.next = ListNode((h2.val + flag) % 10)

flag = (h2.val + flag) // 10

h2 = h2.next

h3 = h3.next

if flag == 1:

h3.next = ListNode(1)

return l3.next