LeetCode 24: 两两交换链表中的节点
题目描述
给定一个链表,两两交换其中相邻的节点,并返回交换后的链表。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
示例:
给定 1->2->3->4, 你应该返回 2->1->4->3.
解题
硬解
从前向后遍历整个链表,两两交换节点,继续向前遍历,我们可以用循环完成这一过程。在循环体中,我们分别记录当前节点的上一节点以及当前节点,假设上一节点用P表示,当前节点为C,后面一节点为N1,再后面为N2,两两交换的过程包括:
P的下一个指向N1
C的下一个指向N2
N1的下一个指向C
用N2更新C
用N1更新P
当当前节点或当前节点的下一节点为空时跳出循环。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* swapPairs(ListNode* head) {
ListNode *pre_head=new ListNode(), *pre_node = pre_head;
pre_head->next = head;
while (head && head->next){
pre_node->next = head->next;
head->next = head->next->next;
pre_node->next->next = head;
pre_node = head;
head = head->next;
}
return pre_head->next;
}
};
巧妙递归
思路和反转链表有那么一点相似,借助递归调用出栈时的返回结果,可以更方便的实现循环体的过程。我们需要实现这样一个递归函数,它返回当前两两节点交换后的第一个节点的指针,这样在每个递归函数中,我们可以:
1. 保存当前节点C的下一节点N1,C的下一节点指向递归函数返回的节点N2
2. N1的的下一位置指向C
3. 返回N1
为保证两两交换,如果当前节点为空或当前节点的下一节点为空,则直接返回当前节点。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* swapPairs(ListNode* head) {
if (!(head && head->next)) return head;
ListNode *temp = head->next;
head->next = swapPairs(temp->next);
temp->next = head;
return temp;
}
};