Created OrganisingContainersOfBalls.java

java_codes/OrganisingContainersOfBalls.java

@@ -0,0 +1,93 @@
+import java.io.*;
+import java.math.*;
+import java.security.*;
+import java.text.*;
+import java.util.*;
+import java.util.concurrent.*;
+import java.util.function.*;
+import java.util.regex.*;
+import java.util.stream.*;
+import static java.util.stream.Collectors.joining;
+import static java.util.stream.Collectors.toList;
+
+class Result {
+
+    /*
+     * Complete the 'organizingContainers' function below.
+     *
+     * The function is expected to return a STRING.
+     * The function accepts 2D_INTEGER_ARRAY container as parameter.
+     */
+
+    public static String organizingContainers(List<List<Integer>> container) {
+    // Write your code here
+        int len = container.size();
+        int[] count = new int[len];
+        int[] space = new int[len];
+
+        for(int i=0;i<len;i++){
+            int sum =0 ;
+            for(int j=0;j<len;j++){
+                sum+=container.get(i).get(j);
+                count[j] += container.get(i).get(j);
+            }
+            space[i] = sum;
+        }
+
+        for(int i=0;i<len;i++){
+            boolean check = false;
+            for(int j=0;j<len;j++){
+                if(count[i] == space[j]){
+                    check = true;
+                    space[j] = 0;
+                    break;
+                }
+            }
+            if(!check){
+                return "Impossible";
+            }
+        }
+
+        return "Possible";
+    }
+
+}
+
+public class OrganisingContainersOfBalls{
+    public static void main(String[] args) throws IOException {
+        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
+        BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter(System.getenv("OUTPUT_PATH")));
+
+        int q = Integer.parseInt(bufferedReader.readLine().trim());
+
+        IntStream.range(0, q).forEach(qItr -> {
+            try {
+                int n = Integer.parseInt(bufferedReader.readLine().trim());
+
+                List<List<Integer>> container = new ArrayList<>();
+
+                IntStream.range(0, n).forEach(i -> {
+                    try {
+                        container.add(
+                            Stream.of(bufferedReader.readLine().replaceAll("\\s+$", "").split(" "))
+                                .map(Integer::parseInt)
+                                .collect(toList())
+                        );
+                    } catch (IOException ex) {
+                        throw new RuntimeException(ex);
+                    }
+                });
+
+                String result = Result.organizingContainers(container);
+
+                bufferedWriter.write(result);
+                bufferedWriter.newLine();
+            } catch (IOException ex) {
+                throw new RuntimeException(ex);
+            }
+        });
+
+        bufferedReader.close();
+        bufferedWriter.close();
+    }
+}