CodingExercise: Rotate List
Medium
Topics
Companies
Given the head of a linked list, rotate the list to the right by k places.
Example 1:
Input: head = [1,2,3,4,5], k = 2
Output: [4,5,1,2,3]
Example 2:
Input: head = [0,1,2], k = 4
Output: [2,0,1]
Constraints:
The number of nodes in the list is in the range [0, 500].
-100 <= Node.val <= 100
0 <= k <= 2 * 109
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode rotateRight(ListNode head, int k) {
if (head == null || k == 0) return head;
Listhead current = head;
int n = 0;
while (current){
n++;
current = current.next;
}
if (k > n) { k = k % n;}
int offset = n - k;
current = head;
if (offset > 0) {
while (offset > 1) {
current = current.next;
offset--;
}
ListNode remaining = current.next;
current.next = null;
ListHead end = remaning;
while (end && end.next) {
end = end.next;
}
end.next = head;
return remaining;
}
return head;
}
}
No comments:
Post a Comment