AcWing 35 反转链表
题目
定义一个函数,输入一个链表的头结点,反转该链表并输出反转后链表的头结点。
思考题:
- 请同时实现迭代版本和递归版本。
样例
输入:1->2->3->4->5->NULL
输出:5->4->3->2->1->NULL
思路
迭代版本
这道题使用迭代的思想很容易思考,我们只需要准备三个指针,分别是上一个节点prev
、当前节点curr
和curr
的下一个节点node
通过遍历整个链表,没到一个节点,我们先保存curr.next
给node
,再将curr.next
赋给prev
,也就改变了curr
的指向,随后我们把curr
和prev
都后移一位即可(在原来的顺序上,因为我们已经保留了node
),当curr == null
时,说明链表已经到结尾了,我们返回prev
即可
递归版本
递归版本如何思考呢,我们只需要思考一种情况,由于是递归实现,当递归到head
节点时,我们需要递归函数返回的结果是一个从head.next
开始已经被反转的链表,所以我们只需要将head
接到链表的结尾,并将head.next
指空即可
代码
c++ 迭代
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode *prev = NULL, *cur = head;
while (cur) {
auto node = cur->next;
cur->next = prev;
prev = cur;
cur = node;
}
return prev;
}
};
c++ 递归
class Solution {
public:
ListNode* reverseList(ListNode* head) {
if (!head || !head->next)
return head;
auto tail = reverseList(head->next);
head->next->next = head;
head->next = NULL;
return tail;
}
};
java 迭代
class Solution {
public ListNode reverseList(ListNode head) {
ListNode prev = null;
ListNode curr = head;
while (curr != null) {
ListNode node = curr.next;
curr.next = prev;
prev = curr;
curr = node;
}
return prev;
}
}
java 递归
class Solution {
public ListNode reverseList(ListNode head) {
if (head == null || head.next == null)
return head;
ListNode tail = head.next;
ListNode node = reverseList(head.next);
tail.next = head;
head.next = null;
return node;
}
}