leetcode61 旋转链表

题目

给你一个链表的头节点 head ，旋转链表，将链表每个节点向右移动 k 个位置。

示例

输入：head = [1,2,3,4,5], k = 2
输出：[4,5,1,2,3]

解析

这道题属实不好想：需要计算出链表的长度，然后在k > n的时候，需要k%n才知道右移几位；然后用n减去上面的那个取模后的值，表示从尾节点开始，还需要走几步；如果是走n步的话，那就没什么意义了，还是旋转后与原链表相同；然后将已经到达尾节点的next指向head，这样就形成了一个环；然后走上面的偏移量，就达到了新链表开始的前一个节点。将这个节点的next赋值成nil后，相当于断开了后面的环，此时拿到了预期结果

代码如下：

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func rotateRight(head *ListNode, k int) *ListNode {
	if k == 0 || head == nil || head.Next == nil {
		return head
	}
	n := 1 // 链表的长度
	cur := head
	for cur.Next != nil {
		cur = cur.Next
		n++ // 比如链表是12345的话，n就是5
	}
	add := n - k%n // k是向右移动的次数，k=12取模后是2，也就是45后面，还需要5-2=3个数
	if add == n { // 链表的长度为n的倍数，旋转后与原链表相同，不用处理
		return head
	}
	cur.Next = head // 尾节点连接到头节点，闭环
	for add > 0 {
		cur = cur.Next // 循环走完后，cur此时到了数字3处
		add--
	}
	ret := cur.Next // 数字4，确实是返回结果的头节点
	cur.Next = nil // 从这里阶段，也就是从上面的4开始，再走一个闭环，到3之后就被截断了
	return ret
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30