Maximum number of elements greater than X after equally distributing subset of array

• Last Updated : 05 May, 2021

Given an array, arr[] and an integer X, the task is to count the number of elements greater than X after equally dividing the subset of elements. That is each element of the subset will be equal to Examples:

Input: arr[] = {5, 1, 2, 1}, X = 3
Output:
Explanation:
Subset which is equally distributes is {5, 2}.
After which the elements will be 3.5 each.
Array => {3.5, 1, 3.5, 1}
Total number of elements greater than X = 2

Input: arr[] = {3, 4, 5}, X = 6
Output:
Explanation:
There is no way to distribute any subset of array to make the elements greater than 6.

Approach: The idea is to sort the array and include the largest elements of the array such that their average is greater than or equal to X. Count of such elements whose average is greater than or equal to X is the desired subset which can be equally divided and each element is greater than X.

Below is the implementation of the above approach:

C++

 // C++ implementation to find the// maximum number of elements greater// than X by equally distributing #include using namespace std; // Function to find the// maximum number of elements greater// than X by equally distributingvoid redistribute(int arr[], int n, int x){    // Sorting the array    sort(arr, arr + n, greater());     int i, sum = 0;     // Loop to iterate over the elements    // of the array    for (i = 0; i < n; i++) {        sum += arr[i];         // If no more elements can        // become larger than x        if (sum / (i + 1) < x) {            cout << i << endl;            break;        }    }    if (i == n)        cout << n << endl;} // Driver Codeint main(){    int arr[] = { 5, 1, 2, 1 };    int x = 3;    redistribute(arr, 4, x);    return 0;}

Java

 // Java implementation to find the// maximum number of elements greater// than X by equally distributingimport java.util.*; class GFG{ // Function to find the maximum// number of elements greater// than X by equally distributingstatic void redistribute(Integer arr[], int n,                                        int x){         // Sorting the array    Arrays.sort(arr, Collections.reverseOrder());     int i, sum = 0;     // Loop to iterate over the elements    // of the array    for(i = 0; i < n; i++)    {       sum += arr[i];               // If no more elements can       // become larger than x       if (sum / (i + 1) < x)       {           System.out.print(i + "\n");           break;       }    }    if (i == n)        System.out.print(n + "\n");} // Driver Codepublic static void main(String[] args){    Integer arr[] = { 5, 1, 2, 1 };    int x = 3;         redistribute(arr, 4, x);}} // This code is contributed by Rajput-Ji

Python3

 # Python3 implementation to find the# maximum number of elements greater# than X by equally distributing # Function to find the# maximum number of elements greater# than X by equally distributingdef redistribute(arr, n, x):     # Sorting the array    arr.sort(reverse = True)     sum = 0     # Loop to iterate over the    # elements of the array    for i in range(n):        sum += arr[i]         # If no more elements can        # become larger than x        if (sum / (i + 1) < x):            print(i)            break             if (i == n):        print(n) # Driver Codearr = [ 5, 1, 2, 1 ]x = 3 # Function callredistribute(arr, 4, x) # This code is contributed by Vishal Maurya.

C#

 // C# implementation to find the// maximum number of elements greater// than X by equally distributingusing System; class GFG{ // Function to find the maximum// number of elements greater// than X by equally distributingstatic void redistribute(int []arr, int n,                                    int x){         // Sorting the array    Array.Sort(arr);    Array.Reverse(arr);     int i, sum = 0;     // Loop to iterate over the elements    // of the array    for(i = 0; i < n; i++)    {       sum += arr[i];               // If no more elements can       // become larger than x       if (sum / (i + 1) < x)       {           Console.Write(i + "\n");           break;       }    }    if (i == n)        Console.Write(n + "\n");} // Driver Codepublic static void Main(String[] args){    int []arr = { 5, 1, 2, 1 };    int x = 3;         redistribute(arr, 4, x);}} // This code is contributed by Rajput-Ji

Javascript


Output:
2

My Personal Notes arrow_drop_up