题目
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例:
输入: 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
注意点
1、该题思路是将链表处理成环,然后计算出结果链表的尾节点和头节点,最后在尾节点和头节点的中间截断;
2、size - k % size - 1为所求结果为节点的距离头节点的长度;k % size是为了确定移动的次数;(如size=3,k=4时,其实跟k=1是一样)
实现
public ListNode rotateRight(ListNode head, int k) {
//链表为null或链表只有一个节点,直接返回
if(head == null || head.next == null) return head;
ListNode tail = head; //记录链表尾节点
int size = 1; //记录链表的长度
//获取链表的尾节点和长度
while(tail.next != null){
tail = tail.next;
size ++;
}
//尾节点连向头节点,成环
tail.next = head;
ListNode newTail = head; //记录所求结果的尾节点
//获取所求结果的尾节点
for(int i = size - k % size - 1; i > 0; i --){
newTail = newTail.next;
}
//头节点指向尾节点的下一个节点(环)
head = newTail.next;
//尾节点的下一点置空(截断)
newTail.next = null;
return head;
}