leetcode做题记录
给你一个链表的头节点 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
代码如下:
/**
* 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) {
//head为空时返回nullptr
if(!head)
{
return nullptr;
}
//不移动时直接返回head
if(k == 0)
{
return head;
}
ListNode* dummy = new ListNode(0,head);
ListNode* End,*temp;
//n为链表长度,End指向链表末尾
int n = 0;
while(head)
{
End = head;
head=head->next;
n++;
}
//链表长度为1时,直接返回链表
if(n==1)
{
return dummy->next;
}
//将向右移动转化为向左移动,因为没有指向前的指针不能循环
k = n-k%n;
//当链表长度大于1时循环才有效
for(int i = 0 ;i<k;i++)
{
temp = dummy->next;
dummy->next = temp->next;
End->next = temp;
temp->next = nullptr;
End = temp;
}
return dummy->next;
}
};
个人理解:
因为循环中无法记录最后一节点的上一节点,因此将向左移动转化为向右移动,每次移动将链表首位元素放置到链表末尾。