Leetcode（鏈表）兩數相加(c++)

給出兩個 非空 的鏈表用來表示兩個非負的整數。其中，它們各自的位數是按照 逆序 的方式存儲的，並且它們的每個節點只能存儲 一位 數字。

如果，我們將這兩個數相加起來，則會返回一個新的鏈表來表示它們的和。

您可以假設除了數字 0 之外，這兩個數都不會以 0 開頭。

示例：

輸入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出：7 -> 0 -> 8
原因：342 + 465 = 807

這題的測試數據邊界條件特別多，要一一考慮。比如[5] [5], [9,8][1], [1][9,9,9]等。
添加兩個bool函數，用於判斷。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    //判斷l1是否大於l2
    bool if_morethan(ListNode* l1, ListNode* l2){
        while(l1 !=NULL && l2 !=NULL){
            l1=l1->next;
            l2=l2->next;
        }
        if(l1 !=NULL) return true;
        else return false;
    }
    //判斷l1==l2
    bool if_equal(ListNode* l1, ListNode* l2){
        while(l1 !=NULL && l2 !=NULL){
            l1=l1->next;
            l2=l2->next;
        }
        if(l1==NULL && l2==NULL) return true;
        else return false;
    }
  
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        // 判斷是否存在[0]數組，存在則直接返回另一個鏈表即可
        if(l1->val==0 && l1->next==NULL)
            return l2;
        if(l2->val==0 && l2->next==NULL )
            return l1;
        //創建一個新結點和兩個結點指針
        ListNode* new_list=new ListNode(0);
        ListNode* temp;
        ListNode* head;
        //將temp和head定義爲鏈表長度較長的那個鏈表的頭結點
        if(if_morethan(l1,l2))  //l1長度大於l2
        {
            new_list->next=l1;
            temp=l1;head=l1;
        }
        else {
            new_list->next=l2;
            temp=l2;head=l2;
        }
        //創建一個flag，用於判斷l1,l2的長度是否相等
        bool flag=if_equal(l1, l2);
        //開始進入迭代相加的循環
        int num=0;
        while(l1 !=NULL && l2 !=NULL){
            if((l1->val+l2->val+num) >9){
                temp->val=((l1->val + l2->val + num) %10);
                num=1;
            }
            else{
                temp->val=((l1->val + l2->val + num) %10 ) ;
                num=0;
            }    
            temp=temp->next;
            l1=l1->next;
            l2=l2->next;
        }
        //創建一個val=1的結點
        ListNode* new_num=new ListNode(1);
        //如果num==1表示循環完了，但還要進位。如[5][5]
        if(num==1 && flag)//既要進位，且兩個鏈表等長，要在末尾插入一個結點
        {
            while(head->next!=NULL){
                head=head->next;
            }
            head->next=new_num;
        }
        //既要進位，且兩個鏈表不等長，如[1][9,9,9]
        if(num==1 && !flag)
        {
            temp->val +=1;
            while(temp->val==10){
                temp->val=0;
                if(temp->next==NULL)
                    temp->next=new_num;
                else{
                    temp->next->val +=1;
                }
                temp=temp->next;
            }
            
        }
        return new_list->next;
    }
};

第一遍寫的非常複雜，因爲在不斷完善測試數據的邊界條件。

參考了網友的解答，第二遍的while循環改成（l1 || l2）

class Solution {
public:
     ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode *result = new ListNode(0);
        ListNode *tmp = result;
        int sum = 0;
        while(l1 || l2){      
            if(l1){
                sum += l1->val;
                l1 = l1->next;
            }
            if(l2){
                sum += l2->val;
                l2 = l2->next;
            }
            tmp->next = new ListNode(sum%10);
            sum /= 10;
            tmp = tmp->next;
        }
        if(sum)
            tmp->next = new ListNode(1);
        return result->next;
    }


};