链表正向、反向相加

LeetCode - 445. Add Two Numbers II

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.
You may assume the two numbers do not contain any leading zero, except the number 0 itself.
Follow up: What if you cannot modify the input lists? In other words, reversing the lists is not allowed.

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

  就是链表的反向相加,题目里写了能不能不翻转链表,用正向相加的方式,那用两个stack存起来再算应该是比较简单好懂的方法了,AC代码如下:

ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
    stack<int> s1, s2;
    while(l1){
        s1.push(l1->val);
        l1 = l1->next;
    }
    while(l2){
        s2.push(l2->val);
        l2 = l2->next;
    }
    int sum = 0;
    bool carry = false;
    ListNode *head, *post = nullptr;
    while(!s1.empty() || !s2.empty()) {
        sum = carry ? 1 : 0;
        if(!s1.empty()) {
            sum += s1.top();
            s1.pop();
        }
        if(!s2.empty()) {
            sum += s2.top();
            s2.pop();
        }
        carry = sum >= 10;
        head = new ListNode(sum % 10);
        head->next = post;
        post = head;
    }
    if(carry) {
        head = new ListNode(1);
        head->next = post;
    }
    return head;
}

  学习起见,翻转再正向加的方法也记录一下,也就是 LeetCode - 2. Add Two Numbers206. Reverse Linked List 两道题加起来:

ListNode* reverseList(ListNode* head) {
    if(head==NULL || head->next==NULL) return head;
    ListNode *pre = head, *p = head->next, *nxt;  
    pre->next = nullptr;   // 重要!
    while(p) {  
        nxt = p->next;  
        p->next = pre;  
        pre = p;  
        p = nxt;  
    }  
    return pre;  
}
    
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
    // 翻转链表
    l1 = reverseList(l1);
    l2 = reverseList(l2);
    
    // 正向相加
    ListNode *dummyHead = new ListNode(-1);
    ListNode *cur = dummyHead;
    bool carry = false;
    while (l1 || l2) {
        int n1 = l1 ? l1->val : 0, n2 = l2 ? l2->val : 0;
        int sum = n1 + n2 + (carry ? 1 : 0);
        carry = sum >= 10;
        cur->next = new ListNode(sum % 10);
        cur = cur->next;
        if (l1) l1 = l1->next;
        if (l2) l2 = l2->next;
    }
    if (carry)
        cur->next = new ListNode(1);
    return reverseList(dummyHead->next);    // 结果还要再反转
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值