【Leetcode】【Medium】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

and k =

2

,
return

4->5->1->2->3->NULL

.

解题思路：

题目看上去很简单，但是需要注意一下点：

1、任何输入都有可能，即使是无效的输入（例如链表为空，k不为0），所以程序开始时的检查在任何时候都非常重要。

2、另一个边界条件，就是是否在空指针中，使用了->next操作，很多链表错误都发生在此处。

3、K是旋转的结点数，因此K可能大于整个链表的结点数。

4、K不仅可能大于结点数，还可能非常大，因此传统的循环耗费时间太多，要对K做取余操作。

代码：

/**
* Definition for singly-linked list.
* struct ListNode {
*     int val;
*     ListNode *next;
*     ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if (k == 0 || !head)
return head;
ListNode* pre = head;
ListNode* last = head;
int num = 1;
while (k--) {
if (last->next) {
num++;
last = last->next;
} else {
k = k % num;
last = head;
}
}

while (last->next) {
pre = pre->next;
last = last->next;
}

last->next = head;
head = pre->next;
pre->next = NULL;
return head;
}
}