刷题-力扣-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
题目分析
- 根据题目描述执行k次把链表末尾的结点放到头节点
- 把所有结点放到vector中
- 根据k%vec.size()的大小选择排序后链表的头结点
- 依次按顺序排列结点
代码
/**
* 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];
}
};