2. 两数相加 Go/模拟

162 阅读1分钟

2. 两数相加 Go/模拟

给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例 1:

输入: l1 = [2,4,3], l2 = [5,6,4]
输出: [7,0,8]
解释: 342 + 465 = 807.

示例 2:

输入: l1 = [0], l2 = [0]
输出: [0]

示例 3:

输入: l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出: [8,9,9,9,0,0,0,1]

解题思路

模拟
坑1:要考虑到,如果相加等于9,同时之前的要进一位
坑2:要考虑到,如果最后flag=1 要进一位

代码

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
    var l3, head *ListNode
    var flag int
    for i:=0; i<110; i++{
        var n, m,num int = 0, 0, 0
        if l1 != nil{
            n =  l1.Val
            l1 = l1.Next
        }
        
        if l2 != nil{
            m = l2.Val
            l2 = l2.Next
        }

        if n+m>=10{
            num = (n+m)%10+flag
            flag=1
        }else{
            num= n+m+flag
            if(num>=10){
                num = num%10
                flag=1
            }else{
                flag=0
            }
        }

        if head==nil{
            head = &ListNode{Val:num}
            l3 = head
        }else{
            l3.Next = & ListNode{Val:num}
            l3 = l3.Next
        }

        if l1==nil && l2==nil{
            if flag==1{
                l3.Next = & ListNode{Val:1}
                l3 = l3.Next
            }
            break
        }
    }
    return head
}