LeetCode 2. 兩數相加(C++)
阿新 • • 發佈:2018-12-16
給出兩個 非空 的連結串列用來表示兩個非負的整數。其中,它們各自的位數是按照 逆序 的方式儲存的,並且它們的每個節點只能儲存 一位 數字。
如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。
您可以假設除了數字 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; } };