【leetcode3】两非空链表相加

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

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

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

示例:

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

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

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
     
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
     
        ListNode root(0);
        ListNode *res = &root;
        int add_flag = 0;
        while (l1 || l2 || add_flag){
     
            int l1_val = l1 == NULL ? 0 : l1->val;
            int l2_val = l2 == NULL ? 0 : l2->val;
            int sum = l1_val + l2_val + add_flag;
            add_flag = sum / 10;
            ListNode *tmpNode = new ListNode(sum % 10);
            tmpNode->val = sum % 10;
            res->next = tmpNode;
            res = tmpNode;
            
            l1 = l1 == NULL ? nullptr : l1->next;
            l2 = l2 == NULL ? nullptr : l2->next;
        }
        return root.next;
    }
};

你可能感兴趣的:(算法学习)