Update rearranging_fruits.py

Author: Drag0nop

greedy_methods/rearranging_fruits.py

@@ -3,6 +3,35 @@
 
 
 def min_cost(basket1: List[int], basket2: List[int]) -> int:
+    """
+    Compute the minimum cost to make two baskets identical by swapping fruits.
+
+    Each fruit is represented by an integer value. The goal is to make both baskets
+    have the same multiset of elements with the minimum swap cost. Each swap cost
+    is defined as the minimum of the swapped fruit’s value and twice the smallest
+    fruit value in both baskets.
+
+    If it's impossible to make the baskets identical, return -1.
+
+    Args:
+        basket1 (List[int]): The first basket of fruits.
+        basket2 (List[int]): The second basket of fruits.
+
+    Returns:
+        int: The minimum total cost, or -1 if impossible.
+
+    Examples:
+        >>> min_cost([4, 2, 2, 2], [1, 4, 1, 2])
+        1
+        >>> min_cost([1, 2, 3, 4], [2, 3, 4, 1])
+        0
+        >>> min_cost([1, 1, 1, 1], [1, 1, 1, 1])
+        0
+        >>> min_cost([1, 2, 2], [2, 1, 1])
+        -1
+        >>> min_cost([5, 3, 3, 2], [2, 5, 5, 3])
+        -1
+    """
     n = len(basket1)
     freq: DefaultDict[int, int] = defaultdict(int)
     mn: float = float("inf")
@@ -21,22 +50,12 @@ def min_cost(basket1: List[int], basket2: List[int]) -> int:
     to_swap.sort()
     res: int = 0
     for i in range(len(to_swap) // 2):
-        # Ensure mn is treated as int during arithmetic
         res += min(to_swap[i], 2 * int(mn))
 
     return res
 
 
 if __name__ == "__main__":
-    # ---- Test Cases ----
-    test_cases = [
-        ([4, 2, 2, 2], [1, 4, 1, 2]),  # Expected: 1
-        ([1, 2, 3, 4], [2, 3, 4, 1]),  # Expected: 0
-        ([1, 1, 1, 1], [1, 1, 1, 1]),  # Expected: 0
-        ([1, 2, 2], [2, 1, 1]),  # Expected: -1
-        ([5, 3, 3, 2], [2, 5, 5, 3]),  # Expected: -1
-    ]
-
-    print("Running test cases...\n")
-    for b1, b2 in test_cases:
-        print(f"basket1 = {b1}, basket2 = {b2} → minCost = {min_cost(b1, b2)}")
+    import doctest
+    doctest.testmod()
+    print("All doctests passed.")