一、题目描述
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
示例 1:
输入:head = [1,2,3,4,5], k = 2
输出:[4,5,1,2,3]
示例 2:
输入:head = [0,1,2], k = 4
输出:[2,0,1]
二、解题
快慢指针
使用快慢指针,间隔为K,然后将链表断开,然后在合并。并且K有可能大于链表的长度,所以对K取模。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode rotateRight(ListNode head, int k) {
//取模 链表的断开 快慢指针,然后合并
if(head == null){
return head;
}
ListNode slow = head,fast = head;
int cnt = 0;
while(slow != null){
cnt++;
slow = slow.next;
}
k = k % cnt;
if( k == 0){
return head;
}
slow = head;
for(int i = 0;i<k;i++){
fast = fast.next;
}
while(fast.next != null){
slow = slow.next;
fast = fast.next;
}
//断开链表
ListNode res = slow.next;
slow.next = null;
fast.next = head;
return res;
}
}
时间复杂度:O(n);
空间复杂度:O(1)。