LeetCode刷题C++算法第二关:
题目:
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.
这道题涉及到C++的链表,算法逻辑很清晰,就是建立一个新链表,然后把输入的两个链表按照从头往后的顺序,每两个相加,然后添加一个新节点到新链表后面。为了避免两个输入的链表同时为空,我们建立一个dummy结点(也就是传说中的哑结点),将两个结点相加生成的新结点按顺序加到dummy结点之后,由于dummy结点本身不能变,所以我们用一个指针curr来指向新链表的最后一个结点。
由于本题目的要求为最低位在链表的开头,所以我们可以在遍历链表的同时按从低到高的顺序直接相加。while循环的条件两个链表中只要有一个不为空行,由于链表可能为空,所以我们在取当前结点值的时候,先判断一下,若为空则取0,否则取结点值。然后把两个结点值相加,同时还要加上进位carry,用来防止溢出(也就是逢十进一)。然后更新carry,直接用 sum/10 即可,然后以 sum%10 为值建立一个新结点,连到curr后面,然后curr移动到下一个结点。之后再更新两个结点,若存在,则指向下一个位置。while循环退出之后,最高位的进位问题要最后特殊处理一下,若carry为1,则再建一个值为1的结点,代码如下:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* dummy = new ListNode(-1);
ListNode* curr = dummy;
int carry = 0;
while(l1 || l2) {
int val1 = l1 ? l1->val : 0;
int val2 = l2 ? l2->val : 0;
int sum = val1 + val2 + carry;
carry = sum / 10;
curr->next = new ListNode(sum % 10);
curr = curr->next;
if(l1) l1 = l1->next;
if(l2) l2 = l2->next;
}
if(carry != 0) {
curr->next = new ListNode(1);
}
return dummy->next;
}
提交的结果如下:
当然,你也可以不直接使用链表指针l1和l2,代码如下:
ListNode* addTwoNumbers_1(ListNode* l1, ListNode* l2) {
ListNode* dummy = new ListNode(0);
ListNode* p = l1, *q = l2, *curr = dummy;
int carry = 0;
while(p || q ) {
int x = p ? p->val : 0;
int y = q ? q->val : 0;
int sum = x + y + carry;
carry = sum / 10;
curr->next = new ListNode(sum % 10);
curr = curr->next;
if(p) p = p->next;
if(q) q = q->next;
}
if(carry) {
curr->next = new ListNode(carry);
}
dummy = dummy->next;
return dummy;
}
提交结果如下:
你或许有注意到,两段代码的思路和结构甚至编写方式几乎是一样的,硬要说有区别的地方可能是在最后return的地方,由于dummy所在链表的头结点存储的值为0(或者-1, 也可以是其他的数,这个数自己定义就行),所以dummy中会有四个数(说法不专业),我们可以直接用dummy->next来跳过头结点,也可以先跳过头结点,从第二个结点开始输出,详情见上述代码。