[백준-Python] 누울 자리를 찾아라 : 1652번

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

 

 

❎ 1차 풀이

import sys
input=sys.stdin.readline

n=int(input())
row=0
col=0

rooms=[]

#가로 확인
for i in range(n):
    x_list=list(input().rstrip())
    rooms.append(x_list)
    #col 순회
    for i in range(n-1):
        #2개이상 연속인지 확인
        if x_list[i]=='.' and x_list[i+1]=='.':
            row+=1
            break
        


#세로 확인
for i in range(n):
    for j in range(n-1):
        if rooms[j][i]=='.' and rooms[j+1][i]=='.':
            col+=1
            break
        
print(row,col)

각 세로, 가로줄에 2개이상의 누울곳이 최대 1개라고 생각했기에 

연속2개의 . 이 나온다면 바로 break로 for문을 종료했었다. 

바로 틀렸습니다가 출력되는것을 보고 

또 내가 엣지케이스를 놓쳤구나 싶어서 다시 살펴보니,

예제엔 존재하지 않지만 ..x.. 이런 경우엔 최대 1개가 아닌, 누울곳이 2개로 되겠구나 싶어서 코드를 변경하였다.

 

 

✅ 정답 풀이

# X를 기준으로 누울수 있는 영역을 나눈다 (split) -> 2이상이면 카운트

import sys
input=sys.stdin.readline

n=int(input())
row=0
col=0

rooms=[]

#가로 
for i in range(n):
    x_list=input().rstrip()
    rooms.append(list(x_list))
    row+=len(list(filter(lambda x:len(x)>=2,x_list.split('X'))))

#세로
for i in range(n):
    temp=''
    for j in range(n):
        temp+=rooms[j][i]
    col+=len(list(filter(lambda x:len(x)>=2,temp.split('X'))))     
       
        
print(row,col)

#31120	44

X(벽)을 기준으로 누울자리를 나누어야하기에 split을 사용해주었고,

..  = len(2) 이상을 길이를 filter를 이용해 걸러주고, 리턴되는 리스트의 길이를 더해주면 된다.

filter대신 in 연산자를 사용하는 방법도 있었다!

 

❗️ filter는 모든 리스트를 순회하지만, in 연산자는 순회하면서 true를 발견하면 바로 리턴하기 때문에

조금 더 실행시간이 짧은것을 확인해볼 수 있었다.