알고리즘/알고리즘_백준

[Python][백준] 1652. 누울 자리를 찾아라 / 구현, 문자열 (S5)

Jerry_K 2024. 7. 16. 20:49

링크🔗

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


🗒️파이썬 코드 풀이

N = int(input())
lst = [list(input()) for _ in range(N)]

width = 0 
for i in range(N): 
    space = 0 
    for j in range(N):
        if lst[i][j] ==".":
            space += 1
            if space == 2:
                width += 1
        else : 
            space = 0

length = 0 
for j in range(N): 
    space = 0 
    for i in range(N):
        if lst[i][j] ==".":
            space += 1
            if space == 2:
                length += 1
        else : 
            space = 0
            
print(width, length)

 

1. 한 줄에 2개 이상 누울 자리가 있을 수도 있다는 것을 주의해야한다.

 

2. 2차원 배열 크기만큼 반복을 하고, lst[i][j]가 "." 일때, space 카운드를 한다.

 

3. space가 2개이면, 몸을 누을 수 있으니, width나 length에 추가한다. 

(3개 이상부터는 신경 안써도 된다.)

 

4. lst[i][j]가 "X"  일 때, space를 다시 0으로 초기화 해준다. 

 

5. 세로 같은 경우 반복문의 i와 j만 적절하게 바꾸면 된다 

 

✏️ 내 코드 풀이

N = int(input())
lst = [list(input()) for _ in range(N)]

rs = [0,0]
for i in range(N):
    sqc = True
    for j in range(N-1):
        if lst[i][j] == "." and lst[i][j+1] == "." and sqc :
            rs[0] += 1 
            sqc = False
        elif lst[i][j] == "X" or j+1==N :
            sqc = True

for j in range(N):
    sqc = True
    for i in range(N-1):
        if lst[i][j] == "." and lst[i+1][j] == "." and sqc :
            rs[1] += 1 
            sqc = False
        elif lst[i][j] == "X" or i+1==N :
            sqc = True

print(rs[0],rs[1])

 

나는 다소 복잡하게 풀었다. 

 

lst[i][j]  와 lst[i][j+1]가 모두 "."이고, sqc(연속성)이 True일 때, 

rs 리스트의 값이 올라간다.

 

그리고 lst[i][j]가 "X"이거나 j+1이 해당 범위를 넘어 갈때, 

sqc를 다시 True로 만들어서 연속성을 찾아준다.

 

이 방법도 정답으로 제출은 되었지만, 

좀 더 간단하게 할 수 있는 법을 생각해보자 ! 

 

📌  문제 코멘트

 

문제가 좀 이해가 잘 안됐었다. 

2칸 이상이 왜 3칸,4칸,5칸 ... 이랑 같은지 말이다.

4칸이면 3자리 남는거 아닌가 ??


📚문제

일 년 동안 세계일주를 하던 영식이는 여행을 하다 너무 피곤해서 근처에 있는 코레스코 콘도에서 하룻밤 잠을 자기로 하고 방을 잡았다.

코레스코 콘도에 있는 방은 NxN의 정사각형모양으로 생겼다. 방 안에는 옮길 수 없는 짐들이 이것저것 많이 있어서 영식이의 누울 자리를 차지하고 있었다. 영식이는 이 열악한 환경에서 누울 수 있는 자리를 찾아야 한다. 영식이가 누울 수 있는 자리에는 조건이 있다. 똑바로 연속해서 2칸 이상의 빈 칸이 존재하면 그 곳에 몸을 양 옆으로 쭉 뻗으면서 누울 수 있다. 가로로 누울 수도 있고 세로로 누울 수도 있다. 누울 때는 무조건 몸을 쭉 뻗기 때문에 반드시 벽이나 짐에 닿게 된다. (중간에 어정쩡하게 눕는 경우가 없다.)

만약 방의 구조가 위의 그림처럼 생겼다면, 가로로 누울 수 있는 자리는 5개이고, 세로로 누울 수 있는 자리는 4개 이다. 방의 크기 N과 방의 구조가 주어졌을 때, 가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 수를 구하는 프로그램을 작성하시오.

입력

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

출력

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

예제 입력 1 복사

5
....X
..XX.
.....
.XX..
X....

예제 출력 1 복사

5 4