力扣LeetCode 热题 HOT 100 C++ （2/100) 两数相加

最新推荐文章于 2023-10-07 09:00:00 发布

Randome_

最新推荐文章于 2023-10-07 09:00:00 发布

阅读量204

点赞数

分类专栏：寒假刷题计划文章标签：链表 leetcode 算法

本文链接：https://blog.csdn.net/weixin_51617698/article/details/128699440

版权

寒假刷题计划专栏收录该内容

3 篇文章 0 订阅

订阅专栏

这是一个用链表实现的高精度加法，对两个数的每一位进行相加并记录进位。要注意的是不要漏掉执行完carry不为0的情况。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* dummy = new ListNode(0); //虚拟头结点
        ListNode* cur = dummy; //cur指针指向dummy
        int carry = 0; //进位初始化为0
        while(l1 || l2){ //l1和l2有一个链表没有遍历完就循环
            int val1 = l1 ? l1 -> val : 0; //如果l1不为空，val1就为l1所指结点的值，否则为空
            int val2 = l2 ? l2 -> val : 0;
            int sum = val1 + val2 + carry;
            cur -> next = new ListNode(sum % 10); //将sum的值赋给cur的下一个结点
            carry = sum / 10;
            cur = cur -> next;
            if(l1){ //l1不为空就指向下一个
                l1 = l1 -> next;
            }
            if(l2){
                l2 = l2 -> next;
            }
        }
        if(carry){ //如果carry不为0的话要进一位
            cur -> next = new ListNode(carry);
        }
        return dummy -> next; //从虚拟头结点的下一位开始返回
    }
};

跑出来的结果不是很好看，进行了一些小小的优化，用三目运算符代替了两个if。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* dummy = new ListNode(0);
        ListNode* cur = dummy;
        int carry = 0;
        while(l1 || l2){
            int val1 = l1 ? l1 -> val : 0;
            int val2 = l2 ? l2 -> val : 0;
            int sum = val1 + val2 + carry;
            cur -> next = new ListNode(sum % 10);
            carry = sum / 10;
            cur = cur -> next;
            l1 = l1 ? l1 -> next : l1; //优化内容
            l2 = l2 ? l2 -> next : l2;
        }
        if(carry){
            cur -> next = new ListNode(carry);
        }
        return dummy -> next;
    }
};

运算时间有了一些提高，暂时也没想到别的优化方法。