问题描述:
You are given two linked lists representing two non-negative numbers. 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.
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
答案之一:
class Solution {
public:
ListNode *addTwoNumbers(ListNode *l1, ListNode *l2)
{
ListNode stackAnchor(0);
ListNode* tail = &stackAnchor;
div_t sum = { 0, 0 };
while(sum.quot > 0 || l1 || l2)
{
if (l1)
{
sum.quot += l1->val;
l1 = l1->next;
}
if (l2)
{
sum.quot += l2->val;
l2 = l2->next;
}
sum = div(sum.quot, 10);
tail->next = new ListNode(sum.rem);
tail = tail->next;
}
return stackAnchor.next;
}
};
div_t是一个结构,div返回div_t同时包含商与余数
typedef struct{
int quot;
int rem;
}div_t;
用多了vector,总想着将ListNode转换为vector,因为vector的size是知道的,总是写for循环,不知道循环的次数就不知如何下笔,写到后面ListNode想vector转换就糊涂了,此处用while(l1 || l2 || sum.quot>0)就可以了控制循环的结束了。用div_t就将进位和余数存储,作者提到考虑到效率的问题,希望将商和余数同时得到,我最初的想法是比较两个数之和与10的大小,用bool值标志是否要进位。
注意返回的结果是stackAnchor.next;我开始一直想让返回值是stackAnchor,磨叽了半天,实现不了。
还有一个超简洁版结果:
ListNode *addTwoNumbers(ListNode *l1, ListNode *l2) {
ListNode preHead(0), *p = &preHead;
int extra = 0;
while (l1 || l2 || extra) {
int sum = (l1 ? l1->val : 0) + (l2 ? l2->val : 0) + extra;
extra = sum / 10;
p->next = new ListNode(sum % 10);
p = p->next;
l1 = l1 ? l1->next : l1;
l2 = l2 ? l2->next : l2;
}
return preHead.next;
}
前一种方法在if(l1)中完成了求和与l1的更新,这种方法用? :操作符,不能一次实现,所以l1是否为NULL判断了两次。
extra和sum可以再精简一点,只留下sum即可
ListNode *addTwoNumbers(ListNode *l1, ListNode *l2) {
ListNode preHead(0), *p = &preHead;
int sum = 0;
while (l1 || l2 || extra) {
sum = (l1 ? l1->val : 0) + (l2 ? l2->val : 0) + sum;
p->next = new ListNode(sum % 10);
p = p->next;
sum=sum/10;
l1 = l1 ? l1->next : l1;
l2 = l2 ? l2->next : l2;
}
return preHead.next;
}
【总结】
1. 熟练使用while循环,不要总想着转换为vector,for循环
2.return preHead.next;
3. 用div_t结构和div运算,一步得到商和余数
4.多个数相加,或者结果保留为其他进制的形式都可以在这个程序的基础上修改得到
5. ListNode这种数据结构可以借鉴,因为加法或者减法乘法都是由低位向高位计算的,所以低位放在根节点