每日一题,很久没做链表的题了,今天做l一道相对简单的力扣中等难度题。
题目要求
给你一个链表的头节点 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]
题目解析
这道题的要求很简单,根据它给出的数字,将链表移动即可。
首先计算链表的长度,当k等于链表长度时链表就正好转完一圈,相当于没有移动,所以一旦k的值超过了链表的长度,那么移动的值就相当于k除以链表长度取余即可。
拿到真正要移动的位置后,就可以寻找到链表旋转后新头节点和新尾节点,具体操作为让一个指针从头节点移动(链表长度减去移动的次数再减一)次,就指向到了新尾节点,下一个节点就是链表的新头节点,再将链表尾部指向头部,将新尾部置空即可,这样就得到了新的旋转链表。
例如示例1中:
这样就完成了旋转,整体逻辑比较简单,具体实现可以查看下面代码。
代码实现
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode rotateRight(ListNode head, int k) {
if(head==null) return null;
int len=1;
ListNode h = head;
while(h.next!=null){
len++;
h=h.next;
}
int n = k%len;
if(n==0) return head;
ListNode newTail = head;
for(int i=0;i<len-n-1;i++){
newTail = newTail.next;
}
ListNode newHead = newTail.next;
h.next = head;
newTail.next = null;
return newHead;
}
}