Check if the frequency of all the digits in a number is same in Python

Suppose we have a number num we have to check whether is balanced or not. A number is balanced when frequencies of all digits are same or not.

So, if the input is like num = 562256, then the output will be True as frequency of each digit is 2.

To solve this, we will follow these steps −

• number := convert num as string
• freq := a map containing frequencies of digits of number
• freq_values := make a new set by taking all digit frequency values from number
• if size of freq_values is same as 1, then
  • return True
• return False

Let us see the following implementation to get better understanding −

Example Code

Live Demo

from collections import defaultdict

def solve(num):
   number = str(num)
 
   freq = defaultdict(int)
   n = len(number)
 
   for i in range(n):
      freq[int(number[i])] += 1
 
   freq_values = set(freq.values())
 
   if len(freq_values) == 1:
      return True
   return False
   
num = 562256
print(solve(num))

Input

562256

Output

True

Arnab Chakraborty

Updated on: 15-Jan-2021

1K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started