LeetCode 206. 反转链表
给你单链表的头节点head
,请你反转链表,并返回反转后的链表。示例 1:
输入:head = [1,2,3,4,5] 输出:[5,4,3,2,1]示例 2:
输入:head = [1,2] 输出:[2,1]示例 3:
输入:head = [] 输出:[]提示:
- 链表中节点的数目范围是
[0, 5000]
-5000 <= Node.val <= 5000
进阶:链表可以选用迭代或递归方式完成反转。你能否用两种方法解决这道题?
思路:迭代 & 递归
时间复杂度:O(N)
空间复杂度:O(1)
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
// 迭代
func reverseList(head *ListNode) *ListNode {
// 该初始化方式会在反转后的尾节点添加值为0的空节点,导致错误
// pre, mid, end := &ListNode{}, head, &ListNode{} // xxx
var pre, mid, end *ListNode = nil, head, nil
for mid != nil {
end = mid.Next
mid.Next = pre
pre = mid
mid = end
}
return pre
}
// 递归
func reverseList(head *ListNode) *ListNode {
// 直到当前节点的下一个节点为空时返回当前节点
// 由于5没有下一个节点了,所以此处返回节点5
if head == nil || head.Next == nil {
return head
}
// 递归传入下一个节点,目的是为了到达最后一个节点
newHead := reverseList(head.Next)
head.Next.Next = head
head.Next = nil
return newHead
}
/*
第一轮出栈,head为5,head.next为空,返回5
第二轮出栈,head为4,head.next为5,执行head.next.next=head也就是5.next=4,
把当前节点的子节点的子节点指向当前节点
此时链表为1->2->3->4<->5,由于4与5互相指向,所以此处要断开4.next=null
此时链表为1->2->3->4<-5
返回节点5
第三轮出栈,head为3,head.next为4,执行head.next.next=head也就是4.next=3,
此时链表为1->2->3<->4<-5,由于3与4互相指向,所以此处要断开3.next=null
此时链表为1->2->3<-4<-5
返回节点5
第四轮出栈,head为2,head.next为3,执行head.next.next=head也就是3.next=2,
此时链表为1->2<->3<-4<-5,由于2与3互相指向,所以此处要断开2.next=null
此时链表为1->2<-3<-4<-5
返回节点5
第五轮出栈,head为1,head.next为2,执行head.next.next=head也就是2.next=1,
此时链表为1<->2<-3<-4<-5,由于1与2互相指向,所以此处要断开1.next=null
此时链表为1<-2<-3<-4<-5
返回节点5
出栈完成,最终头节点5->4->3->2->1
*/