[알고리즘] BOJ 1303번: 전쟁 - 전투(python3)

* 알고리즘 너무 약해서 기초 문제 50개 목표로 푸는 중...

* 3시간 안에 답을 내지 못하면 답지를 보고 30분 내로 정답 판정을 받고, 3일 뒤 다시 풀어보기

---

Greedy

9 / 50

 

탐색

9 / 50(NEW!)

 

기초 동적 프로그래밍

7 / 50

---

투포인터

2 / 10

 

이분탐색

0 / 10

---

문제

https://www.acmicpc.net/problem/1303

1303번: 전쟁 - 전투

→ solved.ac 기준 실버 1

---

문제 해결 아이디어

이전에 풀었던 문제와 비슷.

DFS 로 순회하면서 카운트 해준다음,

마지막에 저장해주고 반복.

---

구현

내 풀이

import sys
input = sys.stdin.readline

def dfs(x, y, team):
  global count
  if x < 0 or x >= n or y < 0 or y >= m:
    return False
  if graph[x][y] == team and not visited[x][y] :
    visited[x][y] = True
    count += 1
    dfs(x - 1, y, team)
    dfs(x, y - 1, team)
    dfs(x + 1, y, team)
    dfs(x, y + 1, team)
    return True
  return False

m, n = map(int, input().split())
graph = [list(input().rstrip()) for _ in range(n)]
visited = [[False] * m for _ in range(n)]

white_sum = 0
blue_sum = 0
count = 0

for i in range(n):
  for j in range(m):
    if dfs(i, j, graph[i][j]) == True:
      if graph[i][j] == 'W':
        white_sum += count ** 2
      else :
        blue_sum += count ** 2
      count = 0

print(white_sum, blue_sum)

---

메모

• 걸린 시간: 12분
• 보통 세로 길이, 가로 길이 순으로 받는데, 해당 문제는 가로 길이 먼저 받아서 IndexError가 나서 한참을 헤맸다..

---

마치며

비슷한 문제만 거의 3개 보는 것 같아서 이제 거의 반사적으로 푸는 것 같당..

DP는 아직 헥헥 하지만 BFS랑 DFS 문제는 난이도를 조금 올려서 풀어도 될 듯!!