diff --git a/bit_manipulation/binary_count_trailing_zeros.py b/bit_manipulation/binary_count_trailing_zeros.py index f401c4ab9266..1168521d4f3d 100644 --- a/bit_manipulation/binary_count_trailing_zeros.py +++ b/bit_manipulation/binary_count_trailing_zeros.py @@ -42,3 +42,20 @@ def binary_count_trailing_zeros(a: int) -> int: import doctest doctest.testmod() + + +# counting number of 0s and 1s in a binary number +def count_zeros_and_ones(binary_number): + # Convert the binary number to a string if it's not already + binary_str = str(binary_number) + + count_zeros = binary_str.count("0") + count_ones = binary_str.count("1") + + return count_zeros, count_ones + + +# Example usage +binary_number = 1011001 +zeros, ones = count_zeros_and_ones(binary_number) +print(f"Number of 0s: {zeros}, Number of 1s: {ones}")