백준 11723 집합

https://www.acmicpc.net/problem/11723

11723번: 집합

 

import sys
 
 
if __name__ == '__main__':
    m = int(sys.stdin.readline())
    s = set()
    for i in range(m):
        operation = list(sys.stdin.readline().strip().split())
        if len(operation) == 1:
            if operation[0] == 'all':
                s = set([i for i in range(1, 21)])
            else:
                s = set()
        else:
            if operation[0] == 'add':
                s.add(int(operation[1]))
            elif operation[0] == 'remove':
                s.discard(int(operation[1]))
            elif operation[0] == 'check':
                if int(operation[1]) in s:
                    print(1)
                else:
                    print(0)
            elif operation[0] == 'toggle':
                if int(operation[1]) in s:
                    s.discard(int(operation[1]))
                else:
                    s.add(int(operation[1]))

입력을 readline으로 해주셔야 속도 초과가 안날거같네요옹