面试题——反转链表

AcWing 35 反转链表

题目

定义一个函数,输入一个链表的头结点,反转该链表并输出反转后链表的头结点

思考题:

  • 请同时实现迭代版本和递归版本

样例

1
2
3
输入:1->2->3->4->5->NULL

输出:5->4->3->2->1->NULL

思路

迭代版本

​ 这道题使用迭代的思想很容易思考,我们只需要准备三个指针,分别是上一个节点prev、当前节点currcurr的下一个节点node

​ 通过遍历整个链表,没到一个节点,我们先保存curr.nextnode,再将curr.next赋给prev,也就改变了curr的指向,随后我们把currprev都后移一位即可(在原来的顺序上,因为我们已经保留了node),当curr == null时,说明链表已经到结尾了,我们返回prev即可

递归版本

​ 递归版本如何思考呢,我们只需要思考一种情况,由于是递归实现,当递归到head节点时,我们需要递归函数返回的结果是一个从head.next开始已经被反转的链表,所以我们只需要将head接到链表的结尾,并将head.next指空即可

代码

c++ 迭代

1
2
3
4
5
6
7
8
9
10
11
12
13
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++ 递归

1
2
3
4
5
6
7
8
9
10
11
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 迭代

1
2
3
4
5
6
7
8
9
10
11
12
13
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 递归

1
2
3
4
5
6
7
8
9
10
11
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;
}
}
-------------本文结束感谢您的阅读-------------