leetcode 面试题 02.05 链表求和

⭐️ 题目描述

leetcode链接：面试题 02.05 链表求和

ps： 首先定义一个头尾指针 head 、tail，这里的 tail 是方便我们尾插，每次不需要遍历找尾，由于这些数是反向存在的，所以我们直接加起来若大于等于 10 则进位，进位的数字加到下一位的数字和上，需要注意的是，当任意一个链表结束后，另一个链表可能还有余项，还需要检查是否有不为空的链表继续重复上面的操作。还有一种特殊的情况是：5 + 5 = 10 两个链表只有一个数，而且两个链表的 val 相加后是同时都为空，但是进位还有一个 1 所以还要判断这种特殊情况。

代码：

struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2){
    struct ListNode* head = NULL;   // 
    struct ListNode* tail = NULL;
    int unit = 0;
    while (l1 && l2) {
        int sum = l1->val + l2->val;
        // 尾插
        struct ListNode* newNode = (struct ListNode*)malloc(sizeof(struct ListNode));
        newNode->val = (sum % 10 + unit) % 10;
        newNode->next = NULL;
        if (head == NULL) {
            head = tail = newNode;
        } else {
            tail->next = newNode;
            tail = newNode;
        }
        // 检查
        if ((sum + unit) >= 10) {
            unit = 1;   // 进位
        } else {
            unit = 0;
        }
        // 迭代
        l1 = l1->next;
        l2 = l2->next;
    }
    // 结束可能会有一个链表还有节点
    struct ListNode* empty = l1;
    struct ListNode* noneEmpty = l2;
    if (l1 != NULL) {
        noneEmpty = l1;
        empty = l2;
    }
    // 遍历当前非空链表
    while (noneEmpty) {
        struct ListNode* newNode = (struct ListNode*)malloc(sizeof(struct ListNode));
        newNode->val = (noneEmpty->val + unit) % 10;
        newNode->next = NULL;
        tail->next = newNode;
        tail = newNode;
        // 检查
        if ((noneEmpty->val + unit) >= 10) {
            unit = 1;   // 进位
        } else {
            unit = 0;
        }
        noneEmpty = noneEmpty->next;
    }
    // 结束可能进位还存在
    if (unit) {
        struct ListNode* newNode = (struct ListNode*)malloc(sizeof(struct ListNode));
        newNode->val = unit;
        newNode->next = NULL;
        tail->next = newNode;
    }

    return head;
}

---