Leetcode-2. 两数相加

120 阅读2分钟

题目

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

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

你可以假设除了数字 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]

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

解题思路

一开始我是处理两个链都不为空的情况,然后while循环后判断两个链是否有一条不为空,然后处理进位和剩余链的求和,算法很麻烦。

后来发现其实while的条件写成或,情况会简单很多,进入循环后如果链当前位置为空就给一个0,这样循环结束后,只需要判断是否有进位就行了,如果有就结尾加一个1.

另外,我最开始求和链是没有一个空heading的,在循环中处理求和链的下标很麻烦,其实一开始创建一个空的节点,之后在循环中给next创建空间,最后返回heading.next就很方便,代码量很小。

代码

public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
    // empty Heading
    ListNode ret = new ListNode();
    ListNode retHeading = ret;
    // 用来存储进位
    int carry = 0;

    // 这里用或来处理逻辑
    while (l1 != null || l2 != null) {
        // 如果链为空就给0
        int x = l1 != null ? l1.val : 0;
        int y = l2 != null ? l2.val : 0;
        int sum = x + y + carry;

        // 因为加入了空链头,这里直接放到next中
        retHeading.next = new ListNode(sum%10);
        carry = sum/10;
        
        // 因为是或,所以这里要进行判断,非空链往后移
        if(l1 != null) l1 = l1.next;
        if(l2 != null) l2 = l2.next;
        retHeading = retHeading.next;
    }

    // 处理最后的进位
    if (carry != 0) {
        retHeading.next = new ListNode(carry);
    }
    
    // 这里一定要返回.next,因为引入了一个空头
    return ret.next;
}