题目描述
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
思路:
- 如果链表为空,则返回null
- 如果链表只有一个头节点,则不论怎么旋转都是链表本身,返回head
- 因为向右移动k的len倍数和移动k是一样的效果(len为链表的长度),所以先求出链表的长度
ListNode tail = head;
int len ;
for(len = 1;tail.next != null;len++){
tail = tail.next;
}
k = k%len;
- 定义两个指针,fast快指针和slow慢指针,fast先移动k步,如果fast和slow指向一样,则返回head,否则fast和slow一起移动,直到fast.next为空
- 将fast的next指向头节点,头节点重新改为slow的下一个节点,slow的下一个节点为空
代码:
/**
* 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;
}
if(head.next == null){
return head;
}
ListNode tail = head;
int len ;
for(len = 1;tail.next != null;len++){
tail = tail.next;
}
k = k%len;
ListNode fast = head;
ListNode slow = head;
for(int i = 0;i<k;i++){
fast = fast.next;
}
if(fast == slow){
return head;
}
while(fast.next != null){
fast = fast.next;
slow = slow.next;
}
fast.next = head;
head = slow.next;
slow.next = null;
return head;
}
}