1. 程式人生 > >LeetCode 2. 兩數相加(C++)

LeetCode 2. 兩數相加(C++)

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

如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。

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

示例:

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

C++

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    int getlen(ListNode* head)
    {
        int count=0;
        while(head)
        {
            count++;
            head=head->next;
        }
        return count;
    }
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) 
    {
        int count1=getlen(l1);
        int count2=getlen(l2);
        ListNode* large;
        ListNode* small;
        if(count1>count2)
        {
            large=l1;
            small=l2;
        }
        else
        {
            large=l2;
            small=l1;
        }
        ListNode* p;
        int add=0;
        int num=large->val+small->val+add;
        if(num<10)
        {
            p=new ListNode(num);
            add=0;
        }
        else
        {
            p=new ListNode(num%10);
            add=num/10;
        }
        large=large->next;
        small=small->next;
        ListNode* head=p;
        while(small)
        {
            num=large->val+small->val+add;
            if(num<10)
            {
                p->next=new ListNode(num);
                add=0;
            }
            else
            {
                p->next=new ListNode(num%10);
                add=num/10;
            }
            p=p->next;
            small=small->next;
            large=large->next;
        }
        while(large)
        {
            num=large->val+add;
            if(num<10)
            {
                p->next=new ListNode(num);
                add=0;
            }
            else
            {
                p->next=new ListNode(num%10);
                add=num/10;
            }
            p=p->next;
            large=large->next;
        }
        if(add)
        {
            p->next=new ListNode(add);
        }
        return head;        
    }
};