【Leetcode】之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个节点问题的变种，问题的关键点自然也就是双指针的策略。采用两个指针，一个指针指向链表头，另一个指针指向从头开始走k步所到达的节点。然后开始遍历链表，每次都将两个指针同时往后移，直至到达链表尾。

本题有一个地方说明不太清，就是k>链表长度len的情况。根据测试案例的情况来看，k采用的递增方式应该是k mod len。所以需要预先遍历一下链表得到链表长度len，然后再得到k mod len。 测试通过的程序如下，复杂度应该是o(n):

/**
 * 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) {
        ListNode * res;
        if(head==NULL) return NULL;
 
        ListNode * curr = head;ListNode * k_next = head; ListNode * pre=head;
        int cnt=1;
        while(k_next->next!=NULL){
              k_next = k_next->next;
              cnt++;
        }
        k_next = head;
        int len =cnt;
        k = k%len;
        if(k==0) return head;
        cnt=1;
        while(k_next->next!=NULL && cnt<k){
              k_next = k_next->next;
              cnt++;
        }
        if(cnt<k) return head;
        if(k_next->next==NULL) return head;
        cnt=1;
        while(k_next->next!=NULL){
            pre=curr;
            curr=curr->next;
            k_next=k_next->next;
            cnt++;
        }
        
        k_next->next = head;
        pre->next=NULL;
        res = curr;
        return res;
    }
};