forked from PawanJaiswal08/leetcode-solutions
-
Notifications
You must be signed in to change notification settings - Fork 22
/
Copy path25. Reverse Nodes in k-Group.cpp
53 lines (46 loc) · 1.22 KB
/
25. Reverse Nodes in k-Group.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
class Solution {
public:
void reverse(ListNode *start, ListNode *end)
{
ListNode *prev = NULL;
ListNode *curr = start;
ListNode *nexty = start->next;
while(prev!=end)
{
curr->next = prev;
prev = curr;
curr = nexty;
if(nexty!=NULL)
{
nexty = nexty->next;
}
}
}
ListNode* reverseKGroup(ListNode* head, int k) {
if(head==NULL || head->next == NULL || k==1)
return head;
ListNode *dummy = new ListNode(-1);
dummy->next = head;
ListNode *beforeStart = dummy, *end = head;
int i=0;
while(end!=NULL)
{
i++;
if(i%k==0)
{
//reversal
ListNode *start = beforeStart->next, *temp = end->next;
reverse(start,end);
beforeStart->next = end;
start->next = temp;
beforeStart = start;
end = temp;
}
else
{
end = end->next;
}
}
return dummy->next;
}
};