原题链接:Add Two Numbers
题目原文:
You are given two non-empty linked lists representing two non-negative integers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.
You may assume the two numbers do not contain any leading zero, except the number 0 itself.
Example:
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.
|ू・ω・` )
这是一个相对简单的两数和,两个加数以链表形式反向存储,要求输出同样是一个链表的形式,并消除高位占位零。用到链表的基本知识,写详细一点就当复习了 φ(..)
遍历时间复杂度O(n),(并不是前50%,有更快的算法欢迎评论区交流。)每次相加需要记录两个数字:模十之后的和数sum、进位carry。若两个加数对应的每一位数字分别记为a、b,用两个指针P1和P2指向它们,那么
a = ( P1 == NULL) ? 0 : P1 -> val; b = (P2 == NULL) ? 0 : P2 -> val;
sum = (a + b)%10; carry = (a + b) / 10;
在执行相加的循环体开始之前,需要申请一个新的链表空间作为结果以返回给主函数
struct ListNode *ans = (struct ListNode*)malloc(sizeof(struct ListNode));
ans->val = 0;
在循环体内另用一个指针P代替ans进行扩展以防返回ans的末尾。循环内部若一个加数所有位都已完成相加但是另一个加数还没有结束,就把已结束的加数指针设为NULL,val置0。因此无需考虑加数长度。
下面列出了几个导致TLE的小bug:
1.sum与carry的计算公式有没有写对;
2.每一次相加对P应该新生成新节点,而不是修改原始节点。
3.特殊基本问题:循环的结束条件。
完整代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2) {
int carry = 0;
struct ListNode *p1, *p2, *p;
struct ListNode *ans = (struct ListNode*)malloc(sizeof(struct ListNode));
ans->val = 0;
p1 = l1; p2 = l2;
p = NULL;
while (p1 != NULL || p2 != NULL || carry != 0) {
int a, b, sum;
if (p == NULL) p = ans;
else {
p->next = (struct ListNode*)malloc(sizeof(struct ListNode));
p->next->val = 0;
p = p->next;
}
a = (p1 == NULL ? 0 : p1->val);
b = (p2 == NULL ? 0 : p2->val);
sum = (a + b + carry) % 10;
carry = (a + b + carry) / 10;
p->val = sum;
p->next = NULL;
p1 = (p1 == NULL ? NULL : p1->next);
p2 = (p2 == NULL ? NULL : p2->next);
}
return ans;
}