题目地址:
https://www.lintcode.com/problem/rotate-list/description
给定一个链表,再给定一个正整数 k k k,要求将链表向右旋转 k k k步。旋转 1 1 1步的意思是,将每个node的位置向右移动一格,最后一个node移动到链表首位。
思路是,找到倒数第 k + 1 k+1 k+1个节点,将链表尾部连接到链表头,然后将倒数第 k + 1 k+1 k+1个节点与后面断开,再返回新链表的头即可。需要注意的是,如果 k k k大于链表长度,在找倒数第 k + 1 k+1 k+1个节点的时候会出错,所以要先求表长,再取余。代码如下:
public class Solution {
/**
* @param head: the List
* @param k: rotate to the right k places
* @return: the list after rotation
*/
public ListNode rotateRight(ListNode head, int k) {
// write your code here
// 如果链表长度为0,取余会出错,所以对空表单独考虑
if (head == null) {
return null;
}
ListNode dummy = new ListNode(0);
dummy.next = head;
int l = len(head);
k %= l;
ListNode slow = dummy, fast = dummy;
for (int i = 0; i < k; i++) {
fast = fast.next;
}
while (fast.next != null) {
slow = slow.next;
fast = fast.next;
}
fast.next = head;
// 先缓存新链表的表头,再断开
ListNode res = slow.next;
slow.next = null;
return res;
}
private int len(ListNode head) {
int res = 0;
while (head != null) {
res++;
head = head.next;
}
return res;
}
}
class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。