Golang刷题-2. 两数相加

1. 题目描述【中等】

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题思路

本题并不算难,思路也很清晰,只不过是对链表的一些操作细节需要注意。

2.1 直接对链表进行操作动画参考

两数之和
注:图片来自五分钟学算法公众号

算法流程:

  1. 首先对给定链表进行非空判断,并定义保存结果的head链表
  2. l1l2同时进行遍历,若l1l2已经为空则不再遍历该链表
  3. 将得到的数字相加结果保存为sum
  4. 如果sum>=10则代表需要进位,则用flag作为标记,同时将sum=sum-10
  5. 否则不需要进位,将flag标记为0
  6. 采用尾插法将结果保存在head链表中

3. 代码实现

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
	if l1 == nil && l2 == nil {
		return nil
	}else if l1 == nil{
		return l2
	}else if l2 == nil{
		return l1
	}
	head := &ListNode{
		Val:-1,
		Next:nil,
	}
	tail := head

	flag := 0
	for l1 != nil || l2 != nil{
		val1, val2 := 0, 0
		if l1 != nil{
			val1 = l1.Val
			l1 = l1.Next
		}
		if l2 != nil{
			val2 = l2.Val
			l2 = l2.Next
		}

		sum := val1 + val2 + flag
		if sum >= 10 {
			flag = 1
			sum -= 10
		}else{
			flag = 0
		}
		tail = listAdd(tail, sum)
	}
	if flag == 1{
		tail = listAdd(tail, 1)
	}
	return head.Next
}

//根据题目要求采用尾插法创建链表
func listAdd(tail *ListNode, num int)*ListNode{
	numNode := &ListNode{
		Val:num,
		Next:nil,
	}
	tail.Next  = numNode
	tail = numNode
	return tail
}
发布了31 篇原创文章 · 获赞 25 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/s_842499467/article/details/104327018