【算法百题之五十三】合并两个有序链表
大家好,我是Lampard~~
很高兴又能和大家见面了,接下来准备系列更新的是算法题,一日一练,早日升仙!
今天的问题是:合并两个有序链表
示例:
输入:1->2->4, 1->3->4 输出:1->1->2->3->4->4
链表结点结构:
struct ListNode {
int val;
ListNode *next;
ListNode() : val(0), next(nullptr) {}
ListNode(int x) : val(x), next(nullptr) {}
ListNode(int x, ListNode *next) : val(x), next(next) {}
};
思路:这道题我拿到手的时候我以为是手到拿来的一题,不就是定义一个res结点,然后遍历一下两个链表并找到大的结点,让res->next指向大的结点然后循环这个过程而已吗。于是我就写下了一下代码
测试代码:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode();
ListNode* curNode = head;
while (l1 && l2) {
ListNode* tmp = l1->val < l2->val ? l1 : l2;
curNode->next = tmp;
tmp = tmp->next;
curNode = curNode->next;
}
curNode->next = l1 ? l1 : l2;
return head->next;
}
整个过程简洁易懂,若两个链表都不为空,则curNode指向大的结点,若其中一个链表为空,则把curNode的next指向剩余结点即可。但是没想到老司机也会翻车,程序一执行就进入了死循环。我百思不得其解,在力扣的帮助下找到正确答案。
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode();
ListNode* curNode = head;
while (l1 && l2) {
ListNode** tmp = l1->val < l2->val ? &l1 : &l2;
curNode->next = *tmp;
*tmp = (*tmp)->next;
curNode = curNode->next;
}
curNode->next = l1 ? l1 : l2;
return head->next;
}
乍一看好像没有区别,唯一的差别是我的tmp是ListNode*类型,指向的是l1和l2。正确答案的tmp是ListNode** 类型,指向的是&l1,&l2。这时我不禁疑惑,**?指针的指针是什么鬼,我ListNode*不是直接就取到链表的地址了吗?为什么还要ListNode**?最终在和大学舍友们进行探讨,以及在csdn中查阅之后得出了结果。我错在对c++的指针和引用概念区分不清。
我是用指针变量生成的ListNode* tmp,系统会给它分配一个地址,然后其记录的内容是l1,l2的地址,执行tmp = tmp -> next这段语句的时候,tmp是指向了下一结点,但是它只是影响了自己,而l1,l2这两个链表并没有指向下一结点,所以l1和l2永远不会指向下一节点,也永远走不出循环。但是ListNode** tmp 指向的是l1和l2的引用&,系统没有给它们生成一个新的地址,而是tmp和链表使用同一地址,只不过这个地址有两个名字罢了。所以当执行tmp = tmp -> next这段语句的时候,l1和l2会指向下一结点,才能走出循环。
测试代码:
int main() {
ListNode *node1 = new ListNode(1);
ListNode *node2 = new ListNode(2);
ListNode *node3 = new ListNode(4);
node1->next = node2;
node2->next = node3;
ListNode *node4 = new ListNode(1);
ListNode *node5 = new ListNode(3);
ListNode *node6 = new ListNode(4);
node4->next = node5;
node5->next = node6;
ListNode* res = mergeTwoLists(node1, node4);
ListNode* curNode = res;
while (curNode != nullptr) {
cout << curNode->val << " ";
curNode = curNode->next;
}
}
测试结果: