题意
基于add two number的将2个链表相加,但是要求:不能改变输入的链表
思路
这道题有很多思路啊,如果能够多开空间的话,直接用一个数组或者栈保存一下链表然后相加即可。
如果要求 O(1) 的空间,有一个很tricky的思路就是:将两个输入链表从高位往低位相加,然后翻转得到的结果链表,并且进行进位即可。
然后还有一个思路就是:要解决的问题就是如何将两个数从高位往低位相加?
首先,我们假设两个链表长度相同,链表长度不同的话很好处理,将长的那部分拿出来即可。
然后关键问题就是进位,如果用pre指向当前加的节点的前一个节点的话,假设当前节点的值大于10,那么前一个节点的值就需要+1,但是前一个节点如果进位后等于10,那么需要继续向前进位,问题就出在我们这时候不能获得再前一个节点了,于是不能再进行进位了。
我们换一个思路,假设如果需要向前进位的话,我们能够知道进位到的最高位的话,那么是否好处理呢?
比如 x+y>10 ,这时候我们需要向前进位,前1位继续向前进位的条件是:当前仅当当前位是9。
那么,假设我们现在的结果是:a9999x,其中x是我们处理的当前节点,a是第一个不为9的数,那么,此时进位后的结果是:a + 1, 0, 0, 0, 0, x % 10。
于是,我们用combo来记录连续的9前面的第一个不为9的节点,具体过程见代码。
代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
int getLength(ListNode *p) {;
int len = 0;
while (p) {
++len;
p = p->next;
}
return len;
}
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
if (!l1 && !l2) return NULL;
int len1 = getLength(l1), len2 = getLength(l2);
if (len1 < len2) swap(l1, l2);
int diff = abs(len1 - len2);
ListNode* vhead = new ListNode(0);
ListNode* cur = vhead, *combo = vhead;
while (diff--) {
cur->next = new ListNode(l1->val);
cur = cur->next;
if (cur->val != 9) combo = cur;
l1 = l1->next;
}
while (l1) {
int val = l1->val + l2->val;
if (val > 9) {
val %= 10;
combo->val++;
combo = combo->next;
while (combo) {
combo->val = 0;
combo = combo->next;
}
}
cur->next = new ListNode(val);
cur = cur->next;
if (cur->val != 9) combo = cur;
l1 = l1->next, l2 = l2->next;
}
ListNode* head = vhead;
if (!vhead->val) {
head = vhead->next;
delete(vhead);
}
return head;
}
};