Queue/Stack Python Code

VSFe · VSFe · commit 510c230d3d66 · 2021-01-15T00:23:40.000+09:00
diff --git a/Concept/01_Queue_Stack/SampleCode/Python/BOJ_10773.py b/Concept/01_Queue_Stack/SampleCode/Python/BOJ_10773.py
@@ -0,0 +1,19 @@
+"""
+문제: 제로 (http://boj.kr/10773)
+Tier: Silver 4 (2021.01.14 기준)
+Comment: 0이 나올때 가장 최근의 값을 뺍니다.
+즉, FIFO 구조를 띄고 있고, 자연스럽게 스택을 사용하시면 됩니다.
+"""
+
+import sys
+
+N = int(sys.stdin.readline().rstrip())
+_list = []
+for i in range(N):
+    tmp = int(sys.stdin.readline().rstrip())
+    if tmp == 0:
+        _list.pop()
+    else:
+        _list.append(tmp)
+
+print(sum(_list))
diff --git a/Concept/01_Queue_Stack/SampleCode/Python/BOJ_1158.py b/Concept/01_Queue_Stack/SampleCode/Python/BOJ_1158.py
@@ -0,0 +1,18 @@
+"""
+문제: 요세푸스 문제 0 (http://boj.kr/1158)
+Tier: Silver 5 (2021.01.12 기준)
+Comment: 큐를 활용할 수 있는 대표적인 문제입니다.
+출력이 조금 특이하기 때문에 출력만 유의해서 풀어주세요.
+"""
+
+from collections import deque
+
+N, K = map(int, input().split())
+_list = []
+q = deque([i + 1 for i in range(N)])
+
+while len(q) != 0:
+    q.rotate(-K)
+    _list.append(q.pop())
+
+print('<' + ', '.join(map(str, _list)) + '>')
diff --git a/Concept/01_Queue_Stack/SampleCode/Python/BOJ_2164.py b/Concept/01_Queue_Stack/SampleCode/Python/BOJ_2164.py
@@ -0,0 +1,19 @@
+"""
+문제: 카드 2 (http://boj.kr/2164)
+Tier: Silver 4 (2021.01.12 기준)
+Comment: 큐를 배웠으니 큐를 통해서 풀이 하였습니다.
+한번은 그냥 빼고, 한번은 뺀 걸 다시 넣는데, 결국 두 연산을 한 번씩 하면 크기가 1씩 감소합니다.
+즉 크기가 1이 될 때 까지 빼고, 빼서 넣고를 반복하면 쉽게 해결할 수 있겠네요.
+물론 큐를 사용하지 않고 수식으로도 풀 수 있습니다.
+어떻게 할 수 있을까요?
+"""
+
+from collections import deque
+
+li = deque([i for i in range(1, int(input()) + 1)])
+
+while len(li) >= 2:
+    li.popleft()
+    li.rotate(-1)
+
+print(li[0])
diff --git a/Concept/01_Queue_Stack/SampleCode/Python/BOJ_9012.py b/Concept/01_Queue_Stack/SampleCode/Python/BOJ_9012.py
@@ -0,0 +1,25 @@
+"""
+문제: 괄호 (http://boj.kr/9012)
+Tier: Silver 4 (2021.01.14 기준)
+Comment: 스택 하면 떠올릴 수 있는 가장 유명한 문제입니다.
+그런데, 어차피 괄호 종류는 하나인데 굳이 스택을 써야 할까요?
+(백준 pypy3 기준 효율성 1위 코드입니다 ㅎㅎ;;)
+"""
+
+import sys
+
+N = int(sys.stdin.readline().rstrip())
+result = ""
+
+for i in range(N):
+    testcase = sys.stdin.readline().rstrip()
+    cnt = 0
+    for c in testcase:
+        cnt += 1 if c == '(' else -1
+        if cnt < 0:
+            result += "NO\n"
+            break
+    else:
+        result += "YES\n" if cnt == 0 else "NO\n"
+
+print(result)
