1. 程式人生 > 實用技巧 >兩數相加

兩數相加

題目

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

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

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

示例

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

連結:https://leetcode-cn.com/problems/add-two-numbers

分析

連結串列 考慮進位

程式碼

C++ 40 ms 9.3 MB

/**
 * 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* l3 = new ListNode(0);
        ListNode* p = l3;
        int c = 0,sum = 0;

        while(l1!=NULL||l2!=NULL||c!=0){
          
          
            int a,b;
            if(l1==NULL&&l2==NULL){
                a = b = 0;
            }else if(l1==NULL){
                a = 0;
                b = l2->val;
                l2 = l2 -> next;
            }else if(l2 == NULL){
                a = l1->val;
                b = 0;
                l1 = l1 -> next;
            }else{
                a = l1->val;
                b = l2->val;
                l1 = l1 -> next;
                l2 = l2 -> next;
            }
          //可以把上面這段換成 
          //int a = l1!=NULL? l1->val:0;
          //int b = l2!=NULL? l2->val:0;
          //if(l1!=NULL) l1 = l1->next;
          //if(l2!=NULL) l2 = l2->next;
          
            sum = a + b + c;
            ListNode* h = new ListNode(sum % 10);
            c = sum / 10;
            p->next = h;
            p = h;
    
        }
        return l3->next;
    }
};

python 80ms 13.7MB

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        l3 = ListNode(0)
        p = l3
        c = 0
        while(l1 or l2 or c!=0):
            a = l1.val if l1 else 0
            b = l2.val if l2 else 0

            
            p.next = ListNode((a+b+c)%10)
            c = (a+b+c)//10
            p = p.next

            if(l1!=None):
                l1 = l1.next
            if(l2!=None):
                l2 = l2.next
        return l3.next


收穫

int a = l1!=NULL? l1->val:0;
ListNode* l3 = new ListNode(0);

b = l2.val if l2 else 0