LeetCode 之路

LeetCode 2——两数相加

2018-10-28  本文已影响4人  seniusen

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」!


上一篇下一篇

猜你喜欢

热点阅读