编程语言
首页 > 编程语言> > NC40 链表相加(二)(c++)

NC40 链表相加(二)(c++)

作者:互联网

在这里插入图片描述
在这里插入图片描述

方法一:反转链表

所以我们的入手则是对链表进行对齐,我们都是从后面开始对齐与计算的,所以很容易想到反转链表后进行相加。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

class Solution {
public:
    /**
     * 
     * @param head1 ListNode类 
     * @param head2 ListNode类 
     * @return ListNode类
     */
    ListNode* addInList(ListNode* head1, ListNode* head2) {
        // write code here
        if(head1 == nullptr){
            return head2;
        }
        if(head2 == nullptr){
            return head1;
        }
        
        head1 = reverseList(head1);//反转链表
        head2 = reverseList(head2);
        
        ListNode* head = nullptr;//保存结果
        int carry = 0;//进位
        int sum = 0;
        
        while(head1 || head2 || carry){
            sum = carry;
            if(head1){
                sum += head1->val;
                head1 = head1 -> next;
            }
            if(head2){
                sum +=head2->val;
                head2 = head2 -> next;
            }
            
            ListNode* pnext = new ListNode(sum % 10);
            //把node插入到头结点
            pnext->next = head;
            head = pnext;
          //  head = head -> next = pnext; //正序插入,需要逆转结果
            carry = sum / 10;
            
        }
        return head;
        
    }
    
    ListNode* reverseList(ListNode* head){
        if(head == nullptr){
            return head;
        }
        ListNode* cur = nullptr;
        while(head){
            ListNode* temp = head -> next;
            head->next = cur;
            cur = head;
            head = temp;
        }
        return cur;
    }
   
};

在这里插入图片描述

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

class Solution {
public:
    /**
     * 
     * @param head1 ListNode类 
     * @param head2 ListNode类 
     * @return ListNode类
     */
    ListNode* addInList(ListNode* head1, ListNode* head2) {
        // write code here
        if(head1 == nullptr){
            return head2;
        }
        if(head2 == nullptr){
            return head1;
        }
        
        stack<ListNode*>stk1;
        stack<ListNode*>stk2;
        
        //把head1,head2节点存入stack中
        ListNode*p1 = head1;
        ListNode*p2 = head2;
        while(p1){
            stk1.push(p1);
            p1 = p1 -> next;
        }
        while(p2){
            stk2.push(p2);
            p2 = p2 -> next;
        }
        ListNode* head = nullptr;//保存结果
        int carry = 0;//进位
        int sum = 0;
        while(!stk1.empty() || !stk2.empty() || carry){
            sum = carry;
            if(!stk1.empty()){//加上栈顶元素
                sum += stk1.top()->val;
                stk1.pop();
            }
            if(!stk2.empty()){
                sum += stk2.top()->val;
                stk2.pop();
            }
            
            ListNode* pnext = new ListNode(sum % 10);
            //把node插入到头结点
            pnext->next = head;
            head = pnext;
          //  head = head -> next = pnext;
            carry = sum / 10;
            
        }
        return head;
        
    }
};

标签:head,ListNode,head2,head1,sum,c++,next,链表,NC40
来源: https://blog.csdn.net/qq_36421001/article/details/123419767