Leetcode 61. 旋转链表
题目
给定一个链表,旋转链表,将链表每个节点向右移动 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
题解
旋转是一个循环的过程,因此先求出链表的长度,根据长度判断我们要将后面几个节点挪到前面。然后再根据求出的位置,进行相关操作,具体过程见代码
代码
ListNode* rotateRight(ListNode* head, int k) {
int length=0;
ListNode*p = head;
while(p){
length++;
p = p->next;
}
if(length == 0) return head;
int place = length-k%length,i=0; //根据链表的长度,旋转步数求出需要移动的节点的位置
if(place == length) return head;
p = head;
ListNode *newhead;
while(p){
i++;
if(i == place){
newhead=p->next; //新的头节点
p->next=NULL;
p=newhead;
while(p){
if(p->next==NULL){ //找到最后一个节点,让其指向原头节点
p->next=head;
break;
}
p=p->next;
}
break;
}
p = p->next;
}
return newhead;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。