2. Add Two Numbers
题目链接:2. Add Two Numbers
题意
给出两条链表,链表的每个节点存储一个个位数,每条链表各表示一个倒着的整数,求表示这两个数的和的链表。
解法
为了代码简洁,可以采用冗余节点,因为如果不采用冗余节点,新链表的第一个节点必须单独创建,而且创建方式与循环体内的方法一样,产生重复代码。所以创建两个链表节点head和current,head指向新链表的头部的前一个节点,current指向当前节点。声明表示进位的变量carry,然后循环遍历两条链表,因为有可能第一条链表先结束,也可能第二条链表先结束,或者两条链表都遍历完毕,但是最后一次加法产生了一个进位,所以循环条件应该是l1、l2、carry这三者取或,因为这三个变量剩下哪一个都意味着加法还没结束。在每次循环中,先使用new创建current指向的节点的后一个节点,然后将current、l1、l2都后移一个节点(如果l1、l2都非空)。
代码
class Solution {
public:
ListNode* addTwoNumbers(ListNode *l1, ListNode *l2) {
ListNode *head = new ListNode(0), *current = head;
int carry = 0, temp;
while (l1 || l2 || carry) {
temp = (l1 ? l1->val : 0) + (l2 ? l2->val : 0) + carry;
current->next = new ListNode(temp % 10);
current = current->next;
carry = temp / 10;
l1 = (l1 ? l1->next : NULL);
l2 = (l2 ? l2->next : NULL);
}
return head->next;
}
};
算法复杂度
算法复杂度为O(m+n),其中m、n分别是两条链表的长度。