问题描述:
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
and k = 2
,
return 4->5->1->2->3->NULL
.
原问题链接:https://leetcode.com/problems/rotate-list/
问题分析
对于这个问题来说,它本身很容易找到思路去解决。主要是实现容易出现各种小的错误。因为要将整个链表循环向右移动k位,所以我们就需要找到从链表的最后面往前的k个元素。因为按照最终的调整,这个位置的元素将作为新链表的结尾,而它后面的那个节点则将作为新链表的开头。
那么,我们该怎么来实现呢?从具体实现的步骤来看,我们需要找到倒数第k个元素,同时要找到它前面一个的元素,来进行操作。另外,既然是循环移位,原来链表最末尾的元素要和原来的链表头连接起来。另外一个,对于参数k来说,如果它本身比较大的话,我们就这么去循环的找也不是个办法。我们可以通过求得链表的总长度,再将k对链表长度取模,这样就可以减少遍历的距离。如果按照这种方式实现的话,我们需要有两个指针,一个遍历到链表最后将整个链表给连起来。另外一个记录倒数第k个元素的位置,然后将它前面的元素和它断开。
结合前面我们需要计算链表的长度来对k取模,我们可以将整个过程简化成只需要一个指针。首先我们用一个指针遍历到链表的末尾,并同时计算链表的长度。这样我们将k对链表长度取模之后再往前移动n - k 个位置。这个时候这个位置就是我们需要断开的节点,我们将这个节点断开,并返回它后面的元素作为新链表的头,则得到我们期望的结果。详细的代码实现如下:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) {
* val = x;
* next = null;
* }
* }
*/
public class Solution {
public ListNode rotateRight(ListNode head, int k) {
if(head == null || head.next == null || k == 0) return head;
ListNode p = head;
int len = 1;
while(p.next != null) {
p = p.next;
len++;
}
p.next = head;
k %= len;
for(int i = 0; i < len - k; i++) p = p.next;
head = p.next;
p.next = null;
return head;
}
}