Create K Largest Elements.java

Java/K Largest Elements.java

@@ -0,0 +1,33 @@
+import java.io.*;
+import java.util.*;
+
+public class Main {
+
+   public static void main(String[] args) throws Exception {
+      BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+      int n = Integer.parseInt(br.readLine());
+      int[] arr = new int[n];
+
+      for (int i = 0; i < n; i++) {
+         arr[i] = Integer.parseInt(br.readLine());
+      }
+
+      int k = Integer.parseInt(br.readLine());
+      // write your code here
+      PriorityQueue<Integer> pq = new PriorityQueue<>();
+      for(int i=0;i<arr.length;i++){
+          if(i <k){
+              pq.add(arr[i]);
+          }else{
+              if(arr[i] > pq.peek()){
+                  pq.remove();
+                  pq.add(arr[i]);
+              }
+          }
+      }
+      while(pq.size() >0){
+          System.out.println(pq.remove());
+      }
+    }
+
+}