力扣 两数相加 C++

#include<malloc.h>

/*
给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。

如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。

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

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

struct ListNode {
	int val;
	ListNode *next;
	ListNode(int x) : val(x), next(NULL) {}
};

ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
	ListNode *lr = NULL, *node;
	lr = (ListNode *)malloc(sizeof(ListNode));
	node = lr;
	int count = 0;
	while (l1 != NULL && l2 != NULL)
	{

		if (count >= 10)
		{
			node->val = (l1->val + l2->val + 1) % 10;
			count = l1->val + l2->val + 1;
		}
		else
		{
			node->val = (l1->val + l2->val) % 10;
			count = l1->val + l2->val;
		}
		l1 = l1->next;
		l2 = l2->next;
		node->next = (ListNode *)malloc(sizeof(ListNode));
		node = node->next;
	}
	return lr;
}

int main()
{
	ListNode *l1 = (ListNode *)malloc(sizeof(ListNode));
	ListNode *l2 = (ListNode *)malloc(sizeof(ListNode));

	l1->val = 2;
	l1->next = (ListNode *)malloc(sizeof(ListNode));
	l1->next->val = 4;
	l1->next->next = (ListNode *)malloc(sizeof(ListNode));
	l1->next->next->val = 3;
	l1->next->next->next = NULL;

	l2->val = 5;
	l2->next = (ListNode *)malloc(sizeof(ListNode));
	l2->next->val = 6;
	l2->next->next = (ListNode *)malloc(sizeof(ListNode));
	l2->next->next->val = 4;
	l2->next->next->next = NULL;

	ListNode *lr = addTwoNumbers(l1, l2);

	printf("%d ---- %d ---- %d", lr->val, lr->next->val, lr->next->next->val);
	getchar();

    return 0;
}