Reverse Nodes in k-Group
Given a linked list, reverse the nodes of a linked list k at a time and return its modified list.
k is a positive integer and is less than or equal to the length of the linked list. If the number of nodes is not a multiple of k then left-out nodes in the end should remain as it is.
Example:
Given this linked list: 1->2->3->4->5
For k = 2, you should return: 2->1->4->3->5
For k = 3, you should return: 3->2->1->4->5
Note:
Only constant extra memory is allowed.
You may not alter the values in the list's nodes, only nodes itself may be changed.
class Solution {
public ListNode reverse(ListNode head) {
ListNode prev = null, current = head;
while (current != null) {
ListNode temp = current.next;
current.next = prev;
prev = current;
current = temp;
}
return prev;
}
public ListNode reverseKGroup(ListNode head, int k) {
if (head == null || head.next == null || k == 1)
return head;
ListNode previous = null, current = head;
int index = 1;
while (current != null) {
if (index == k) {
ListNode next = current.next;
current.next = null;
if (previous == null) {
previous = head;
head = reverse(head);
previous.next = next;
} else {
ListNode temp = previous.next;
previous.next = reverse(previous.next);
temp.next = next;
previous = temp;
}
current = next;
index = 1;
continue;
}
current = current.next;
index++;
}
return head;
}
}
Last updated