这是一道关于链表的题
首先来看一下题目:
给定两个非空链表分别代表两个非负的整数,数字是按照逆序存储(链表第一个节点表示数的个位)并且链表中的每一个节点代表一位数字,现将两个数字相加并返回使用链表表示的结果。
思路:
这道题的思路是非常直接的,将两个链表中的对应节点一次相加即可,有两个需要注意的点:
- 进位:当两个节点之和大于等于10的时候,其下一位的计算结果要加1;
- 末尾(最高位)需要进位:这个时候就要新添加一个节点,其值为1;
怎么样,是不是很简单呢~
不过链表的题嘛,总是需要注意一些特殊情况:
- 如果输入的一个链表为空,程序会不会崩溃?
- 如果两个链表都为空呢?
那么实现的细节有哪些需要注意的地方呢,怎样才能让代码变得简洁,我们来看一下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2)
{
//这是一个头节点,我们要返回的第一个节点在它之后
ListNode res = ListNode(0);
ListNode *curr = &res;
//这个变量表示进位
int carry = 0;
//开始遍历链表
//遍历直到两个链表都达到末尾才结束
while(l1 != nullptr || l2 != nullptr)
{
//如果其中某一个链表先达到了末尾,则将其值直接置为0
//否则,就取出链表节点的值
int n1 = l1 ? l1->val : 0;
int n2 = l2 ? l2->val : 0;
//相加结果,别忘了加上进位
int number = n1 + n2 + carry;
//计算进位
carry = number / 10;
//计算结果若超过10则取余
number %= 10;
//添加新的节点用来存储计算得到的结果
curr->next = new ListNode(number);
//递进链表指示器
curr = curr->next;
l1 = l1 ? l1->next : nullptr;
l2 = l2 ? l2->next : nullptr;
}
//别忘了如果最后还有进位,要添加一个新的值为1的节点
if(carry)
curr->next = new ListNode(1);
//返回头节点的后一个节点
//注意不要返回头节点,我们的第一个计算结果是从头节点之后开始的
return res.next;
}
};
下图是运行结果: