ตรวจสอบว่าตัวเลขประกอบด้วยหลัก 0 และ 1 เท่านั้น ใน Python

ปัญหา

ให้ตัวเลข num ตรวจสอบว่าประกอบด้วยหลัก 0 และ 1 เท่านั้นหรือไม่

ตัวอย่าง

num = 101101 → True

แนวคิดการแก้ปัญหา

1. สร้างเซตของหลักทั้งหมดในตัวเลข
2. ลบ 0 และ 1 ออกจากเซต
3. ถ้าเซตว่าง → ประกอบด้วย 0,1 เท่านั้น → True

โค้ดตัวอย่าง

def solve(num):
    digits = set()
    while num > 0:
        digits.add(num % 10)
        num //= 10
    digits.discard(0)
    digits.discard(1)
    return len(digits) == 0

print(solve(101101))  # True

อินพุต / เอาต์พุต

Input:  101101
Output: True