题目:旋转链表(Java)
内容
链接: 旋转链表
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
用例
- 示例 1:
输入: 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
- 示例 2:
输入: 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
思路
这个旋转的过程,其实就是先构造一个环,然后在某一个位置断开环,返回结果。例如示例2,先连成环,再移动到2,断开1和2,并把1的next指向NULL,返回节点2即可。观察示例2,发现当输入k值比链表本身长度还要大的时候,可以发现一个规律,移动n(0<= n< len,len为链表长度)和n+m*len(m为正整数)的效果是一样的。
AC代码
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode rotateRight(ListNode head, int k) {
if (head == null) {
return null;
}
ListNode phead = new ListNode(-1);
ListNode resNode = null;
phead.next = head;
int len = 0;
//获取链表长度以及最后一位ListNode
while (phead.next != null) {
len++;
phead = phead.next;
}
//计算需要移动的步数
int stepNum = len - k % len;
//连接成环,已经移动了一步
phead.next = head;
phead = phead.next;
//再移动stepNum - 1步
for (int i = 0; i < stepNum - 1; i++) {
phead = phead.next;
}
//断开链表,返回新的头结点
resNode = phead.next;
phead.next = null;
return resNode;
}
}