[Leetcode] 002 Add Two Numbers

You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8

Solution
思路并不难想,从头节点开始,逐个节点相加,并保存进位,计算下一个节点时计算进位。
不过这一题需要考虑较全面,很容易漏掉处理一些边界条件,这题提交了5次才通过。
对于一些看似简单的题目,一定要更加小心,不要阴沟翻船。
下面代码中的测试Case就是教训。

Code
common头文件的代码比较简单,就不贴上了。

#include <iostream>                                                                                                                                           

#include "../common/list/list.h"

class Solution
{
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) 
    {   
        if (l1 == NULL && l2 == NULL)
        {
            return new ListNode(0);
        }
        if (l1 == NULL)
        {
            return l2; 
        }
        if (l2 == NULL)
        {
            return l1; 
        }
        ListNode* result = NULL;
        ListNode* pL1 = l1; 
        ListNode* pL2 = l2; 
        ListNode* pResult = NULL;
        int flag = 0;
        while (pL1 != NULL && pL2 != NULL)
        {
            int curNodeVal = (pL1->val + pL2->val + flag) % 10; 
            flag = (pL1->val + pL2->val + flag) / 10; 
            ListNode* curNode = new ListNode(curNodeVal);
            if (result == NULL)
            {
                result = curNode;
                pResult = curNode;
            }
            else
            {
                pResult->next = curNode;
                pResult = curNode;
            }
            pL1 = pL1->next;
            pL2 = pL2->next;
        }
        while (pL1 != NULL)
        {
            int curNodeVal = (pL1->val + flag) % 10; 
            flag = (pL1->val + flag) / 10; 
            ListNode* curNode = new ListNode(curNodeVal);
            pResult->next = curNode;
            pResult = pResult->next;
            pL1 = pL1->next;
        }
        while (pL2 != NULL)
        {
            int curNodeVal = (pL2->val + flag) % 10;
            flag = (pL2->val + flag) / 10;
            ListNode* curNode = new ListNode(curNodeVal);
            pResult->next = curNode;
            pResult = pResult->next;
            pL2 = pL2->next;
        }
        if (flag)
        {
            ListNode* curNode = new ListNode(flag);
            pResult->next = curNode;
        }
        return result;
    }
};

int main()
{
    /* Case: 1
    ListNode* l1 = new ListNode(2,
                                new ListNode(4,
                                             new ListNode(3)));
    ListNode* l2 = new ListNode(5,
                                new ListNode(6,
                                             new ListNode(4)));
    */

    /* Case: 2
    ListNode* l1 = new ListNode(5);
    ListNode* l2 = new ListNode(5);
    */

    /* Case: 3
    ListNode* l1 = new ListNode(9,
                                new ListNode(9));
    ListNode* l2 = new ListNode(1);
    */

    /* Case: 4
    ListNode* l1 = new ListNode(0);
    ListNode* l2 = new ListNode(7,
                                new ListNode(8));
    */

    /* Case 5 */
    ListNode* l1 = new ListNode(1);
    ListNode* l2 = new ListNode(9,
                                new ListNode(9));
    Solution solution;
    ListNode* result = solution.addTwoNumbers(l1, l2);
    PrintList(result);
    return 0;
}
    原文作者:周肃
    原文地址: https://www.jianshu.com/p/8e6f82cfd2e2
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞