兩數相加-鏈表 - Java實現

24. 兩數相加-鏈表

題目描述

給出兩個 非空 的鏈表用來表示兩個非負的整數。其中，它們各自的位數是按照 逆序 的方式存儲的，並且它們的每個節點只能存儲 一位 數字。

如果，我們將這兩個數相加起來，則會返回一個新的鏈表來表示它們的和。

您可以假設除了數字 0 之外，這兩個數都不會以 0 開頭。

示例

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

leetcode 地址

https://leetcode-cn.com/problems/add-two-numbers/

代碼

package day;

import day.data.ListNode;
/**
 * Class day24_兩數相加_鏈表 ...
 *
 * @author LiJun
 * Created on 2019/4/8
 */
public class day24_兩數相加_鏈表 {
    /**
     * Definition for singly-linked list.
     * public class ListNode {
     * int val;
     * ListNode next;
     * ListNode(int x) { val = x; }
     * }
     */
    static class Solution {
        public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
            ListNode head, p = head = new ListNode(-1);
            int flag = 0;
            while (l1 != null && l2 != null) {
                p.next = l1;
                p = p.next;
                int t = l1.val;
                p.val = (l1.val + l2.val + flag) % 10;
                flag = (t + l2.val + flag) / 10;
                l1 = l1.next;
                l2 = l2.next;
            }
            while (l1 != null) {
                p.next = l1;
                p = p.next;
                int t = l1.val;
                p.val = (t + flag) % 10;
                flag = (t + flag) / 10;
                l1 = l1.next;
            }
            while (l2 != null) {
                p.next = l2;
                p = p.next;
                int t = l2.val;
                p.val = (l2.val + flag) % 10;
                flag = (t + flag) / 10;
                l2 = l2.next;
            }
            if (flag == 1) {
                p.next = new ListNode(1);
            }
            return head.next;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        ListNode l1 = new ListNode(9);
        l1.next = new ListNode(8);
        // l1.next.next = new ListNode(3);
        ListNode l2 = new ListNode(1);
        // l2.next = new ListNode(6);
        // l2.next.next = new ListNode(4);
        ListNode.print(solution.addTwoNumbers(l2, l1));
    }
}