LeetCode 2 两数相加

题目:

给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。

请你将两个数相加,并以相同形式返回一个表示和的链表。

你可以假设除了数字 0 之外,这两个数都不会以 0 开头。

解题:

把链表相同位置的值加起来,满10就向下一位进一

代码

type ListNode struct {
   Val  int
   Next *ListNode
}

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
   var head, node *ListNode
   flag := 0
   for l1 != nil || l2 != nil {
      n1, n2 := 0, 0
      if l1 != nil {
         n1 = l1.Val
         l1 = l1.Next
      }
      if l2 != nil {
         n2 = l2.Val
         l2 = l2.Next
      }
      sum := n1 + n2 + flag
      flag = sum / 10
      sum = sum % 10
      if head == nil {
         head = &ListNode{Val: sum}
         node = head
      } else {
         node.Next = &ListNode{Val: sum}
         node = node.Next
      }
   }
   if flag > 0 {
      node.Next = &ListNode{Val: 1}
   }
   return head
}

刚开始忘了最后对进位做判断,出错了。循环结束后,需对进位做判断,如果大于0,则需生成一个新节点,值为1。

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注