백준 9466번 텀 프로젝트 v2

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

Author: skysign

백준 9466번 텀 프로젝트 v2/main.py

@@ -0,0 +1,38 @@
+import sys
+from collections import deque
+
+
+def solve():
+    input = sys.stdin.readline
+    T = int(input().strip())
+
+    for _ in range(T):
+        N = int(input().strip())
+        dts = [0] + list(map(int, input().strip().split()))
+        incoming = [0 for _ in range(len(dts))]
+        for i in range(len(dts)):
+            fr, to = i, dts[i]
+            incoming[to] += 1
+
+        queue = deque()
+
+        for i in range(len(dts)):
+            if incoming[i] == 0:
+                queue.append(i)
+
+        while queue:
+            fr = queue.popleft()
+            to = dts[fr]
+            incoming[to] -= 1
+            if incoming[to] == 0:
+                queue.append(to)
+
+        answer = 0
+        for i in range(len(dts)):
+            if incoming[i] == 0:
+                answer += 1
+
+        print(answer)
+
+if __name__ == '__main__':
+    solve()

백준 9466번 텀 프로젝트 v2/test1.txt

@@ -0,0 +1,5 @@
+2
+7
+3 1 3 7 3 4 6
+8
+1 2 3 4 5 6 7 8

백준 9466번 텀 프로젝트 v2/test1_answer.txt

@@ -0,0 +1,2 @@
+3
+0

백준 9466번 텀 프로젝트 v2/test_main.py

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