【leetcode C语言实现】面试题 02.05-链表求和

题目描述

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

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

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

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912
进阶：假设这些数位是正向存放的，请再做一遍。

示例：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-lists-lcci

解题思路

当两个整数相加时，从个位开始，依次将两个数的对应位置进行相加，将所得结果的个位数作为相加后对应位置的结果，若有进位将进位的值在更高一位进行相加。此处两个数是以链表存储，同样的思路，依次将两个数的对应位置进行相加，并将所得结果保存到一个链表中。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */


struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2){
    int carry = 0, sum = 0; // 进位值, 初始化为0
    struct ListNode * head = NULL; // (struct ListNode *)malloc(sizeof(struct ListNode));
    struct ListNode * p = head;

    while(l1 || l2 || carry)
    {
        sum = 0;
        if(l1)
        {
            sum += l1->val;
            l1 = l1->next;
        }

        if(l2)
        {
            sum += l2->val;
            l2 = l2->next;
        }

        sum += carry;

        struct ListNode * newNode = (struct ListNode *) malloc(sizeof(struct ListNode));
        newNode->val = sum >= 10 ? (sum % 10) : sum;
        newNode->next = NULL;
        carry = sum / 10;

        if(head == NULL)
        {
            head = newNode;
            p = newNode;
        }
        else
        {
            p->next = newNode;
            p = p->next;
        }
    }

    return head;
}

运行结果：