Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def solution(num_str):
- zeros = ones = 0
- for i in range(len(num_str)):
- num = num_str[i]
- if num == "0": zeros += 1
- else: ones += 1
- if ones>1 and zeros>1: break
- if zeros == 1 or ones == 1: print("YES")
- else: print("NO")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement