forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution.cs
52 lines (51 loc) · 1.38 KB
/
Solution.cs
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
public class Solution {
public ListNode ReverseKGroup(ListNode head, int k) {
if (k < 2) return head;
ListNode newHead = null;
ListNode newTail = null;
var current = head;
while (current != null)
{
ListNode segmentHead = null;
ListNode segmentTail = null;
var count = 0;
while (current != null && count < k)
{
if (segmentHead == null) segmentHead = current;
segmentTail = current;
current = current.next;
++count;
}
segmentTail.next = null;
if (count == k)
{
segmentTail = segmentHead;
segmentHead = ReverseList(segmentHead);
}
if (newHead == null)
{
newHead = segmentHead;
newTail = segmentTail;
}
else
{
newTail.next = segmentHead;
newTail = segmentTail;
}
}
return newHead;
}
private ListNode ReverseList(ListNode head)
{
var current = head;
head = null;
while (current != null)
{
var next = current.next;
current.next = head;
head = current;
current = next;
}
return head;
}
}