LeetCode 第2题 Add Two Numbers

坚持刷LeetCode，有时间每日一题，能学到很多数据结构的内容~

题目描述

  给定两个非空链表，倒序存储着两个数，链表中的每个元素为一个单独数字，假设没有前导零。求这两个数的和，并且以倒序存储的链表形式返回。

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.

分析

  这道题目并不难，其实只需要考虑三点，1、考虑两个数相加进位的情况 2、考虑两个链表不等长的情况 3、考虑两个链表最后的数字相加时还有进位的情况。我的代码如下：

/**
 * 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) {
        int sum = 0;
        //创建一个空列表
        ListNode* sum_list  = new ListNode(0);
        ListNode* head = sum_list;
        ListNode* tail = sum_list;
        //将两个链表中长度一致的部分相加
        while(l1 != NULL && l2 != NULL)
        {
            sum += l1->val+l2->val; // 计算两者的和
            //使用尾插法
            tail->next = new ListNode(sum%10);
            tail = tail->next;
            
            sum/=10;
            l1 = l1->next;  //更新l1指针
            l2 = l2->next;  //更新l2指针
        }
        
        ListNode* remain = l1==NULL?l2:l1;
        while(remain!=NULL)
        {
            sum+=remain->val;
            tail->next = new ListNode(sum%10);
            tail = tail->next;
            sum/=10;
            remain = remain->next;
            
        }
        while(sum!=0)
        {
            ListNode *new_number = new ListNode(sum%10);
            tail->next = new ListNode(sum%10);
            tail = tail->next;
            sum/=10;
        }
        return head->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* root = new ListNode(0), *first = root;
        int carry = 0;
        
        while ( l1 || l2 )
        {
            int n1 = l1 ? l1->val : 0;
            int n2 = l2 ? l2->val : 0;
            int ans = n1 + n2 + carry;
            carry = ans / 10;
            root->next = new ListNode(ans%10);
            root = root->next;
            if ( l1 ) l1 = l1->next;
            if ( l2 ) l2 = l2->next;
        }
        if (carry)
            root->next = new ListNode(1);
        
        return first->next;
        
    }
};

总结

  答案的代码有一点很好的思想，就是使用三元运算符，当指针为null时，直接将取出的值赋值为0。以后这种指针题，可以考虑多用几个指针，同时移动，可能会减少编程难度。