mirror of
https://github.com/TheAlgorithms/Python.git
synced 2025-04-06 22:05:54 +00:00
Update binary_count_trailing_zeros.py
The program prompts the user to enter a binary number. It checks if the input contains only 0s and 1s using a generator expression. If the input is valid, it calls the count_zeros_and_ones function and prints the results. If not, it notifies the user of the invalid input.
This commit is contained in:
parent
e9e7c96465
commit
f992b5eac6
@ -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.")
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user