题目描述:
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
示例:
输入:head = [1, 2, 3, 4, 5],k = 2
输出:[4, 5, 1, 2, 3]
输入:head = [0, 1, 2],k = 4
输出:[2, 0, 1]
个人解法:
首先对特殊情况进行处理
若链表为空或仅有一个节点,则不做处理,直接返回,然后计算链表长度,并令 k 对链表长度取余,结果用 num 保存,表示等效移动位数,若 num 为 0 ,则不做处理,直接返回
对链表的右移可以看作从链表中间某一节点开始到链表尾,与链表前端剩余部分的拼接,如示例中的 [1, 2, 3, 4, 5] 右移两位后,可以看作是 [4, 5] 和 [1, 2, 3] 的拼接,得到 [4, 5, 1, 2, 3]
具体步骤如下:
(1)用临时指针指向 head,向后移动 length - num - 1位,此时临时指针为应断开节点
(2)该节点的后继节点为新链表的起始位置,用 start 保存其后继节点,并置空该节点的 next
(3)用指针 result 保存新链表的头节点,以便最后返回
(4)将 start 指针向后移动 num - 1 位,到达原链表末尾,修改末尾节点的 next 为 head
最后返回 result 即可
public ListNode rotateRight(ListNode head, int k) {
if (head == null || head.next == null) {
return head;
}
ListNode temp = head;
ListNode start;
int length = 0;
while (temp != null) {
length++;
temp = temp.next;
}
int num = k % length;
if (num == 0) {
return head;
}
temp = head;
for (int i = 0; i < length - num - 1; i++) {
temp = temp.next;
}
start = temp.next;
temp.next = null;
ListNode result = start;
for (int i = 0; i < num - 1; i++) {
start = start.next;
}
start.next = head;
return result;
}
官方解法(闭合为环):
特殊情况的处理与个人解法类似
计算链表长度与个人解法有所不同,最终临时指针将停留在链表的末尾节点上,以便修改末尾节点的 next 为 head , 实现链表闭合为环
然后临时指针向后移动 length - k % length 位,即可到达新链表的最后一个节点,其后继节点为新链表的起始节点,故只需要保存其后继节点,并将当前节点的 next 置空,返回后继节点即可
总的来说,就是先将链表闭合为环,再通过计算,在相应的位置断开,即可得到新链表
public ListNode rotateRight1(ListNode head, int k) {
if (k == 0 || head == null || head.next == null) {
return head;
}
int length = 1;
ListNode temp = head;
while (temp.next != null) {
length++;
temp = temp.next;
}
if (k % length == 0) {
return head;
}
temp.next = head;
int count = length - k % length;
while (count > 0) {
temp = temp.next;
count--;
}
ListNode result = temp.next;
temp.next = null;
return result;
}