leetcode 2.两数相加

leetcode 2.两数相加

题目描述

给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外，这两个数都不会以 0 开头。

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

解题思路

该题是链表对应值相加，没有难度，但是要考虑的情况比较多
1.两个链表一样长
2.链表的长度不相等
3.进位情况
4.最后一次相加进位的情况
5.单个链表进位的情况

/**
 * 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* cur = new ListNode(-1); // 定义一个链表存储结果
        ListNode* resNum = cur;  
        bool flag = false;      // 进位标志

        ListNode* cur1 = l1;
        ListNode* cur2 = l2;
        // 为了防止两个链表不是一样长，只有两个节点都不为空的时候，才进行相加操作
        while((cur1 != NULL) && (cur2 !=NULL)){
            cur->next = new ListNode(0);
            cur = cur->next;
            // 进位的情况下，需要在两数相加的结果上加1, 同时判断当前相加是否进位
            if(flag){
                if((cur1->val + cur2->val + 1)/10 == 0){
                    flag = false;
                }
                else{
                    flag = true;
                }
                cur->val = (cur1->val + cur2->val + 1)%10;
            }
            else{   // 不进位的情况下，只要判读相加是否进位。
                if((cur1->val + cur2->val)/10 == 0){
                    flag = false;
                }
                else{
                    flag = true;
                }
                cur->val = (cur1->val + cur2->val)%10;
            }
            // 移动指针位置
            cur1 = cur1->next;
            cur2 = cur2->next;
        }
        // l1比较长的情况，这里需要考虑有多个9的情况，所以需要遍历一下
        while(cur1 != NULL){
            cur->next = cur1;
            cur = cur->next;
            if(flag){
                cur->val += 1;
                if(cur->val == 10){
                    cur->val = 0;
                    flag = true;
                }
                else{
                    flag = false;
                }
            }
            else{
                break;
            }
            cur1 = cur1->next;
        }
        // l2比较长的情况，同样需要考虑多个9的情况
        while(cur2 != NULL){
            cur->next = cur2;
            cur = cur->next;
            if(flag){
                cur->val += 1;
                if(cur->val == 10){
                    cur->val = 0;
                    flag = true;
                }
                else{
                    flag = false;
                }
            }
            else{
                break;
            }
            cur2 = cur2->next;
        }
        // 最后相加进位的情况
        if(flag){
            cur->next = new ListNode(1);
        }
        return resNum->next;
    }
};

上面是直接按照思路写的，下面是精简一下代码

/**
 * 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* head = new ListNode(-1);	 // 存储结果的链表	
        ListNode* cur = head;                // 移动节点

        bool flag = false;                  // 进位
        int sum = 0;

        while(l1 != NULL || l2 != NULL){
            sum = 0;   // sum置位
            // l1是否循环完毕，取val值
            if(l1 != NULL){
                sum += l1->val;
                l1 = l1->next;
            }
            // l2是否循环完毕，取val值
            if(l2 != NULL){
                sum += l2->val;
                l2 = l2->next;
            }
            // 进位加1
            if(flag){
                sum += 1;
            }
            
            cur->next = new ListNode(sum%10);   // 向链表中添加值
            cur = cur->next;  
            flag = (sum/10 != 0) ? true:false;  // 当前相加的结果是否进位
        }
        // 最后一位相加是否有进位情况
        if(flag){
            cur->next = new ListNode(1);
        }
        return head->next;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步