Create Question2.java

DSA/RatnaKhanna_590013141/Day_47/Question2.java

@@ -0,0 +1,57 @@
+import java.util.*;
+public class Main {
+    public static int kthElement(int[] A, int[] B, int k) {
+        int n = A.length;
+        int m = B.length;
+        if (n > m) {
+            return kthElement(B, A, k);
+        }
+        int low = Math.max(0, k - m);
+        int high = Math.min(k, n);
+        while (low <= high) {
+            int cutA = (low + high) / 2;
+            int cutB = k - cutA;
+            int leftA = (cutA == 0) ? Integer.MIN_VALUE : A[cutA - 1];
+            int leftB = (cutB == 0) ? Integer.MIN_VALUE : B[cutB - 1];
+            int rightA = (cutA == n) ? Integer.MAX_VALUE : A[cutA];
+            int rightB = (cutB == m) ? Integer.MAX_VALUE : B[cutB];
+            if (leftA <= rightB && leftB <= rightA) {
+                return Math.max(leftA, leftB);
+            } else if (leftA > rightB) {
+                high = cutA - 1;
+            } else {
+                low = cutA + 1;
+            }
+        }
+        return -1;
+    }
+
+    public static void main(String[] args) {
+        Scanner sc = new Scanner(System.in);
+
+        // Input first array
+        int n = sc.nextInt();
+        int[] A = new int[n];
+        for (int i = 0; i < n; i++) {
+            A[i] = sc.nextInt();
+        }
+
+        // Input second array
+        int m = sc.nextInt();
+        int[] B = new int[m];
+        for (int i = 0; i < m; i++) {
+            B[i] = sc.nextInt();
+        }
+
+        // Input k
+        int k = sc.nextInt();
+
+        if (k < 1 || k > n + m) {
+            System.out.println("Invalid k");
+        } else {
+            System.out.println(kthElement(A, B, k));
+        }
+
+        sc.close();
+    }
+}