题目:
给你一个链表的头节点 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大于结点个数时,可以对其求余。即设长度为i,当k>=i, k%i 的值就是首元素结点在原链表中的次序,而 i-k%i 就是尾结点在原链表中的次序,我们让新的链表在尾结点断开,就得到了想要的旋转后的链表
主要函数代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* rotateRight(struct ListNode* head, int k){
if(k<=0||head==NULL||head->next==NULL){ //判断链表是否符合要求
return head;
}
struct ListNode*p1=head; //设置用来循环的指针,先定位于head
int i=1; //设置计数器,或者链表共有多少个结点
while(p1->next!=NULL){ //求链表长度 并且找到最后一个结点
p1=p1->next;
i++;
}
int j=i-k%i; //找到新链表中尾结点在原链表中的次序
p1->next=head; //首位相连
while(j--){ //将指针定位到尾结点
p1=p1->next;
}
struct ListNode*p2;
p2=p1->next; //用新定义的p2接管新链表的首元素结点
p1->next=NULL; //在新链表的尾结点处断开链表
return p2;
}
测试用例
in:{1,2,3,4,5} k=2
out:{3,4,5,1,2}
in:{1,2} k=1
out:{2,1}
in :{0,1} k=1
out{1,0}
in:{1,2,3,4,5} k=5
out:{1,2,3,4,5}
in:{1,2,3,4,5} k=0
out{1,2,3,4,5}
int:{1} k=1
out:{1}