Kiểm tra số thập phân chỉ chứa chữ số 0 và 1 trong Python

Cho một số num. Kiểm tra xem các chữ số của nó có chỉ gồm 0 và 1 hay không.

Ví dụ: 101101 → True.

Giải thuật

• Tạo tập hợp chứa các chữ số của num.
• Loại bỏ 0 và 1 khỏi tập hợp.
• Nếu tập hợp rỗng → True, ngược lại False.

Mã ví dụ

def solve(num):
    digits_set = set()
    while num > 0:
        digit = num % 10
        digits_set.add(digit)
        num = int(num / 10)
    digits_set.discard(0)
    digits_set.discard(1)
    if len(digits_set) == 0:
        return True
    return False

num = 101101
print(solve(num))

Đầu vào

101101

Đầu ra

True