题目
给你一个链表的头节点 head
,旋转链表,将链表每个节点向右移动 k
个位置。
示例 1:
输入:head = [1,2,3,4,5], k = 2
输出:[4,5,1,2,3]
1
2
2
示例 2:
输入:head = [0,1,2], k = 4
输出:[2,0,1]
1
2
2
提示:
- 链表中节点的数目在范围
[0, 500]
内 -100 <= Node.val <= 100
0 <= k <= 2 * 10
9
题解
java
public ListNode rotateRight(ListNode head, int k) {
// 尾节点
ListNode t = head;
int count = 0;
while (null != t) {
t = t.next;
count++;
}
// 节点数量为0或移动次数是节点数量整数倍 直接返回头结点
if (0 == count || k % count == 0) {
return head;
}
// 移动后n个节点至头部
int offset = k % count;
t = head;
int index = 0;
// 前半部分
while (index++ < (count - offset - 1)) {
t = t.next;
}
// 中间截断节点
ListNode m = t;
// 后半部分
while (offset-- > 0) {
t = t.next;
}
// 后半部分链接到头结点
t.next = head;
// 新的头结点
ListNode newHead = m.next;
// 新的尾节点
m.next = null;
return newHead;
}
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
31
32
33
34
35
36
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
31
32
33
34
35
36