Tuesday, May 10, 2016

LeetCode Online Judge-2. Add Two Numbers

You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8

//C Program: Add Two Numbers
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2) {
    int val;
    struct ListNode* q = l1;
    struct ListNode* p;
    while (l1&&l2)
    {
        p = l1;
        val = l1->val + l2->val;
        l1->val = val % 10;
        if (val >= 10)
        {
            if (!l1->next)
            {
                l1->next = (struct ListNode*)malloc(sizeof(struct ListNode));
                l1->next->val = 0;
                l1->next->next = NULL;
                if (!l2->next)
                {
                    l2->next = (struct ListNode*)malloc(sizeof(struct ListNode));
                    l2->next->val = 0;
                    l2->next->next = NULL;
                }
            }
            l1->next->val += 1;
        }
        l1 = l1->next;
        l2 = l2->next;
    }
    while (l1&&l1->val>=10)
    {
        l1->val %= 10;
        if (!l1->next)
        {
            l1->next = (struct ListNode*)malloc(sizeof(struct ListNode));
            l1->next->val = 1;
            l1->next->next = NULL;
            break;
        }
        l1->next->val += 1;
        l1 = l1->next;
    }
    if (l2)
    {
        p->next = l2;
    }
    return q;
}