백준 1236번 파이썬 풀이 : 성 지키기

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

1236번: 성 지키기

문제

영식이는 직사각형 모양의 성을 가지고 있다. 성의 1층은 몇 명의 경비원에 의해서 보호되고 있다. 영식이는 모든 행과 모든 열에 한 명 이상의 경비원이 있으면 좋겠다고 생각했다.

성의 크기와 경비원이 어디있는지 주어졌을 때, 몇 명의 경비원을 최소로 추가해야 영식이를 만족시키는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 성의 세로 크기 N과 가로 크기 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에는 성의 상태가 주어진다. 성의 상태는 .은 빈칸, X는 경비원이 있는 칸이다.

출력

첫째 줄에 추가해야 하는 경비원의 최솟값을 출력한다.

 

# 문제 풀이 (알고리즘 분류 : 구현)

해당 문제는 X가 없는 가로줄의 수와 세로줄의 수 중 큰 수를 리턴하면 된다.

 

n, m = map(int, input().split())
 
answer = []
 
for i in range(n):
    answer.append(input())
 
x_count, y_count = 0, 0

n과 m을 각각 입력 받아 준 후 n의 길이만큼 다시 input 받고 answer에 append 해주며 2차원 배열을 완성한다.

 

for i in range(n):
    if "X" not in answer[i]:
        x_count += 1
 
for j in range(m):
    if "X" not in [answer[i][j] for i in range(n)]:
        y_count += 1
 
print(max(x_count, y_count))

가로와 세로에서 각각 X가 필요한 줄의 수를 count 한 후 max값을 리턴하면 된다.

 

여기서 if 문 안에 반복문을 넣을 수 있다는 파이썬의 편안함을 알고 있으면 문제푸는데 많은 도움이 될 것 같다.