Add Partition Problem (#4182)

src/main/java/com/thealgorithms/dynamicprogramming/PartitionProblem.java

@@ -0,0 +1,40 @@
+/**
+ * @author Md Asif Joardar
+ *
+ * Description: The partition problem is a classic problem in computer science
+ * that asks whether a given set can be partitioned into two subsets such that
+ * the sum of elements in each subset is the same.
+ *
+ * Example:
+ * Consider nums = {1, 2, 3}
+ * We can split the array "nums" into two partitions, where each having a sum of 3.
+ * nums1 = {1, 2}
+ * nums2 = {3}
+ *
+ * The time complexity of the solution is O(n × sum) and requires O(n × sum) space
+ */
+
+package com.thealgorithms.dynamicprogramming;
+
+import java.util.Arrays;
+
+public class PartitionProblem {
+
+    /**
+     * Test if a set of integers can be partitioned into two subsets such that the sum of elements
+     * in each subset is the same.
+     *
+     * @param nums the array contains integers.
+     * @return {@code true} if two subset exists, otherwise {@code false}.
+     */
+    public static boolean partition(int[] nums)
+    {
+        // calculate the sum of all the elements in the array
+        int sum = Arrays.stream(nums).sum();
+
+        // it will return true if the sum is even and the array can be divided into two subarrays/subset with equal sum.
+        // and here i reuse the SubsetSum class from dynamic programming section to check if there is exists a
+        // subsetsum into nums[] array same as the given sum
+        return (sum & 1) == 0 && SubsetSum.subsetSum(nums, sum/2);
+    }
+}

src/main/java/com/thealgorithms/dynamicprogramming/SubsetSum.java

@@ -22,7 +22,7 @@ public class SubsetSum {
      * @param sum target sum of subset.
      * @return {@code true} if subset exists, otherwise {@code false}.
      */
-    private static boolean subsetSum(int[] arr, int sum) {
+    public static boolean subsetSum(int[] arr, int sum) {
         int n = arr.length;
         boolean[][] isSum = new boolean[n + 2][sum + 1];
 

src/test/java/com/thealgorithms/dynamicprogramming/PartitionProblemTest.java

@@ -0,0 +1,24 @@
+package com.thealgorithms.dynamicprogramming;
+
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class PartitionProblemTest {
+    @Test
+    public void testIfSumOfTheArrayIsOdd(){
+        assertFalse(PartitionProblem.partition(new int[]{1, 2, 2}));
+    }
+    @Test
+    public void testIfSizeOfTheArrayIsOne(){
+        assertFalse(PartitionProblem.partition(new int[]{2}));
+    }
+    @Test
+    public void testIfSumOfTheArrayIsEven1(){
+        assertTrue(PartitionProblem.partition(new int[]{1, 2, 3, 6}));
+    }
+    @Test
+    public void testIfSumOfTheArrayIsEven2(){
+        assertFalse(PartitionProblem.partition(new int[]{1, 2, 3, 8}));
+    }
+}