3.leetcode-两数相加

Posted on 2021-04-03 11:27  陈夫子  阅读(50)  评论(0)    收藏  举报

https://leetcode-cn.com/problems/add-two-numbers/

给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。

请你将两个数相加,并以相同形式返回一个表示和的链表。

你可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例 1:

输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.
示例 2:

输入:l1 = [0], l2 = [0]
输出:[0]
示例 3:

输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出:[8,9,9,9,0,0,0,1]

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


struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2){
    struct ListNode *phead = NULL;
    struct ListNode *pnext = NULL;
    struct ListNode *pnode = NULL;
    int carry = 0;

    phead = (struct ListNode *)malloc(sizeof(struct ListNode));
    if (NULL == phead)
        return NULL;
    memset(phead, 0, sizeof(struct ListNode));
    pnext = phead;

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

        pnode = (struct ListNode *)malloc(sizeof(struct ListNode));
        if (NULL == pnode)
            return NULL;
        memset(pnode, 0, sizeof(struct ListNode));

        pnode->val = sum % 10;
        pnext->next = pnode;
        pnext = pnext->next;

        carry = sum / 10;         
        if (sum > 9) {
            if (NULL == l1 && NULL == l2) {
                pnode = (struct ListNode *)malloc(sizeof(struct ListNode));
                if (NULL == pnode)
                    return NULL;
                memset(pnode, 0, sizeof(struct ListNode));
                pnode->val = carry;
                pnext->next = pnode;
            }
        }
    } 
    return phead->next;
}