파이썬 누울 자리를 찾아라(백준 BOJ 1652)
문제
일 년 동안 세계일주를 하던 영식이는 여행을 하다 너무 피곤해서 근처에 있는 코레스코 콘도에서 하룻밤 잠을 자기로 하고 방을 잡았다.
코레스코 콘도에 있는 방은 NxN의 정사각형모양으로 생겼다. 방 안에는 옮길 수 없는 짐들이 이것저것 많이 있어서 영식이의 누울 자리를 차지하고 있었다. 영식이는 이 열악한 환경에서 누울 수 있는 자리를 찾아야 한다. 영식이가 누울 수 있는 자리에는 조건이 있다. 똑바로 연속해서 2칸 이상의 빈 칸이 존재하면 그 곳에 몸을 양 옆으로 쭉 뻗으면서 누울 수 있다. 가로로 누울 수도 있고 세로로 누울 수도 있다. 누울 때는 무조건 몸을 쭉 뻗기 때문에 반드시 벽이나 짐에 닿게 된다. (중간에 어정쩡하게 눕는 경우가 없다.)
만약 방의 구조가 위의 그림처럼 생겼다면, 가로로 누울 수 있는 자리는 5개이고, 세로로 누울 수 있는 자리는 4개 이다. 방의 크기 N과 방의 구조가 주어졌을 때, 가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 방의 크기 N이 주어진다. N은 1이상 100이하의 정수이다. 그 다음 N줄에 걸쳐 N개의 문자가 들어오는데 ‘.’은 아무것도 없는 곳을 의미하고, ‘X’는 짐이 있는 곳을 의미한다.
출력
첫째 줄에 가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 개수를 출력한다.
예제 입력 1
5
....X
..XX.
.....
.XX..
X....
예제 출력 1
5 4
📝 풀어보기
📌 방의 크기 N을 입력받고 N의 크기만큼 반복하며 room list를 만든다. 그리고 가로와 세로의 누울 자리를 확인하기 위한 변수를 생성한다.
N = int(input())
room = [list(map(str, input())) for i in range(N)]
r_seat = 0
c_seat = 0
📌 N의 범위만큼 가로와 세로를 돌며 가로를 먼저 탐색한다.
가로의 자리가 빈자리일때 카운트를 1 증가시키고 짐을 만나면 초기화한다. 빈자리 카운트가 2가 되면 r_seat를 1씩 누적시킨다.
for r in range(N):
cnt_r = 0
cnt_c = 0
for c in range(N):
# 가로
if room[r][c] == '.': # 빈자리일때
cnt_r += 1 # 카운트 1 증가
else:
cnt_r = 0 # 짐을 만나면? 초기화
if cnt_r == 2: # 누울자리가 생기면
r_seat += 1 # 누울자리 카운트 증가
📌 세로도 마찬가지로 탐색한다.
# 세로
if room[c][r] == '.': # 빈자리일때
cnt_c += 1 # 카운트 1 증가
else:
cnt_c = 0 # 짐을 만나면 초기화
if cnt_c == 2: # 누울자리가 생기면
c_seat += 1 # 카운트 증가
print(r_seat, c_seat) # 가로 세로 출력
전체 코드
N = int(input())
room = [list(map(str, input())) for i in range(N)]
r_seat = 0
c_seat = 0
for r in range(N):
cnt_r = 0
cnt_c = 0
for c in range(N):
# 가로
if room[r][c] == '.': # 빈자리일때
cnt_r += 1 # 카운트 1 증가
else:
cnt_r = 0 # 짐을 만나면? 초기화
if cnt_r == 2: # 누울자리가 생기면
r_seat += 1 # 누울자리 카운트 증가
# 세로
if room[c][r] == '.': # 빈자리일때
cnt_c += 1 # 카운트 1 증가
else:
cnt_c = 0 # 짐을 만나면 초기화
if cnt_c == 2: # 누울자리가 생기면
c_seat += 1 # 카운트 증가
print(r_seat, c_seat) # 가로 세로 출력