[链表]–旋转链表
题目链接
题目
给定一个链表,旋转链表,将链表每个节点向右移动 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
输入: 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
解析
两种方法
第一种:从示例就可以看出来将前面一部分节点整体连接剩余节点的后面
特殊情况:k == size 的时候,链表相当于没动,所以在进行截取时,需要k=(k%size);
- 首先计算链表长度,并判断 k 的情况;
- 遍历链表截取前面(size-k)个节点,第(size-k)个节点就是旋转后的尾结点,它的next = null;
- 此时原链表就被分割成两部分,继续之前的遍历遍历到最后一个节点连接前半部分链表即可。
代码实现
public class Solution61 {
/**
* Definition for a singly-linked node
*/
class ListNode {
int val;
ListNode next;
public ListNode(int val) {
this.val = val;
}
}
/**
* 将旧链表截成两部分, 前半部分长度(size-k); 后半部分长度 k , 很显然只需要将前半部分连接到后半部分后面即可
* 1.计算链表长度 size, 由于向右移动 size 个位置正好回到原始位置, 所以不做考虑, k %= size 即可
* 2.上一步完成后 k=0 说明链表不需要移动, return head
* 3.新链表头结点的位置就是倒数第 k 个结点的位置, 定义 cur 遍历原链表, 找出头结点, 并将头结点的前驱置为 null(新链表的尾结点就是它了)
* 4.原链表经过上一步分为两部分, 遍历后半部分到尾结点, 将前半部分连接到尾结点之后就行
*/
public ListNode rotateRight(ListNode head, int k) {
if (head == null ) return null;
int size = 0;
// 计算结点个数
for (ListNode cur = head; cur != null; cur = cur.next) {
size++;
}
// 当每次旋转的 k=size 时, 正好回到原始位置, 不做考虑
k %= size;
if (k == 0) return head;
// 头结点后移 step 就是旋转后新链表的头结点 newHead
int step = size - k;
ListNode cur = head;
// 移至 newHead 前驱结点处, 这个前驱结点很明显是新链表尾结点, 需要置为null
while (cur.next != null && step-- > 1) {
cur = cur.next;
}
// 链表分割成两部分
ListNode newHead = cur.next;
cur.next = null;
// 遍历新链表至尾结点, 将原来的头结点连接在新链表末尾
cur = newHead;
while (cur.next != null) {
cur = cur.next;
}
cur.next = head;
return newHead;
}
}
第二种方法比较容易了:结环然后再适当的位置断开就好
- 前面和上一个方法一样进行判断特殊情况;
- 结环:tail.next = head;
- 寻找合适位置断开(从 head 开始向后 size-k-1个节点),该节点就是尾结点了,下一节点为新链表头节点。
public class Solution61 {
/**
* Definition for a singly-linked node
*/
class ListNode {
int val;
ListNode next;
public ListNode(int val) {
this.val = val;
}
}
/**
* 将原链表变成环, 在合适位置(新头结点之前)断开
* 1.计算链表长度 size, 由于向右移动 size 个位置正好回到原始位置, 所以不做考虑, k %= size 即可
* 2.上一步完成后 k=0 说明链表不需要移动, return head
* 3.将尾结点的 next 置为 head, 结环
* 4.新链表头结点的位置就是第 (size-k) 个结点的位置, 定义 cur 遍历原链表, 找出头结点, 并将头结点的前驱置为 null(断开)
*/
public ListNode rotateRight(ListNode head, int k) {
if (head == null ) return null;
int size = 1;
// 计算结点个数
ListNode cur = head;
for (; cur.next != null; cur = cur.next) {
size++;
}
// 当每次旋转的 k=size 时, 正好回到原始位置, 不做考虑
k %= size;
if (k == 0) return head;
// 结环
cur.next = head;
// 寻找新链表头结点并断开
int step = size - k;
// 移至 newHead 前驱结点处, 这个前驱结点很明显是新链表尾结点, 需要置为null
while (head.next != null && step-- > 1) {
head = head.next;
}
ListNode newHead = head.next;
head.next = null;
return newHead;
}
}
-----------------------------------------------------------------------------有始有终分割线----------------------------------------------------------------------------------