两数之和(C++解法)
题目
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 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]
C++代码
#include <iostream>
using namespace std;
//创建链表结构
struct ListNode {
int val;
ListNode* next;
ListNode() : val(0), next(nullptr) {}
ListNode(int x) : val(x), next(nullptr) {}
};
/*
* 两数之和问题
* 用变量sum存储每一轮两个链表的值之和,carry存储进位值
* 将sum值放入新的链表中,进入下一次循环
*/
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* head = nullptr, * tail = nullptr;
int carry = 0;
while (l1 || l2) {
int n1 = l1 ? l1->val : 0;
int n2 = l2 ? l2->val : 0;
int sum = n1 + n2 + carry;
carry = sum / 10;
sum = sum % 10;
if (!head) {
head = tail = new ListNode(sum);
}
else {
tail->next = new ListNode(sum);
tail = tail->next;
}
if (l1) {
l1 = l1->next;
}
if (l2) {
l2 = l2->next;
}
}
if (carry > 0) {
tail->next = new ListNode(carry);
}
return head;
}
int main() {
ListNode* i1 = new ListNode(2);
ListNode* i2 = new ListNode(4);
ListNode* i3 = new ListNode(3);
i1->next = i2;
i2->next = i3;
i3->next = nullptr;
ListNode* l1 = i1;
ListNode* j1 = new ListNode(5);
ListNode* j2 = new ListNode(6);
ListNode* j3 = new ListNode(4);
j1->next = j2;
j2->next = j3;
j3->next = nullptr;
ListNode* l2 = j1;
ListNode* ans = addTwoNumbers(l1, l2);
while (ans) {
cout << ans->val << " ";
ans = ans->next;
}
return 0;
}
分析
两数之和问题,用变量 sum 存储每一轮两个链表的值之和,carry 存储进位值,将 sum 值放入新的链表中,进入下一次循环。