2. 两数相加

题目:https://leetcode-cn.com/problems/add-two-numbers/
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。

请你将两个数相加,并以相同形式返回一个表示和的链表。

你可以假设除了数字 0 之外,这两个数都不会以 0 开头。

我的方法一:直接从头遍历,O(N),O(N)

链表的首个节点恰好是数字的最低位,所以直接从头遍历两个链表,对应位置上的值相加,如果大于等于10,下两个元素相加时记得+1,最后注意,如果最后一位相加依然大于10,那么结果的链表个数会比两个链表较长那个长度多一个元素,这个元素就是1,是进位的那个1。

边界条件

  1. 当两个链表都为空,并且进位为0时,停止

复杂度

时间:O(N),因为遍历两个链表是O(N),所以整体是O(N)
空间:O(1),注意返回值虽然自己创建了N个节点,但是返回值不记录空间复杂度。

代码

优化点,链表操作时,记得考虑是否需要dummy哑节点,这样可以简化非空判断的逻辑。

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode dummy;
        
        int carry = 0;
        ListNode* cur = &dummy;
        ListNode* tmp;
        int sum;

        //时间O(N),空间O(N)
        while(l1 || l2 || carry){
            tmp = new ListNode();
            sum = carry;
            if(l1){
                sum += l1->val;
            }
            if(l2){
                sum += l2->val;
            }

            if(sum >= 10){
                tmp->val = sum % 10;
                carry = 1;
            }else{
                tmp->val = sum;
                carry = 0;
            }

            if(l1){
                l1 = l1->next;
            }

            if(l2){
                l2 = l2->next;
            }

            cur->next = tmp;
            cur = tmp;
        }

        return dummy.next;
    }
};

你可能感兴趣的:(2. 两数相加)