1.题目61. 旋转链表
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
输入: 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
将链表分成段,第一段是k%len长,是链表的后k%len个元素 需要到队头
第二段是len - k%len 是链表的前面几个元素 需要放到第一段的后面
2.思路
算法:
* 1.先计算链表长度len
* 2.记录旧队头oldHead,先走len - k%len,找到新队头newHead
* 3.记得断开新队尾
* 4.走到队尾,tail.next=oldHead
* 5.返回新队头
3.代码实现
public ListNode rotateRight(ListNode head, int k) {
if(head==null) return null;
int len = getLen(head);
int secLen = len - k%len;
ListNode oldHead = head;
for (int i=0;i<secLen-1;i++) {
head = head.next;
}
ListNode newHead = head.next;
head.next = null; //断开新队尾
head = newHead;
// 找到链尾
while (head!=null && head.next!=null) {
head = head.next;
}
if (head!=null){
head.next = oldHead;
}
return newHead!=null?newHead:oldHead;
}
private int getLen(ListNode head) {
int cnt = 0;
while (head!=null) {
cnt++;
head = head.next;
}
return cnt;
}
4.注意点
1.边界问题 当k=0或者k=len时,新队头为空,记得返回旧队头
2.一定要记得衔接和断开,否则一定是死循环