LeetCode-445|两数相加(2)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jmu201521121021/article/details/83756816

LeetCode-445

  • 题目:
    给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。

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

示例:

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

  • 解法
    通过两个栈来存放两个链表,最终栈顶为数的低位,栈顶为栈的高位,然后两个栈元素对应相加。最后注意是否有进位。
  • 源码
/**
 * 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;  // 栈
        while(l1 != NULL){ 
            s1.push(l1->val);
            l1 = l1->next;
        }
         while(l2 != NULL){
            s2.push(l2->val);
            l2 = l2->next;
        }
        int sum = 0;
        ListNode* res = new ListNode(0);  // 初始化头结点
        while(!s1.empty() || !s2.empty()){ 
           if(!s1.empty()){
               sum += s1.top();
               s1.pop();
           }
           if(!s2.empty()){
               sum += s2.top();
               s2.pop();
           }
           res->val = sum%10;     //  采用头插法
           sum /= 10;
           ListNode* head = new ListNode(sum);
           head->next = res;
           res = head;
           
        }
        if(!res -> val){     // 判断是否最高位为0
              res = res->next;
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/jmu201521121021/article/details/83756816