题目链接: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. 

这道题的要求是向右旋转链表k步。

其实就是把链表后面l-k个节点放到前面,可以采用快慢指针处理。不过当k大于链表长度l的时候,就会出现问题。这时需要做一点点处理,就是当快指针f指向最后节点后,下一次令其指向头节点,这样问题就解决了。这样的时间复杂度是O(k),因此,当k非常大的时候,会TLE。

考虑到k大于l的时候,可以令k对l取模处理。因此,先计算链表长度,同是令p指向链表最后节点。接下来k对l取模,再令q从头节点往后移动l-k步,这样q就指向要旋转的位置。最后将q后面的节点放到前面即可。

假设k=2,加头之后链表为:
h -> 0 -> 1 -> 2 -> 3 -> 4 -> 5 -> NULL。

领p指向链表最后节点,q指向要旋转的位置,即:
h -> 0 -> 1 -> 2 -> 3 -> 4 -> 5 -> NULL
                    ^         ^
                    |         |
                    q         p

接下来把q后面的节点放到前面:
h -> 0 -> 4 -> 5 -> 1 -> 2 -> 3 -> NULL
               ^              ^
               |              |
               p              q

处理的时候,同样先对链表加头,方便处理,类似于Remove Nth Node From End of ListSwap Nodes in PairsReverse Nodes in k-Group,同样先对链表加头处理。

时间复杂度:O(l)(l是链表长度)

空间复杂度:O(1)

 1 class Solution
 2 {
 3 public:
 4     ListNode *rotateRight(ListNode *head, int k)
 5     {
 6         if(head == NULL)
 7             return NULL;
 8         
 9         // 链表加头,方便处理
10         ListNode *h = new ListNode(0), *p, *q;
11         h -> next = head;
12         
13         int l = 0;
14         // p指向最后节点
15         for(p = h; p -> next != NULL; ++ l, p = p -> next); 
16         // q指向要要旋转位置,即旋转后最后节点
17         for(q = h, k = l - k % l; k > 0; -- k, q = q -> next); 
18         
19         // 把链表后半段放到前面
20         p -> next = h -> next;
21         h ->next = q -> next;
22         q -> next = NULL;
23         
24         return h -> next;
25     }
26 };