记录:2021.8.10 链表:旋转链表(med)
题目:
给你一个链表的头节点 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
Related Topics 链表 双指针
👍 603 👎 0
思路:
将链表每个结点右移k个位置,其实就是从最后一个结点开始,依次移动k个结点到链表头部,
当链表的size为0、1或<k且k为size的倍数的时候,移动都会让链表回归原状,所以直接输出即可。
当链表长度大于k且k>0时,将链表连成一个环,即在遍历链表时,用一个pre结点来保存每个结点的前一个结点,
当while循环结束时,pre结点即为最后一个结点,然后pre.next=head就把链表连成了环。
在这种情况下,链表指针head只需要向前移动size-(k%size)个位置,就走到了需要断开的位置。
同样在循环中使用pre结点来保存前一个节点,当满足条件循环结束后,pre结点即为需要断掉的结点,令pre.next=null就将环链表断开。返回head即可。
代码:
class Solution {
public ListNode rotateRight(ListNode head, int k) {
ListNode start=head;
ListNode pre=head;
ListNode ans=head;
int size=0;
while (start!=null){
size++;
pre=start;
start=start.next;
}
if(size<=1||(size<k&&k%size==0)){
return ans;
}
//这之后的pre就是链表中的最后一个结点,将链表连成一个环
pre.next=head;
int count=0;
while (head!=null){
count++;
if(count>size-(k%size)){
break;
}
pre=head;
head=head.next;
}
//这之后的head走到了需要断开的位置
pre.next=null;
ans=head;
return ans;
}
}