面试题 02.05. 链表求和

最新推荐文章于 2021-01-27 18:14:29 发布

vision泉

最新推荐文章于 2021-01-27 18:14:29 发布

阅读量184

点赞数

本文链接：https://blog.csdn.net/weixin_38603360/article/details/104751975

版权

面试题 02.05. 链表求和

给定两个用链表表示的整数，每个节点包含一个数位。

这些数位是反向存放的，也就是个位排在链表首部。

编写函数对这两个整数求和，并用链表形式返回结果。

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912

思路：用flag记录进位，sum = l1->val + l2->val + flag;然后将没有遍历完的l1或l2继续遍历

/**
 * 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) {
        ListNode *res = new ListNode(0);
        ListNode *cur = res;
        int flag=0;//是否进位
        while(l1 && l2){
            int sum = l1->val + l2->val + flag;
            if(sum<10){
                flag=0;
            }
            else flag=1;
            sum = sum % 10;
            cur->next = new ListNode(sum);
            cur = cur->next;
            l1 = l1->next;
            l2 = l2->next;
        }
        while(l1){
            int sum = l1->val + flag;
            if(sum<10){
                flag=0;
            }
            else flag=1;
            sum = sum % 10;
            cur->next = new ListNode(sum);
            cur = cur->next;
            l1 = l1->next;
        }
        while(l2){
            int sum = l2->val + flag;
            if(sum<10){
                flag=0;
            }
            else flag=1;
            sum = sum % 10;
            cur->next = new ListNode(sum);
            cur = cur->next;
            l2 = l2->next;
        }
        if(flag) cur->next = new ListNode(1);
        return res->next;
    }
};