给你一个链表的头节点
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]
记给定链表的长度为 n,注意到当向右移动的次数 k≥n时,我们仅需要向右移动 k mod n次即可。因为每 n次移动都会让链表变为原状。这样我们可以知道,新链表的最后一个节点为原链表的第 (n−1)−(k mod n)个节点(从 0开始计数)。
这样,我们可以先将给定的链表连接成环,然后将指定位置断开。
具体代码中,我们首先计算出链表的长度 n,并找到该链表的末尾节点,将其与头节点相连。这样就得到了闭合为环的链表。然后我们找到新链表的最后一个节点(即原链表的第 (n−1)−(k mod n)个节点),将当前闭合为环的链表断开,即可得到我们所需要的结果。
特别地,当链表长度不大于 1,或者 k为 n的倍数时,新链表将与原链表相同,我们无需进行任何处理。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* rotateRight(struct ListNode* head, int k) {
//k==0:不移动元素。链表为空。链表只有一个元素。这三种情况链表都不变
if(k==0||head==NULL||head->next==NULL)
return head;
//计算链表长度
struct ListNode* p=head;
int count=1;
while(p->next!=NULL)
{
p=p->next;
count++;
}
//新链表的最后一个节点为原链表的第几个几点
int add = count - k % count;
if(add==count)//结点没变,此时k为n的倍数
{
return head;
}
//将原链表尾节点与头结点连接起来,形成循环链表(环)
p->next=head;
while(add--)//寻找新链表的尾节点
{
p=p->next;
}
struct ListNode* list=p->next;
p->next=NULL;
return list;
}