[BOJ 1652번] 누울 자리를 찾아라(Python)

📝 문제

💡 일 년 동안 세계일주를 하던 영식이는 여행을 하다 너무 피곤해서
근처에 있는 코레스코 콘도에서 하룻밤 잠을 자기로 하고 방을 잡았다.
코레스코 콘도에 있는 방은 NxN의 정사각형모양으로 생겼다.
방 안에는 옮길 수 없는 짐들이 이것저것 많이 있어서 영식이의 누울 자리를 차지하고 있었다.
영식이는 이 열악한 환경에서 누울 수 있는 자리를 찾아야 한다.

영식이가 누울 수 있는 자리에는 조건이 있다.
똑바로 연속해서 2칸 이상의 빈 칸이 존재하면 그 곳에 몸을 양 옆으로 쭉 뻗으면서 누울 수 있다.
가로로 누울 수도 있고 세로로 누울 수도 있다.
누울 때는 무조건 몸을 쭉 뻗기 때문에 반드시 벽이나 짐에 닿게 된다.
(중간에 어정쩡하게 눕는 경우가 없다.)

방의 크기 N과 방의 구조가 주어졌을 때,
가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 수를 구하는 프로그램을 작성하시오.

⚙️ 입력 : 첫째 줄에 방의 크기 N이 주어진다.
N은 1이상 100이하의 정수이다.
그 다음 N줄에 걸쳐 N개의 문자가 들어오는데
'.'은 아무것도 없는 곳을 의미하고, 'X'는 짐이 있는 곳을 의미한다.

⚙️ 출력 : 첫째 줄에 가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 개수를 출력한다.

 

문제링크

1652번: 누울 자리를 찾아라

---

✅ 풀이

n = int(input())
room = []
for i in range(n):
    room.append(list(input()))

row = []
for item in room:
    temp = ''
    for items in item:
        if items == '.':
            temp += items
        else:
            row.append(temp)
            temp = ''
    row.append(temp)

rcnt = len([k for k in row if len(k) >= 2])

croom = [list(k) for k in zip(*room)]
col = []
for item in croom:
    temp = ''
    for items in item:
        if items == '.':
            temp += items
        else:
            col.append(temp)
            temp = ''
    col.append(temp)

ccnt = len([k for k in col if len(k) >= 2])

print(rcnt, ccnt)

💡 입력받은 대로 만들어진 방 구조 리스트로 가로 공간을,
zip을 이용해 방 구조를 전치시킨 리스트로 세로 공간을 구했다.

연속되는 .을 한 묶음으로 묶어서 별도 리스트에 저장해준 뒤,
. 묶음의 길이가 2 이상인 것들을 카운팅해주었다.