Open In App

Distributing all balls without repetition

Improve
Improve
Like Article
Like
Save
Share
Report

Given N balls. For convenience, we denote the color of each ball as — lowercase letters. We have to distribute N balls among K people. They will be upset if they get two balls of the same color. We can give any number of balls to people and they won’t be upset even if they do not get any balls, but, we have to distribute all the balls, such that no one will be upset — print YES, if it is possible, and NO, otherwise.

Examples: 

Input : 4 2 // value of N and K
        aabb // colors of given balls
Output : YES
We can give 1st and 3rd ball to the first person,
and 2nd and 4th to the second.

Input : 6 3 // value of N and K
        aacaab // colors of given balls
Output : NO
We need to give all balls of color a, but one 
ball will stay, that's why answer is NO 

The approach will be really simple. We will create a count array to keep the count of each color that occurs and then we will check if any color occurs more than the number of people we have. If it occurs, we will print NO else YES.

The implementation of the above idea is given below.  

C++





Java




// Java program to find if its possible
// to distribute balls without repetition
import java.io.*;
  
public class GFG {
  
    static int MAX_CHAR = 26;
      
    // function to find if its possible
    // to distribute balls or not
    static boolean distributingBalls(long k,
                         long n, String str)
    
          
        // count array to count how many
        // times each color has occurred
        int []a = new int[MAX_CHAR];
          
        for (int i = 0; i < n; i++)
        {
              
            // increasing count of each
            // color every time it appears
            a[str.charAt(i) - 'a']++; 
        }
          
        for (int i = 0; i < MAX_CHAR; i++) 
      
            // to check if any color appears 
            // more than K times if it does 
            // we will print NO
            if (a[i] > k) 
                return false;
      
        return true;
    }
      
    // Driver code
    static public void main (String[] args)
    {
        long n = 6, k = 3;
        String str = "aacaab";
      
        if (distributingBalls(k, n, str))
            System.out.println("YES");
        else
            System.out.println("NO");
    }
}
  
// This code is contributed by vt_m.


Python3




# Python3 program to find if its possible to 
# distribute balls without repetition
  
  
MAX_CHAR = 26
  
# function to find if its possible to 
# distribute balls or not 
def distributingBalls(k, n, string) :
  
    # count array to count how many times 
    # each color has occurred 
    a = [0] * MAX_CHAR
      
    for i in range(n) : 
        # increasing count of each color
        # every time it appears
        a[ord(string[i]) - ord('a')] += 1
      
    for i in range(MAX_CHAR) :
        # to check if any color appears
        # more than K times if it does
        # we will print NO
        if (a[i] > k) :
            return False 
      
    return True 
  
  
# Driver code 
if __name__ == "__main__" :
   
    n, k = 6, 3
    string = "aacaab" 
  
    if (distributingBalls(k, n, string)) :
        print("YES"
    else :
        print("NO"
  
# This code is contributed by Ryuga


C#




// C# program to find if its possible to
// distribute balls without repetition
using System;
  
public class GFG {
  
    static int MAX_CHAR = 26;
      
    // function to find if its possible
    // to distribute balls or not
    static bool distributingBalls(long k,
                       long n, string str)
    
          
        // count array to count how many
        // times each color has occurred
        int []a = new int[MAX_CHAR];
          
        for (int i = 0; i < n; i++)
        {
              
            // increasing count of each
            // color every time it appears
            a[str[i] - 'a']++; 
        }
          
        for (int i = 0; i < MAX_CHAR; i++) 
      
            // to check if any color
            // appears more than K
            // times if it does we
            // will print NO
            if (a[i] > k) 
                return false;
      
        return true;
    }
      
    // Driver code
    static public void Main ()
    {
        long n = 6, k = 3;
        string str = "aacaab";
      
        if (distributingBalls(k, n, str))
            Console.WriteLine("YES");
        else
            Console.WriteLine("NO");
    }
}
  
// This code is contributed by vt_m.


Javascript




<script>
    // Javascript program to find if its possible to
    // distribute balls without repetition
      
    let MAX_CHAR = 26;
        
    // function to find if its possible
    // to distribute balls or not
    function distributingBalls(k, n, str)
    
            
        // count array to count how many
        // times each color has occurred
        let a = new Array(MAX_CHAR);
        a.fill(0);
            
        for (let i = 0; i < n; i++)
        {
                
            // increasing count of each
            // color every time it appears
            a[str[i].charCodeAt() - 'a'.charCodeAt()]++; 
        }
            
        for (let i = 0; i < MAX_CHAR; i++) 
        
            // to check if any color
            // appears more than K
            // times if it does we
            // will print NO
            if (a[i] > k) 
                return false;
        
        return true;
    }
      
    let n = 6, k = 3;
    let str = "aacaab";
  
    if (distributingBalls(k, n, str))
      document.write("YES");
    else
      document.write("NO");
      
</script>


Output

NO

Time complexity : O(n) 
Auxiliary Space : O(1)

 



Last Updated : 16 Feb, 2023
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads