Skip to content

Commit 7552ecf

Browse files
BamaCharanChhandogipre-commit-ci[bot]cclauss
authored andcommitted
add reverse k group linkedlist (TheAlgorithms#9323)
* add reverse k group linkedlist * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update * Update reverse_k_group.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update reverse_k_group.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update reverse_k_group.py --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Christian Clauss <cclauss@me.com>
1 parent ce8fcec commit 7552ecf

File tree

1 file changed

+118
-0
lines changed

1 file changed

+118
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,118 @@
1+
from __future__ import annotations
2+
3+
from collections.abc import Iterable, Iterator
4+
from dataclasses import dataclass
5+
6+
7+
@dataclass
8+
class Node:
9+
data: int
10+
next_node: Node | None = None
11+
12+
13+
class LinkedList:
14+
def __init__(self, ints: Iterable[int]) -> None:
15+
self.head: Node | None = None
16+
for i in ints:
17+
self.append(i)
18+
19+
def __iter__(self) -> Iterator[int]:
20+
"""
21+
>>> ints = []
22+
>>> list(LinkedList(ints)) == ints
23+
True
24+
>>> ints = tuple(range(5))
25+
>>> tuple(LinkedList(ints)) == ints
26+
True
27+
"""
28+
node = self.head
29+
while node:
30+
yield node.data
31+
node = node.next_node
32+
33+
def __len__(self) -> int:
34+
"""
35+
>>> for i in range(3):
36+
... len(LinkedList(range(i))) == i
37+
True
38+
True
39+
True
40+
>>> len(LinkedList("abcdefgh"))
41+
8
42+
"""
43+
return sum(1 for _ in self)
44+
45+
def __str__(self) -> str:
46+
"""
47+
>>> str(LinkedList([]))
48+
''
49+
>>> str(LinkedList(range(5)))
50+
'0 -> 1 -> 2 -> 3 -> 4'
51+
"""
52+
return " -> ".join([str(node) for node in self])
53+
54+
def append(self, data: int) -> None:
55+
"""
56+
>>> ll = LinkedList([1, 2])
57+
>>> tuple(ll)
58+
(1, 2)
59+
>>> ll.append(3)
60+
>>> tuple(ll)
61+
(1, 2, 3)
62+
>>> ll.append(4)
63+
>>> tuple(ll)
64+
(1, 2, 3, 4)
65+
>>> len(ll)
66+
4
67+
"""
68+
if not self.head:
69+
self.head = Node(data)
70+
return
71+
node = self.head
72+
while node.next_node:
73+
node = node.next_node
74+
node.next_node = Node(data)
75+
76+
def reverse_k_nodes(self, group_size: int) -> None:
77+
"""
78+
reverse nodes within groups of size k
79+
>>> ll = LinkedList([1, 2, 3, 4, 5])
80+
>>> ll.reverse_k_nodes(2)
81+
>>> tuple(ll)
82+
(2, 1, 4, 3, 5)
83+
>>> str(ll)
84+
'2 -> 1 -> 4 -> 3 -> 5'
85+
"""
86+
if self.head is None or self.head.next_node is None:
87+
return
88+
89+
length = len(self)
90+
dummy_head = Node(0)
91+
dummy_head.next_node = self.head
92+
previous_node = dummy_head
93+
94+
while length >= group_size:
95+
current_node = previous_node.next_node
96+
assert current_node
97+
next_node = current_node.next_node
98+
for _ in range(1, group_size):
99+
assert next_node, current_node
100+
current_node.next_node = next_node.next_node
101+
assert previous_node
102+
next_node.next_node = previous_node.next_node
103+
previous_node.next_node = next_node
104+
next_node = current_node.next_node
105+
previous_node = current_node
106+
length -= group_size
107+
self.head = dummy_head.next_node
108+
109+
110+
if __name__ == "__main__":
111+
import doctest
112+
113+
doctest.testmod()
114+
ll = LinkedList([1, 2, 3, 4, 5])
115+
print(f"Original Linked List: {ll}")
116+
k = 2
117+
ll.reverse_k_nodes(k)
118+
print(f"After reversing groups of size {k}: {ll}")

0 commit comments

Comments
 (0)