LeetCode2. 两数相加

67 阅读1分钟

题目描述:

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

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

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

示例:

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

思路:

 遍历这两个链表,把结果存到另一个链表中,最后判断一下是否还有进位即可

/**
 * 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 *result = new ListNode(0);
        ListNode *head = result;
        int t = 0;
        
        while(l1 != NULL || l2 != NULL) {
            int sum = 0;
            if(l1 != NULL && l2 != NULL)
                sum = l1->val + l2->val + t;
            else if(l1 != NULL)
                sum = l1->val + t;
            else
                sum = l2->val + t;
            result->next = new ListNode(sum % 10);
            t = sum / 10;
            if(l1 != NULL)
                l1 = l1->next;
            if(l2 != NULL)
                l2 = l2->next;
            result = result->next;
        }
    
        if(t != 0) {
            ListNode *p = new ListNode(t % 10);
            result->next = p;
        }
        return head->next;
        
    }
};