首先是题目链接
假设链表中每一个节点的值都在 0 - 9 之间,那么链表整体就可以代表一个整数。
给定两个这种链表,请生成代表两个整数相加值的结果链表。
例如:链表 1 为 9->3->7,链表 2 为 6->3,最后生成新的结果链表为 1->0->0->0。
反转链表:代码首先使用两个while循环分别反转两个输入链表head1和head2。这是通过迭代链表并逐个节点反转其next指针实现的。
创建结果链表的头节点:
ListNode* head = new ListNode(-1); ListNode* res = head;
这里创建了一个哑节点head,其值为-1(这个值不会被使用),以及一个指向结果链表的指针res。
进位标志:int c = 0; 用于存储加法过程中的进位。
加法运算:接下来的while循环用于遍历两个链表,进行逐位相加。如果两个链表长度不同,循环会一直执行到其中一个链表结束。
处理进位:在加法过程中,如果两个节点的和加上进位c大于9,则需要计算新的进位和当前位的值。
if (b > 9) { c = b/10; b = b%10; } else c=0;
这行代码处理了进位逻辑。
创建新的节点:ListNode* temp = new ListNode(b); 创建一个新的节点,其值为当前位的和。
连接结果链表:
res->next = temp; res = temp;
将新创建的节点连接到结果链表,并移动res指针。
处理剩余的链表:如果其中一个链表在加法过程中先结束,还需要继续处理另一个链表的剩余部分。
处理最后的进位:
if(c) { ListNode* temp = new ListNode(c); res->next = temp; }
如果在加法结束后还有剩余的进位,需要将其作为一个新节点添加到结果链表。
反转结果链表:最后,再次使用while循环将结果链表反转回正常顺序。
返回结果:return pp; 返回结果链表的头节点。
代码效率分析:
时间复杂度:O(n + m),其中n和m分别是两个链表的长度。这是因为函数需要遍历两个链表各一次。
空间复杂度:O(n + m),除了输入的两个链表外,还需要为结果链表分配新的节点。
具体代码如下:
class Solution {
public:
ListNode* addInList(ListNode* head1, ListNode* head2) {
// write code here
ListNode* pre1 = NULL;
ListNode* cur1 = head1;
while (cur1) {
ListNode* temp = cur1->next;
cur1->next = pre1;
pre1 = cur1;
cur1 = temp;
}
ListNode* pre2 = NULL;
ListNode* cur2 = head2;
while (cur2) {
ListNode* temp = cur2->next;
cur2->next = pre2;
pre2 = cur2;
cur2 = temp;
}
ListNode* head = new ListNode(-1);
ListNode* res = head;
int c = 0;
while (pre1 && pre2) {
int b = pre1->val + pre2->val + c;
if (b > 9) {
c = b/10;
b = b%10;
}
else c=0;
ListNode* temp = new ListNode(b);
res->next = temp;
res = temp;
pre1 = pre1->next;
pre2 = pre2->next;
}
while (pre1) {
int b = pre1->val + c;
if (b > 9) {
c = b/10;
b = b%10;
}
else c=0;
ListNode* temp = new ListNode(b);
res->next = temp;
res = temp;
pre1 = pre1->next;
}
while (pre2) {
int b = pre2->val + c;
if (b > 9) {
c = b/10;
b = b%10;
}
else c=0;
ListNode* temp = new ListNode(b);
res->next = temp;
res = temp;
pre2 = pre2->next;
}
if(c)
{
ListNode* temp = new ListNode(c);
res->next = temp;
}
ListNode* pp=NULL;
ListNode* cc=head->next;
while(cc)
{
ListNode* temp=cc->next;
cc->next=pp;
pp=cc;
cc=temp;
}
return pp;
}
};