刷题-力扣-61. 旋转链表

61. 旋转链表

题目链接

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

题目描述

给你一个链表的头节点 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

题目分析

1. 根据题目描述执行k次把链表末尾的结点放到头节点
2. 把所有结点放到vector中
3. 根据k%vec.size()的大小选择排序后链表的头结点
4. 依次按顺序排列结点

代码

/**
 * 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 || !(head->next)) return head;
        vector<ListNode*> vec;
        ListNode* p = head;
        while (p) {
            vec.push_back(p);
            p = p->next;
        }
        k = k % vec.size();
        if (k == 0) return head;
        vec[vec.size() - 1]->next = vec[0];
        int index = vec.size() - k;
        vec[index - 1 < 0 ? vec.size() - 1 : index - 1]->next = nullptr;
        return vec[index];
    }
};