from __future__ import annotations
from dataclasses import dataclass
from typing import Any
@dataclass
class Node:
item: Any
next: Node | Any = None
class LinkedList:
def __init__(self) -> None:
self.head: Node | None = None
self.size = 0
def __str__(self) -> str:
"""
>>> linked_list = LinkedList()
>>> linked_list.add(23)
>>> linked_list.add(14)
>>> linked_list.add(9)
>>> print(linked_list)
9 --> 14 --> 23
"""
iterate = self.head
item_str = ""
item_list: list[str] = []
while iterate:
item_list.append(str(iterate.item))
iterate = iterate.next
item_str = " --> ".join(item_list)
return item_str
def __len__(self) -> int:
"""
>>> linked_list = LinkedList()
>>> len(linked_list)
0
>>> linked_list.add("a")
>>> len(linked_list)
1
>>> linked_list.add("b")
>>> len(linked_list)
2
"""
return self.size
def add(self, item: Any, position: int = 0) -> None:
"""
Add an item to the LinkedList at the specified position.
Default position is 0 (the head).
Args:
item (Any): The item to add to the LinkedList.
position (int, optional): The position at which to add the item.
Defaults to 0.
Raises:
ValueError: If the position is negative or out of bounds.
>>> linked_list = LinkedList()
>>> linked_list.add(1)
>>> linked_list.add(2)
>>> linked_list.add(3)
>>> linked_list.add(4, 2)
>>> print(linked_list)
3 --> 2 --> 4 --> 1
# Test adding to a negative position
>>> linked_list.add(5, -3)
Traceback (most recent call last):
...
ValueError: Position must be non-negative
# Test adding to an out-of-bounds position
>>> linked_list.add(5,7)
Traceback (most recent call last):
...
ValueError: Out of bounds
>>> linked_list.add(5, 4)
>>> print(linked_list)
3 --> 2 --> 4 --> 1 --> 5
"""
if position < 0:
raise ValueError("Position must be non-negative")
if position == 0 or self.head is None:
new_node = Node(item, self.head)
self.head = new_node
else:
current = self.head
for _ in range(position - 1):
current = current.next
if current is None:
raise ValueError("Out of bounds")
new_node = Node(item, current.next)
current.next = new_node
self.size += 1
def partition_liked_list(self, value: int) -> None:
"""
Partition the linked list based on node elements in order.
All nodes with elements less than value should occur on the left,
while those greater than or equal to value should occur on the right.
>>> linked_list = LinkedList()
>>> linked_list.add(1)
>>> linked_list.add(2)
>>> linked_list.add(3)
>>> linked_list.add(4, 2)
>>> linked_list.add(5, 4)
>>> print(linked_list)
3 --> 2 --> 4 --> 1 --> 5
>>> linked_list.partition_liked_list(3)
>>> print(linked_list)
2 --> 1 --> 3 --> 4 --> 5
>>> linked_list = LinkedList()
>>> linked_list.add(1)
>>> linked_list.add(2)
>>> linked_list.add(3)
>>> print(linked_list)
3 --> 2 --> 1
>>> linked_list.partition_liked_list(3)
>>> print(linked_list)
2 --> 1 --> 3
>>> linked_list = LinkedList()
>>> linked_list.add(5)
>>> linked_list.add(5)
>>> linked_list.add(5)
>>> print(linked_list)
5 --> 5 --> 5
>>> linked_list.partition_liked_list(5)
>>> print(linked_list)
5 --> 5 --> 5
>>> linked_list = LinkedList()
>>> linked_list.add(1, 0)
>>> linked_list.add(2, 1)
>>> linked_list.add(3, 2)
>>> linked_list.add(4, 3)
>>> linked_list.add(5, 4)
>>> print(linked_list)
1 --> 2 --> 3 --> 4 --> 5
>>> linked_list.partition_liked_list(6)
>>> print(linked_list)
1 --> 2 --> 3 --> 4 --> 5
>>> linked_list = LinkedList()
>>> linked_list.add(1)
>>> print(linked_list)
1
>>> linked_list.partition_liked_list(1)
>>> print(linked_list)
1
"""
if self.head is None:
return
less_nodes, greater_nodes = Node(0), Node(0)
current, current_less, current_greater = self.head, less_nodes, greater_nodes
while current is not None:
if current.item < value:
current_less.next = current
current_less = current_less.next
else:
current_greater.next = current
current_greater = current_greater.next
current = current.next
current_less.next = greater_nodes.next
current_greater.next = None
self.head = less_nodes.next