Create 6.py

Author: ndb796

13/6.py

@@ -0,0 +1,78 @@
+from itertools import combinations
+
+n = int(input())
+a = []
+teachers = []
+spaces = []
+for i in range(n):
+    a.append(list(input().split()))
+    for j in range(n):
+        # 선생님이 존재하는 위치 저장
+        if a[i][j] == 'T':
+            teachers.append((i, j))
+        # 장애물을 설치할 수 있는 위치 저장
+        if a[i][j] == 'X':
+            spaces.append((i, j))
+
+# 특정 방향으로 감시를 진행 (학생 발견: True, 학생 미발견: False)
+def watch(x, y, direction):
+    # 왼쪽 방향으로 감시
+    if direction == 0:
+        while y >= 0:
+            if a[x][y] == 'S': # 학생이 있는 경우
+                return True
+            if a[x][y] == 'O': # 장애물이 있는 경우
+                return False
+            y -= 1
+    # 오른쪽 방향으로 감시
+    if direction == 1:
+        while y < n:
+            if a[x][y] == 'S': # 학생이 있는 경우
+                return True
+            if a[x][y] == 'O': # 장애물이 있는 경우
+                return False
+            y += 1
+    # 위쪽 방향으로 감시
+    if direction == 2:
+        while x >= 0:
+            if a[x][y] == 'S': # 학생이 있는 경우
+                return True
+            if a[x][y] == 'O': # 장애물이 있는 경우
+                return False
+            x -= 1
+    # 아래쪽 방향으로 감시
+    if direction == 3:
+        while x < n:
+            if a[x][y] == 'S': # 학생이 있는 경우
+                return True
+            if a[x][y] == 'O': # 장애물이 있는 경우
+                return False
+            x += 1
+    return False
+
+# 장애물 설치 이후에, 한 명이라도 학생이 감지되는지 검사
+def process():
+    # 모든 선생의 위치를 하나씩 확인
+    for x, y in teachers:
+        # 4가지 방향으로 학생을 감지할 수 있는지 확인
+        for i in range(4):
+            if watch(x, y, i):
+                return True
+    return False
+
+find = False
+for data in combinations(spaces, 3):
+    for x, y in data:
+        a[x][y] = 'O'
+    # 학생이 한 명도 감지되지 않는 경우
+    if not process():
+        # 원하는 경우를 발견한 것임
+        find = True
+        break
+    for x, y in data:
+        a[x][y] = 'X'
+
+if find:
+    print('YES')
+else:
+    print('NO')