Rotate List
Given a list, rotate the list to the right by k places, where k is non-negative.
For example: Given
1->2->3->4->5->NULL
andk = 2
, return4->5->1->2->3->NULL
.
双指针法
复杂度
时间 O(N) 空间 O(1)
思路
旋转链表的本质,就是把链表的后半部分放到前面来,所以关键在于如何找链表后半部分的起始节点。实际上,我们用一个快指针先走k步,然后快慢指针同时走,这样当快指针到末尾时,慢指针就到链表后半部分的起始节点了。不过,由于k可能大于链表的长度,所以我们要先对k取链表长度的模。要计算链表长度只要遍历一遍链表就行了。
注意
k取余后为0,或者原链表为空的情况下,都要直接返回原链表,不然将慢指针的下一个作为新链表头会出错。
代码
public class Solution {
public ListNode rotateRight(ListNode head, int k) {
ListNode fast = head;
int length = 0;
// 计算链表长度
while(fast != null){
length++;
fast = fast.next;
}
// 如果不旋转或者原链表为空 则直接返回
if(head == null || k % length == 0){
return head;
}
fast = head;
// 让快指针先走k步
for(int i = 0; i < k % length; i++){
fast = fast.next;
}
// 找到新头节点的前一个节点
ListNode slow = head;
while(fast.next != null){
fast = fast.next;
slow = slow.next;
}
// 将后半部分放到前面来
ListNode newHead = slow.next;
slow.next = null;
fast.next = head;
return newHead;
}
}