一、问题描述
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例1:
输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
示例2:
输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL
二、算法思路
首先,通过遍历整个链表,得到链表的长度和链表的尾结点,使用struct ListNode *end
变量将尾结点记录下来,然后,将k对链表长度求余,得到实际旋转次数,遍历链表,得到经过k次旋转后链表的头结点
使用struct ListNode *new_head
变量将新的头结点记录下来。最后,将尾指针指向头结点,使链表成环,再将新的头结点与前面的结点断开,可得到经过k次旋转后的链表。
三、代码实现(C语言)
struct ListNode* rotateRight(struct ListNode* head, int k)
{
if(head==NULL||k==0)
{
return head;
}
int len=1;//没有执行if语句,表示链表不为空,可以直接将长度设为1
struct ListNode *p;//遍历链表,求长度,寻找旋转后的头结点
struct ListNode *end;//最后一个元素
struct ListNode *new_head;//新的头结点
p=head;//使p指向第一个结点
//遍历链表,求链表的长度
while (p->next!=NULL)
{
len++;
p=p->next;
}
end=p;//保存最后一个元素
k=k%len;//k对len求余,得到实际旋转次数
if(k==0)
{
return head;
}
p=head;
for(int i=0;i<len-k-1;i++)
{
p=p->next;
}
new_head=p->next;//保存新的头结点
end->next=head;//尾指针指向头结点,形成环
head=new_head;
p->next=NULL;//断链
return head;
}