ashapochka
diff --git a/‎dynamic_programming/minimum_partition.py‎
Lines changed: 21 additions & 3 deletions b/‎dynamic_programming/minimum_partition.py‎
Lines changed: 21 additions & 3 deletions
@@ -3,21 +3,33 @@
 """
 
 
-def find_min(arr):
+def find_min(arr: list[int]) -> int:
+    """
+    >>> find_min([1, 2, 3, 4, 5])
+    1
+    >>> find_min([5, 5, 5, 5, 5])
+    5
+    >>> find_min([5, 5, 5, 5])
+    0
+    >>> find_min([3])
+    3
+    >>> find_min([])
+    0
+    """
     n = len(arr)
     s = sum(arr)
 
     dp = [[False for x in range(s + 1)] for y in range(n + 1)]
 
-    for i in range(1, n + 1):
+    for i in range(n + 1):
         dp[i][0] = True
 
     for i in range(1, s + 1):
         dp[0][i] = False
 
     for i in range(1, n + 1):
         for j in range(1, s + 1):
-            dp[i][j] = dp[i][j - 1]
+            dp[i][j] = dp[i - 1][j]
 
             if arr[i - 1] <= j:
                 dp[i][j] = dp[i][j] or dp[i - 1][j - arr[i - 1]]
@@ -28,3 +40,9 @@ def find_min(arr):
             break
 
     return diff
+
+
+if __name__ == "__main__":
+    from doctest import testmod
+
+    testmod()