【leetcode】Rotate List

/**
 * 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) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        assert(k >= 0);
        
        if(head == NULL) return NULL;
        
        ListNode *ret = head;
        int len = 1;
        while(ret->next != NULL){
            len++;
            ret = ret->next;
        }
        
        ret->next = head;   // merge a circle list
        
        k = k % len;
        int step = len - k - 1;     // 1 is a head note
        ret = head;
        while(step > 0){
            ret = ret->next;
            step--;
        }
        
        head = ret->next;
        ret->next = NULL;
        
        return head;
    }
};

/**
 * 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) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        assert(k >= 0);
        
        if(head == NULL) return NULL;
        
        ListNode *ret = head;
        int len = 1;
        while(ret->next != NULL){
            len++;
            ret = ret->next;
        }
        
        
        ListNode *tmp = head;
        
        k = k % len;
        int step = len - k - 1;     // 1 is a head note
        while(step > 0){
            tmp = tmp->next;        // left
            step--;
        }
        
        ListNode *tmp2 = tmp;       // right
        while(tmp2->next != NULL){
            tmp2 = tmp2->next;
        }
        
        tmp2->next = head;      // right->next
        
        ret = tmp->next;        // ret head
        tmp->next = NULL;       // left->next
        
        return ret;
    }
};