Leetcode热门100之两数相加

最新推荐文章于 2024-10-04 20:45:38 发布

Channel_R

最新推荐文章于 2024-10-04 20:45:38 发布

阅读量106

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/Channel_R/article/details/107254842

版权

LeetCode 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

问题描述：

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

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

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

示例：

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

代码：
初始代码：

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* l3 = new ListNode();
        ListNode* node = l3;
        int flag = 0;
        while(l1 != NULL && l2 != NULL){
                int tmp = l1->val + l2->val + flag;
                node->val = tmp % 10;
                if(tmp >= 10) {
                    flag = 1;
                }
                else flag = 0;
                if(l1->next != NULL || l2->next != NULL || flag == 1){
                    ListNode* nxt = new ListNode();
                    node->next = nxt;
                    node = node -> next;
                }
                else node->next = NULL;
                l1 = l1->next;
                l2 = l2->next;
        }
        if(l1 != NULL){
            node->val = l1->val + flag;
            node->next = NULL;
        }
        if(l2 != NULL){
            node->val = l2->val + flag;
            node->next = NULL;
        }
        if (flag == 1){
            node->val = flag;
            node->next = NULL;
        }
        return l3;
    }
};

改进代码：

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* l3;
        ListNode* node = new ListNode();
        l3 = node;
        int tmp=0;
        while(l1 || l2){
            if(node->next){
                node = node->next;//解决了无进位时新节点存0造成结果错误的问题
            }
            if(l1){
                tmp += l1->val;
                l1 = l1->next;//指针移动要考虑当前指针是否为空              
            }
            if(l2){
                tmp += l2->val;
                l2 = l2->next;
            }
            node->val = tmp % 10;
            tmp /= 10;//相当于进位，直接应用到下一次求和运算中，不需要额外保存
            node->next = new ListNode();
        }
        if(tmp){
                node->next->val = tmp;
                node->next->next = NULL;
        }
        else node->next = NULL;
        return l3;
    }
};