题目:https://leetcode-cn.com/problems/add-two-numbers/
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
我的方法一:直接从头遍历,O(N),O(N)
链表的首个节点恰好是数字的最低位,所以直接从头遍历两个链表,对应位置上的值相加,如果大于等于10,下两个元素相加时记得+1,最后注意,如果最后一位相加依然大于10,那么结果的链表个数会比两个链表较长那个长度多一个元素,这个元素就是1,是进位的那个1。
边界条件
- 当两个链表都为空,并且进位为0时,停止
复杂度
时间:O(N),因为遍历两个链表是O(N),所以整体是O(N)
空间:O(1),注意返回值虽然自己创建了N个节点,但是返回值不记录空间复杂度。
代码
优化点,链表操作时,记得考虑是否需要dummy哑节点,这样可以简化非空判断的逻辑。
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode dummy;
int carry = 0;
ListNode* cur = &dummy;
ListNode* tmp;
int sum;
//时间O(N),空间O(N)
while(l1 || l2 || carry){
tmp = new ListNode();
sum = carry;
if(l1){
sum += l1->val;
}
if(l2){
sum += l2->val;
}
if(sum >= 10){
tmp->val = sum % 10;
carry = 1;
}else{
tmp->val = sum;
carry = 0;
}
if(l1){
l1 = l1->next;
}
if(l2){
l2 = l2->next;
}
cur->next = tmp;
cur = tmp;
}
return dummy.next;
}
};