[LeetCode] 2.Add Two Numbers

题目描述

题目大致的意思是给输入是两个非负的数字，数字由链表存储，一个节点存储一个数，比如342存储为2 -> 4 -> 3。
在题目中所给的测试样例

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

思路

这道题和在国内经常做的大数加法由异曲同工之妙，我记得本科那会的竞赛书上会出这样的一道题，只不过那道题容器是数组，这道题容器是链表。
考虑到容器是链表，就需要考虑链表是否为空，以及next指针的操作。比如两个数字位数不相同情况的处理，以及进位的处理。

链表循环

设两个数字的链表为L1和L2，在进行循环加法操作时，只要L1和L2任意一位不为空，循环则需继续。

while(l1 || l2){
……
l1 = l1? l1->next : NULL;
l2 = l2? l2->next : NULL;
}

加数处理

这里有若干个小细节，下面来分别解析。我们遇到的情况有以下三种:

1. L1、L2都不为空
2. L1为空、L2不为空
3. L1不为空、L2为空

最“笨“的方法其实可以用三个if来实现三种情况，但是我们可以想着将这三种情况合成一种情况处理，为空的情况就是为零，为空的话就当0做加法。

int x1 = l1 ? l1->val : 0;
int x2 = l2 ? l2->val : 0;

求和处理

这里就考虑到前面一位的进位，初始化时赋值为0，每次求和运算算出是否有进位

int carry = 0;
int temp = x1 + x2 + carry;
carry = temp / 10;
int val = temp % 10;

建立和的链表

我们需要建立两个链表的指针，一个来存储头节点的地址，一个来进行新节点的生成。在这里需要重新温习一下结构体的初始化。

ListNode* head = NULL;//头节点
ListNode* p =NULL;//实时更新的节点
ListNode* newPointer = new ListNode(val);
//C++中Struct可以通过new来初始化

对于一开始，需要将newPointer赋值于head和p，之后的循环只对p进行操作即可。

if(!head){
    head = newPointer;
}
if(p){//表示p不是第一次赋值
    p->next = newPointer;
}
p = newPointer;

最后的进位

两个数的和最后可能会比两个数的位数多处一位(考虑加和进1)，所以要单独将最后的carry生成一个新的节点。

if(carry){
    ListNode* newPointer = new ListNode(carry);
    p->next = newPointer;
}

源码

/**
 * 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 = NULL;
        ListNode* p = NULL;
        int carry = 0;
        while(l1 || l2){
            int x1 = l1 ? l1->val : 0;
            int x2 = l2 ? l2->val : 0;
            int temp = x1 + x2 + carry;
            int val = temp % 10;
            carry = temp / 10;
            ListNode* newPointer = new ListNode(val);
            if(!head){
                head = newPointer;
            }
            if(p){
                p->next = newPointer;
            }
            p = newPointer;
            l1 = l1 ? l1->next : NULL;
            l2 = l2 ? l2->next : NULL;
        }
        if(carry){
            ListNode* newPointer = new ListNode(carry);
            p->next = newPointer;
        }
        return head;
    }
};