LeetCode 61. 旋转链表

示例 1:

输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
示例 2:

输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL

class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(k == 0 || !head ||!head->next)
            return head;
        int len = 0;
        ListNode *tmp = head, *oldtail;
        while(tmp->next)//求链表长度
        {
            ++len;
            tmp = tmp->next;
        }
        oldtail = tmp;//求到旧的尾节点
        ++len;
        k = k%len;
        if(k == 0)
            return head;
        tmp = head;
        int count = len-k-1;//找到新的头结点移动次数
        while(count--)
        {
            tmp = tmp->next;
        }
        ListNode *newHead = tmp->next;//新头节点
        tmp->next = NULL;//断开连接
        oldtail->next = head;//旧的尾节点连接到头结点
        return newHead;
    }
};