LeetCode 2——两数相加

1. 题目

2. 解答

循环遍历两个链表

• 若两个链表都非空，将两个链表结点的值和进位相加求出和以及新的进位

• 若其中一个链表为空，则将另一个链表结点的值和进位相加求出和以及新的进位

然后将每一位的和添加到新链表中。

如果有一个链表为空，且此时进位为 0，我们则只需要将非空链表后面的值复制到新链表即可，可以通过将非空链表的剩余结点直接接在新链表后面来实现。

如果最后两个链表同时为空，我们还要考虑此时是否有进位，若有进位，进位即为最后一位的和，然后，让链表尾结点指向 NULL 即可。

/**
 * 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(0); // 建立哨兵结点
        ListNode *temp = head; 

        int carry = 0; // 保留进位
        int sum = 0;

        while(l1 || l2)
        {
            if (l1 && l2) // 两个链表都非空
            {
                sum = l1->val + l2->val + carry;
                l1 = l1->next;
                l2 = l2->next;
            }
            else if (l1) // l2 链表为空，只对进位和 l1 元素求和
            {
                sum = l1->val + carry;
                l1 = l1->next;
            }
            else // l1 链表为空，只对进位和 l2 元素求和
            {
                sum = l2->val + carry;
                l2 = l2->next;
            } 

            // 求出和以及进位，将和添加到新链表中
            carry = sum >= 10 ? 1 : 0;
            sum = sum % 10;
            head->next = new ListNode(sum);
            head = head->next;

            if ( (l1 == NULL || l2 == NULL) && carry == 0 )
            {
                head->next = l1 ? l1 : l2;  // 将非空链表剩余结点接在新链表后面
                return temp->next;
            }

        }

        if (carry) // 若最后一位还有进位，进位即为最后一位的和
        {
            head->next = new ListNode(carry);
        }
        head->next->next = NULL;

        return temp->next;

    }
};

获取更多精彩，请关注「seniusen」!