算法学习:leetcode 2 两数相加

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

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

示例:

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

思路:
我们要考虑的情况有:
1-两个数相加的进位问题;
2-两个数的长短问题,比如一个是3位数,一个是4位数。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode currNode = new ListNode(0);
        ListNode p=l1,q=l2,curr=currNode;
        //标记前一位的计算是否有进位,0-无 1-有
        int flag = 0;

        while (p != null || q != null) {
            //取出当前位的值
            int x = (p != null) ? p.val : 0;
            int y = (q != null) ? q.val : 0;
            curr.next = new ListNode(0);
            curr = curr.next;
            //计算当前位的值,包含标记位的值
            int temp = flag + x + y;
            if(temp >=10) {
                //有进位
                curr.val = temp - 10;
                flag = 1;
            } else {
                //无进位
                curr.val = temp;
                flag = 0;
            }
            if (p != null) p = p.next;
            if (q != null) q = q.next;

        }
        //falg不为0说明有进位,直接新建一个next节点设置为1
        if(flag > 0) {
            curr.next = new ListNode(1);
        }
        return currNode.next;
    }
}

时间复杂度O(n)

空间复杂度O(n)

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers