61、旋转链表
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
方法一:双指针/固定滑动窗口
1.1 思路分析
将链表每个节点向右移动 k 个位置,可以借鉴旋转数组的思路,其实就是把末尾的前k个移到链表头部。
定义双指针,让快指针先走k个位置,然后一起走,当快指针走到头,此时慢指针就在末尾数第k个的前一个位置了
还需要注意一些特例的判断,下面两个情况直接返回就行。
- 链表是空链表
- k正好是链表长度的整数倍
1.2 代码实现
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func rotateRight(head *ListNode, k int) *ListNode {
// 双指针/固定滑动窗口
ptr := head
// 1. 获取链表长度
n := 0
for tmp:=head;tmp!=nil;tmp=tmp.Next{
n++
}
// 1.1 特判
if n == 0{ // 空链表
return head
}
k = k % n
if k==0{ // k为0,不需要旋转
return head
}
// 2. 设置窗口
left, right := head, head
for i:=0; i<k; i++{
right = right.Next
}
// 3. 将窗口最右端移动到末尾
for right.Next != nil {
left = left.Next
right = right.Next
}
// 4. 将head切换成此时的left.Next
ptr = left.Next
left.Next = nil
right.Next = head
return ptr
}
1.3 测试结果
1.4 复杂度
- 时间复杂度:O(n)
- 空间复杂度:O(1)