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.一定要记得衔接和断开,否则一定是死循环