61. 旋转链表

61. 旋转链表

题目介绍

给你一个链表的头节点 head ，旋转链表，将链表每个节点向右移动 k 个位置。

示例 1：

输入：head = [1,2,3,4,5], k = 2
输出：[4,5,1,2,3]
示例 2：

输入：head = [0,1,2], k = 4
输出：[2,0,1]

提示：

链表中节点的数目在范围 [0, 500] 内
-100 <= Node.val <= 100
0 <= k <= 2 * 109

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/rotate-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解

我们需要找到新链表的头结点的上一个结点，和旧链表的尾部结点，然后重新给next赋值即可。

先遍历一般计算链表长度，之后 k = k%n，让 k 值一般化；

再从头开始遍历，找到倒数第 k+1 个结点和最后一个结点，之后改变next值。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(head == NULL)
        {
            return head;
        }
        int i=0;
        ListNode *new_head = head;
        // 计算链表长度
        while(new_head!=NULL)
        {
            new_head = new_head->next;
            i++;
        }
        k = k%i;
        if(k==0)
        {
            return head;
        }
        // 向右旋转k个位置，则头结点变成倒数第k个结点
        ListNode *p, *q;
        p = head; // 原链表尾
        q = head; // 新链表尾
        i=0;
        while(p->next!=NULL)
        {
            if(i<k)
            {
                i++;
            }
            else{
                q = q->next;
            }
            p = p->next;
        }
        new_head = q->next;
        p->next = head;
        q->next = NULL;
        return new_head;
    }
};

时间复杂度为$O(n)$，空间复杂度为$O(1)$。