LeetCode:2. 两数相加

题目

LeetCode:Add Two Numbers

给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。

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

示例:

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

思路

最初考虑到可以将链表的数先转换成对应的数,相加后再求每个位,结果测试用例来了个超过整型的数,遂改成long long,结果又有一个测试用例有个超过long long范围的,说明行不通。

正确方法是直接让每个数相同的位直接相加,当做链表一个节点,比如示例,直接2和5相加,4和6相加,3和4相加,但是有一个问题,这么求出来是7->10->7,很明显是错误的。

第一个需要考虑的问题为进位,如上面所述。

第二个需要考虑的问题为循环结束条件,任意一个链表不为空都需要往后走,同时当最后一个数相加为10的时候,最后两个节点应为->0->1,因此结束条件是两链表同时指向结尾空指针,且不需要进位。

代码

查看更多LeetCode代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2) {
    int sum = 0, carry = 0;
    struct ListNode* ret = NULL;
    struct ListNode* cur = NULL;
    struct ListNode* new_node = NULL;
    while ((l1 != NULL || l2 != NULL) || carry == 1)
    {
        sum = 0;
        if (l1 != NULL)
        {
            sum += l1->val;
            l1 = l1->next;
        }
        if (l2 != NULL)
        {
            sum += l2->val;
            l2 = l2->next;
        }
        if (carry != 0)
        {
            sum += 1;
            carry = 0;
        }
        if (sum >= 10)
        {
            carry = 1;
        }
        new_node = (struct ListNode*)malloc(sizeof(struct ListNode));
        new_node->val = sum % 10;
        new_node->next = NULL;
        if (ret == NULL)
        {
            ret = new_node;
            cur = new_node;
        }
        else
        {
            cur->next = new_node;
            cur = cur->next;
        }
    }
    return ret;
}

你可能感兴趣的:(LeetCode)