题目: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.com/problems/rotate-list/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
观察示例可知,右移次数k可能出现重复次数,计算链表长度len,实际右移次数k%len,相当于链表后k%len个元素拼接到链表的前头。比如示例1,k%len = 2,所以后面两个元素拼接到前头:4->5->1->2->3。
考虑几种特殊情况:
- k=0,不需要旋转
- 空链表,不需要改变
- 一个结点的链表,不需要改变
- k%len == 0,即k是len的倍数,也无需改变
解题代码
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(k == 0 || !head || !head->next)//k=0无需转换,小于等于一个也无需
return head;
ListNode *temp = head;
ListNode *last = head;
int len = 1;
while(last->next)//计算长度和找到最后一个
{
len++;
last = last->next;
}
int num = len - k % len;
if(num == len)//k是len的倍数
{
return head;
}
for(int i = 0; i <num - 1; i++)//找到新的头的前一个
{
temp = temp->next;
}
ListNode *newHead = temp->next;
temp->next = nullptr;
last->next = head;
return newHead;
}
};
解题感悟
多观察示例,就可找出其中的规律,解决就并不难了。因为直接旋转链表的每个结点是很难也比较复杂,找到在k次就后k个拼接到前头的规律,解决就很轻松了。