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 * 10^9

代码

/**
 * 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) {
        // 如果 k 为 0 或者链表为空或者链表只有一个节点，则无需旋转
        if (k == 0 || head == nullptr || head->next == nullptr)
            return head;

        // 计算链表的长度
        ListNode *p = head;
        int n = 1;
        while (p->next) {
            n++;
            p = p->next;
        }

        // 计算实际需要旋转的节点数
        k = k % n;
        if (k == 0) return head;

        // 将链表首尾相连
        p->next = head;

        // 找到新的尾节点位置，新的尾节点是从头节点开始的第 (n - k) 个节点
        for (int i = 0; i < n - k; i++) {
            p = p->next;
        }

        // 新的头节点是新的尾节点的下一个节点
        ListNode *newHead = p->next;

        // 断开链表，使其成为非循环链表
        p->next = nullptr;

        return newHead;
    }
};