这道题要求将给定的一个单链表重排,具体要求如下所示:
也就是说要求将原先的第一个结点后面接最后一个结点,然后接第二个结点,第二个结点后面接倒数第二个结点,以此类推。
这道题我一开始用暴力法解的,就是直接按照题目要求重排链表,可以做但是代码效率会比较低。这道题中要求的链表操作本质上是将链表的后一半反转之后再与前面的一半链表相合并构成的(这个观察很细致)。
所以官方题解中提到的一种方法是三种对链表的常见操作组合起来完成本题的功能,也就是查询链表中点、反转链表、合并链表这三个动作。这一个题考查了三个小问题,因此知识点非常浓缩,下面是实现代码:
首先实现查询链表中点的操作,这里使用了快慢指针算法。
ListNode* middleNode(ListNode* head) {
ListNode* slow = head;
ListNode* fast = head;
while (fast->next != nullptr && fast->next->next != nullptr) {
slow = slow->next;
fast = fast->next->next;
}
return slow;
}
注意,如果是偶数长度的链表,这里返回的链表中点是中间节点(有两个)的第一个。
反转链表的代码如下,可以用迭代法或者递归法来解,这里的代码是迭代法:
ListNode* reverseList(ListNode* head) {
ListNode* prev = nullptr;
ListNode* curr = head;
while (curr != nullptr) {
ListNode* nextTemp = curr->next;
curr->next = prev;
prev = curr;
curr = nextTemp;
}
return prev;
}
合并链表的操作如下,这个代码是针对本题的简化版本,因为本题中将链表从中点一分为2,这样造成的结果就是两个链表的长度最多只相差1个节点,合并时只需要将可能较长的链表放置在mergeList的第一个参数处即可。
void mergeList(ListNode* l1, ListNode* l2) {
ListNode* l1_tmp;
ListNode* l2_tmp;
while (l1 != nullptr && l2 != nullptr) {
l1_tmp = l1->next;
l2_tmp = l2->next;
l1->next = l2;
l1 = l1_tmp;
l2->next = l1;
l2 = l2_tmp;
}
}
最后就是主函数,将这三个小部件串联起来使用,注意由于查找中间节点的函数没有将链表前半段设置为空,那么这里是需要做一个额外操作的。
void reorderList(ListNode* head) {
if (head == nullptr) {
return;
}
ListNode* mid = middleNode(head);
ListNode* l1 = head;
ListNode* l2 = mid->next;
mid->next = nullptr;
l2 = reverseList(l2);
mergeList(l1, l2);
}
注意这里的
mid->next = nullptr;
将前半段列表的最后一个节点置为空,这步操作容易忽略。
这就是完成本题的代码,麻雀虽小五脏俱全。