BOJ. 11723 집합 (Python)

[ 문제 ]

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

11723번: 집합

 

[ 나의 풀이 ]

🔸 접근 방법

문제에서 집합 S에 들어가는 값 x가 (1 ≤ x ≤ 20) 로 범위가 정해져있기 때문에 True/False방식을 사용했다.

toggle연산이 있어서 더욱 True/False방식을 사용하면 편할 것이라고 생각했다.

연산의 수는 M (1 ≤ M ≤ 3,000,000) 이기 때문에 갯수가 너무 커서 sys.stdin.readline 을 사용했다.

input값을 이 방식으로 받으니까 뒤에 줄 변환 값이 함께 들어와서 .split()을 사용해서 제거했다.

 

🔸 풀이 코드

import sys
input = sys.stdin.readline

# 연산의 수 M
M = int(input())

# 공집합 S
S = [False] * 21

for _ in range(M):
    msg = input().strip()
    if msg == 'all':
        S = [True] * 21
    elif msg =='empty':
        S = [False] * 21
    else:
        msg, n = msg.split()
        n = int(n)
        if msg == 'add':
            if S[n] == False:
                S[n] = True
        if msg == 'remove':
            if S[n] == True:
                S[n] = False
        if msg == 'check':
            if S[n] == True:
                print(1)
            else:
                print(0)
        if msg == 'toggle':
            if S[n] == True:
                S[n] = False
            else:
                S[n] = True

 

[ pair의 풀이 ]

🔸 접근 방법

pair는 정직하게 append, remove, in, not in 등등을 사용했다..!

pair의 풀이 중에 input 연산을 바로 split하지 않고 list로 변환한 후에 앞의 값만 확인 하는 방법을 사용했다.

이거 완전 좋은 방법인 것 같다. 👍

 

🔸 풀이 코드

import sys
input = sys.stdin.readline

n = int(input())
arr = list()
for _ in range(n):
    x = list(input().split())
    a = x[0]

    if a == 'add':
        if int(x[1]) not in arr:
            arr.append(int(x[1]))

    elif a == 'remove':
        if int(x[1]) in arr:
            arr.remove(int(x[1]))

    elif a == 'check':
        if int(x[1]) in arr:
            print(1)
        else:
            print(0)

    elif a == 'toggle':
        if int(x[1]) in arr:
            arr.remove(int(x[1]))
        else:
            arr.append(int(x[1]))

    elif a == 'all':
        arr = [i for i in range(1, 21)]

    elif a == 'empty':
        arr.clear()

 

pair 의 github : https://github.com/Ui-Seok