Problem: 2057B - Gorilla and the Exam
Language: java

Solution 1 (runtime: 858 ms):

import java.sql.SQLOutput;
import java.util.*;

public class Main {
    public static Scanner sc = new Scanner(System.in);

    public static void no() {
        System.out.println("NO");
        return;
    }

    public static void yes() {
        System.out.println("YES");
        return;
    }

    public static void array_input(int[] arr, int size) {
        for (int i = 0; i < size; i++) {
            arr[i] = sc.nextInt();
        }
    }

    public static void array_output(int[] arr) {
        for (int i = 0; i < arr.length; i++) {
            System.out.print(arr[i] + " ");
        }
    }

    public static void matrix_input(int[][] arr, int rows, int colm) {
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < colm; j++) {
                arr[i][j] = sc.nextInt();
            }
        }
    }

    public static void matrix_output(int[][] arr) {
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr[0].length; j++) {
                System.out.print(arr[i][j]);
            }
        }
    }

//main code starts from line 43


    // main code starts from here
    public static int mex(int[] arr) {
        Arrays.sort(arr);
        int ans = 0;
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] == ans) {
                ans++;
            }
        }
        return ans;
    }
    public static int isBalanced(int [] arr){
        int ans=0;
        if(arr.length==0 || arr.length==1){ans=1;}
        else{
            if(arr.length%2==0 && arr[arr.length/2]==1){
                ans=1;
            }
            else{
                if(arr[(arr.length/2)-1]==0 && arr[(arr.length/2)]==1){
                    ans=1;
                }
                else if(arr[(arr.length/2)+1]==1 && arr[(arr.length/2)]==0){
                    ans=1;
                }
            }
        }
        return ans;
    }


    public static void main(String[] args) {

        int t = sc.nextInt();
        while (t-- > 0) {
            int n = sc.nextInt();
            int k=sc.nextInt();
            int []a=new int[n];
            array_input(a,n);
            Arrays.sort(a);
            ArrayList<Integer> fq=new ArrayList<>();
            int count = 1;

            // Iterate through the list to count frequencies
            for (int i = 0; i < n; i++) {
                if (i + 1 < n && a[i]==a[i+1]){
                    count++;
                } else {
                    fq.add(count);
                    count = 1;
                }
            }
            Collections.sort(fq);
            
            int l=0;
            while (!fq.isEmpty()){
                if(k>=fq.get(l)){

                    k=k-fq.get(l);
                    fq.remove(l);

                }
                else break;
            }


            if(fq.isEmpty()){
                System.out.println("1");
            }
            else {
                System.out.println(fq.size());
            }





            


            


        }


    }
}



Solution 2 (runtime: 390 ms):
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.ArrayList;
import java.util.Scanner;

public class B_Gorilla_And_The_Exam {

	public static void main(String[] args) {
		Scanner input = new Scanner(System.in);
		int t = input.nextInt();
		for(int test = 0; test < t; test++) {
			int n = input.nextInt();
			int k = input.nextInt();
			HashMap<Integer, Integer> counts = new HashMap<Integer, Integer>();
			for(int i = 0; i < n; i++) {
				int a = input.nextInt();
				int count = counts.getOrDefault(a, 0);
				counts.put(a, count + 1);
			}
			List<Integer> sortedCounts = new ArrayList<Integer>(counts.values());
			Collections.sort(sortedCounts);
			int index = 0;
			while(index < sortedCounts.size() && k >= sortedCounts.get(index)) {
				k -= sortedCounts.get(index);
				index++;
			}
			System.out.println(Math.max(1, sortedCounts.size() - index));
		}
		input.close();
	}

}