1. 题目描述
给你一个链表的头节点 head, 旋转链表, 将链表每个节点向右移动 k 个位置
输入:head = [1, 2, 3, 4, 5], k = 2
输出:[4, 5, 1, 2, 3]
输入:head = [0, 1, 2], k = 4
输出:[2, 0, 1]
2. 解题思路
- 当链表 head 为空或只有一个结点的情况, 直接返回 head
- 当向右移动 k 个位置的 k = 0, 不需要处理直接返回 head
- 计算出链表节点总数 length 后, 需要 k = k % length, 用于处理 k > length 场景, 还需要判断 k == 0, 直接返回 head
此题的关键点有三个
- 定义一个最终结果链表 ListNode result = new ListNode(-1, head), 将 result.next 指向倒数第 k 个结点
- 将倒数第 k + 1 个结点与倒数第 k 个结点断开
- 将最后一个结点的 next 指向第一个结点 head
public ListNode rotateRight(ListNode head, int k) {
if (null == head || head.next == null || k == 0) {
return head;
}
ListNode result = new ListNode(-1, head);
ListNode last = head;
int length = 1;
while (last.next != null) {
last = last.next;
length++;
}
k = k % length;
if (k == 0) {
return result.next;
}
ListNode temp = head;
for (int i = 0; i < length - k - 1; i++) {
temp = temp.next;
}
result.next = temp.next;
temp.next = null;
last.next = head;
return result.next;
}
|