难度:【中等】
给定一个链表,旋转链表,将链表每个节点向右移动 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
解题思路:
- 先找到旧的尾部并将其与链表头相连
cur.next = head
,整个链表闭合成环,同时计算出链表的长度len
。 - 找到新的尾部,第 (
len - k % len - 1
) 个节点 ,新的链表头是第 (len- k % len
个节点。 - 断开环
newRear.next = null
,并返回新的链表头newHead
。
代码实现:
/**
* 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 || head.next == null){
return head;
}
ListNode cur = head;
int len = 1;
while(cur.next!=null){
len++;
cur = cur.next;
}
cur.next = head;//找到旧的尾部并将其与链表头相连
ListNode newRear = head;
int i = 0;
while(i<(len-k%len-1)){//找到新的尾部
newRear = newRear.next;
i++;
}
ListNode newHead = newRear.next;
newRear.next = null;
return newHead;
}
}
复杂度分析
- 时间复杂度:O(N),其中 N是链表中的元素个数
- 空间复杂度:O(1)