백준 1012번 유기농 배추

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

Author: skysign

백준 1012번 유기농 배추/main.py

@@ -0,0 +1,39 @@
+import sys
+from typing import List
+
+
+def solve():
+    T = int(sys.stdin.readline().strip())
+    drcs = [[1, 0], [0, 1], [-1, 0], [0, -1]]
+
+    for _ in range(T):
+        answer: int = 0
+        M, N, cabbage_cnt = map(int, sys.stdin.readline().strip().split(' '))
+        board: List[List[int]] = [[0 for _ in range(N)] for _ in range(M)]
+
+        for _ in range(cabbage_cnt):
+            row, col = map(int, sys.stdin.readline().strip().split(' '))
+            board[row][col] = 1
+
+        for col in range(N):
+            for row in range(M):
+                if board[row][col] == 1:
+                    answer += 1
+                    queue: List[List[int]] = [[row, col]]
+                    board[row][col] = 0
+
+                    while queue:
+                        r, c = queue.pop(0)
+
+                        for dr, dc in drcs:
+                            nr, nc = r + dr, c + dc
+
+                            if ((0 <= nr < M) and (0 <= nc < N)) and (board[nr][nc] == 1):
+                                queue.append([nr, nc])
+                                board[nr][nc] = 0
+
+        print(answer)
+
+
+if __name__ == '__main__':
+    solve()

백준 1012번 유기농 배추/test1.txt

@@ -0,0 +1,21 @@
+2
+10 8 17
+0 0
+1 0
+1 1
+4 2
+4 3
+4 5
+2 4
+3 4
+7 4
+8 4
+9 4
+7 5
+8 5
+9 5
+7 6
+8 6
+9 6
+10 10 1
+5 5

백준 1012번 유기농 배추/test1_answer.txt

@@ -0,0 +1,2 @@
+5
+1

백준 1012번 유기농 배추/test2.txt

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

백준 1012번 유기농 배추/test2_answer.txt

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

백준 1012번 유기농 배추/test3.txt

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

백준 1012번 유기농 배추/test3_answer.txt

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

백준 1012번 유기농 배추/test4.txt

@@ -0,0 +1,17 @@
+1
+5 5 15
+0 0
+1 0
+2 0
+4 0
+0 1
+2 1
+3 1
+4 1
+0 2
+0 3
+0 4
+1 4
+2 4
+3 4
+4 4

백준 1012번 유기농 배추/test4_answer.txt

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

백준 1012번 유기농 배추/test5.txt

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

백준 1012번 유기농 배추/test5_answer.txt

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

백준 1012번 유기농 배추/test_main.py

@@ -0,0 +1,38 @@
+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')
+
+    def test2_solve(self):
+        self.test_solve('2')
+
+    def test3_solve(self):
+        # 출처 https://www.acmicpc.net/board/view/139316
+        self.test_solve('3')
+
+    def test4_solve(self):
+        # 출처 https://www.acmicpc.net/board/view/126545
+        self.test_solve('4')
+
+    def test5_solve(self):
+        # 출처 https://www.acmicpc.net/board/view/126545
+        self.test_solve('5')