백준 11723번 집합

문제 링크: https://www.acmicpc.net/problem/11723

Author: skysign

백준 11723번 집합/main.py

@@ -0,0 +1,42 @@
+import sys
+
+
+def solve():
+    # input() 사용하면, 시간초과 발생함
+    N = int(sys.stdin.readline().strip())
+    ans = set()
+
+    for _ in range(N):
+        # input() 사용하면, 시간초과 발생함
+        dt = sys.stdin.readline().strip()
+        op = ''
+        v = 0
+
+        if dt.__contains__(' '):
+            op, v = dt.split(' ')
+            v = int(v)
+        else:
+            op = dt
+
+        if op == 'add':
+            ans.add(v)
+        elif op == 'remove':
+            # if v in ans:
+            #     ans.remove(v)
+            ans.discard(v)
+        elif op == 'check':
+            p = 1 if v in ans else 0
+            print(p)
+        elif op == 'toggle':
+            if v in ans:
+                ans.remove(v)
+            else:
+                ans.add(v)
+        elif op == 'all':
+            t = [i for i in range(1, 21)]
+            ans = set(t)
+        elif op == 'empty':
+            ans = set()
+
+if __name__ == '__main__':
+    solve()

백준 11723번 집합/test1.txt

@@ -0,0 +1,27 @@
+26
+add 1
+add 2
+check 1
+check 2
+check 3
+remove 2
+check 1
+check 2
+toggle 3
+check 1
+check 2
+check 3
+check 4
+all
+check 10
+check 20
+toggle 10
+remove 20
+check 10
+check 20
+empty
+check 1
+toggle 1
+check 1
+toggle 1
+check 1

백준 11723번 집합/test1_answer.txt

@@ -0,0 +1,16 @@
+1
+1
+0
+1
+0
+1
+0
+1
+0
+1
+1
+0
+0
+0
+1
+0

백준 11723번 집합/test_main.py

@@ -0,0 +1,19 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+def my_solve(testcase_input):
+    sys.stdin = open(testcase_input, 'r')
+    stdout = sys.stdout
+    sys.stdout = open('stdout.txt', 'w')
+    solve()
+    sys.stdout.close()
+    sys.stdout = stdout
+
+class Test(TestCase):
+    def test1_solve(self):
+        my_solve('test1.txt')
+        self.assertEqual(
+            Path('test1_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())