LeetCode题解[Add Two Numbers II]

本文首发于个人blog中.node

今天的题目是Add Two Numbers II:c++

题目描述:git

You are given two non-empty linked lists representing two non-negative integers. The most significant digit comes first and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.github

You may assume the two numbers do not contain any leading zero, except the number 0 itself.code

Follow up:
What if you cannot modify the input lists? In other words, reversing the lists is not allowed.blog

Example:递归

Input: (7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 8 -> 0 -> 7ip

求解思路:leetcode

这道题最简练的方法就是用栈了作了,而后还能够先计算出两个链表的长度,再生成链表。get

思路都比较简单,emmm,一开始觉得作成递归的会比较简单,后来发现不只时间复杂度仍是代码的复杂度都挺高的,作成迭代的会好不少。

/**
 * 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) {
        
        stack<int> s1, s2;
        int carry = 0, v1, v2, v;
        ListNode *cur;
        ListNode *pre = nullptr;
        
        while(l1 || l2) {
            if (l1) { s1.push(l1->val); l1 = l1->next; }
            if (l2) { s2.push(l2->val); l2 = l2->next; }
        }
        
       
        while(!s1.empty() || !s2.empty() || carry) {
            v1 = v2 = 0;
            if (!s1.empty()) { v1 = s1.top(); s1.pop(); }
            if (!s2.empty()) { v2 = s2.top(); s2.pop(); }
            
            v = v1 + v2 + carry;
            cur  = new ListNode(v % 10);
            cur->next = pre;
            pre = cur;
            carry = v /10;
        }      
        return pre;
    }
};

给出一个dicuss中的方法,很巧妙,用的是翻转链表(题目说不能修改原来的链表,因此这里是翻转output):

ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
    int n1 = 0, n2 = 0, carry = 0;
    ListNode *curr1 = l1, *curr2 = l2, *res = NULL;
    while( curr1 ){ curr1=curr1->next; n1++; }
    while( curr2 ){ curr2=curr2->next; n2++; } 
    curr1 = l1; curr2 = l2;
    while( n1 > 0 && n2 > 0){
        int sum = 0;
        if( n1 >= n2 ){ sum += curr1->val; curr1=curr1->next; n1--;}
        if( n2 > n1 ){ sum += curr2->val; curr2=curr2->next; n2--;}
        res = addToFront( sum, res );
    }
    curr1 = res; res = NULL;
    while( curr1 ){
        curr1->val += carry; carry = curr1->val/10;
        res = addToFront( curr1->val%10, res );
        curr2 = curr1; 
        curr1 = curr1->next;
        delete curr2;
    }
    if( carry ) res = addToFront( 1, res );
    return res;
}

ListNode* addToFront( int val, ListNode* head ){
    ListNode* temp = new ListNode(val);
    temp->next = head;
    return temp;
}
相关文章
相关标签/搜索