diff --git a/bit_manipulation/binary_count_trailing_zeros.py b/bit_manipulation/binary_count_trailing_zeros.py index f401c4ab9..702672e07 100644 --- a/bit_manipulation/binary_count_trailing_zeros.py +++ b/bit_manipulation/binary_count_trailing_zeros.py @@ -42,3 +42,25 @@ if __name__ == "__main__": 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 + +# Get user input +binary_number = input("Enter a binary number: ") + +# Validate input +if all(bit in '01' for bit in binary_number): + zeros, ones = count_zeros_and_ones(binary_number) + print(f"Number of 0s: {zeros}, Number of 1s: {ones}") +else: + print("Invalid input! Please enter a valid binary number.") +