add a new reverse-linked-list

data_structures/arrays/median_of_two_sorted_arrays.py

@@ -0,0 +1,61 @@
+"""
+https://www.enjoyalgorithms.com/blog/median-of-two-sorted-arrays
+"""
+
+
+def find_median_sorted_arrays(nums1: list[int], nums2: list[int]) -> float:
+    """
+    Find the median of two arrays.
+
+    Args:
+        nums1: The first array.
+        nums2: The second array.
+
+    Returns:
+    The median of the two arrays.
+
+    Examples:
+        >>> find_median_sorted_arrays([1, 3], [2])
+        2.0
+
+        >>> find_median_sorted_arrays([1, 2], [3, 4])
+        2.5
+
+        >>> find_median_sorted_arrays([0, 0], [0, 0])
+        0.0
+
+        >>> find_median_sorted_arrays([], [])
+        Traceback (most recent call last):
+            ...
+        ValueError: Both input arrays are empty.
+
+        >>> find_median_sorted_arrays([], [1])
+        1.0
+
+        >>> find_median_sorted_arrays([-1000], [1000])
+        0.0
+
+        >>> find_median_sorted_arrays([-1.1, -2.2], [-3.3, -4.4])
+        -2.75
+    """
+    if not nums1 and not nums2:
+        raise ValueError("Both input arrays are empty.")
+
+    # Merge the arrays into a single sorted array.
+    merged = sorted(nums1 + nums2)
+    total = len(merged)
+
+    if total % 2 == 1:  # If the total number of elements is odd
+        return float(merged[total // 2])  # then return the middle element
+
+    # If the total number of elements is even, calculate
+    # the average of the two middle elements as the median.
+    middle1 = merged[total // 2 - 1]
+    middle2 = merged[total // 2]
+    return (float(middle1) + float(middle2)) / 2.0
+
+
+if __name__ == "__main__":
+    import doctest
+
+    doctest.testmod()

data_structures/hashing/number_theory/median_of_two_sorted_arrays.py

@@ -0,0 +1,59 @@
+"""
+https://www.enjoyalgorithms.com/blog/median-of-two-sorted-arrays
+"""
+
+def find_median_sorted_arrays(nums1: list[int], nums2: list[int]) -> float:
+    """
+    Find the median of two arrays.
+
+    Args:
+        nums1: The first array.
+        nums2: The second array.
+
+    Returns:
+    The median of the two arrays.
+
+    Examples:
+        >>> find_median_sorted_arrays([1, 3], [2])
+        2.0
+
+        >>> find_median_sorted_arrays([1, 2], [3, 4])
+        2.5
+
+        >>> find_median_sorted_arrays([0, 0], [0, 0])
+        0.0
+
+        >>> find_median_sorted_arrays([], [])
+        Traceback (most recent call last):
+            ...
+        ValueError: Both input arrays are empty.
+
+        >>> find_median_sorted_arrays([], [1])
+        1.0
+
+        >>> find_median_sorted_arrays([-1000], [1000])
+        0.0
+
+        >>> find_median_sorted_arrays([-1.1, -2.2], [-3.3, -4.4])
+        -2.75
+    """
+    if not nums1 and not nums2:
+        raise ValueError("Both input arrays are empty.")
+
+    # Merge the arrays into a single sorted array.
+    merged = sorted(nums1 + nums2)
+    total = len(merged)
+
+    if total % 2 == 1:  # If the total number of elements is odd
+        return float(merged[total // 2])  # then return the middle element
+
+    # If the total number of elements is even, calculate
+    # the average of the two middle elements as the median.
+    middle1 = merged[total // 2 - 1]
+    middle2 = merged[total // 2]
+    return (float(middle1) + float(middle2)) / 2.0
+
+
+if __name__ == "__main__":
+    import doctest
+    doctest.testmod()

data_structures/linked_list/reverse_linked_list.py

@@ -0,0 +1,58 @@
+"""
+https://www.enjoyalgorithms.com/blog/reverse-linked-list
+"""
+
+
+class ListNode:
+    """Definition for singly-linked list."""
+
+    def __init__(self, val=0, next=None):
+        self.val = val
+        self.next = next
+
+
+def reverse_linked_list(head: ListNode) -> ListNode:
+    """
+    Reverse a singly linked list.
+
+    Args:
+        head: The head node of the linked list.
+
+    Returns:
+        The new head node of the reversed linked list.
+
+    Examples:
+        >>> a = ListNode(1)
+        >>> b = ListNode(2)
+        >>> c = ListNode(3)
+        >>> a.next, b.next = b, c
+        >>> head = reverse_linked_list(a)
+        >>> [head.val, head.next.val, head.next.next.val]
+        [3, 2, 1]
+    """
+    prev = None
+    current = head
+    while current:
+        nxt = current.next
+        current.next = prev
+        prev = current
+        current = nxt
+    return prev
+
+
+if __name__ == "__main__":
+    import doctest
+
+    doctest.testmod()
+
+    # Example execution
+    a = ListNode(1)
+    b = ListNode(2)
+    c = ListNode(3)
+    a.next, b.next = b, c
+
+    print("Original Linked List: 1 -> 2 -> 3")
+    new_head = reverse_linked_list(a)
+    print(
+        f"Reversed Linked List: {new_head.val} -> {new_head.next.val} -> {new_head.next.next.val}"
+    )