[프로그래머스/Python] Level 2_거리두기 확인하기 (2021 카카오 채용연계형 인턴십)

📌문제 유형

2021 카카오 채용연계형 인턴십

 

📌문제

프로그래머스

 

📌나의 문제풀이

def mahaton_dist(x1,y1,x2,y2):
    return abs(x1-x2) + abs(y1-y2)

def distance_under_2(maps,i,a1,b1,a2,b2):  # 기준 p의 좌표(a1,b1), 차이나는 거리(da,db)
    # 자리 사이에 파티션 존재 # 파티션 사이에 두고 앉는 경우
    if a1 == a2 and maps[i][a1][max(b1,b2)-1] == 'X':
        return 1
    elif b1 == b2 and maps[i][max(a1,a2)-1][b1] == 'X':
        return 1
    elif (maps[i][a1 + (a2-a1)][b1] == 'X' and maps[i][a1][b1 + (b2-b1)] == 'X'):
        return 1
    else:
        return 0
    
def solution(places):
    answer = []
    
    maps = []
    p_position = []     # p위치 저장 [행, 열]
    for l in range(5):
        class_ = []
        p = []
        for r in range(5):
            class_.append(list(places[l][r]))
            for c in range(5):
                if places[l][r][c] == 'P':
                    p.append([r,c])
        p_position.append(p)
        maps.append(class_)
        print(p_position)
        
    for i in range(5):
        # 한 대기실 내, 모든 좌표끼리 맨해튼 거리 측정
        # 배열이 2개 이상인 경우
        check = 1
        for n in range(len(p_position[i])-1):
            for m in range(n+1,len(p_position[i])):
                if mahaton_dist(p_position[i][n][0],p_position[i][n][1], p_position[i][m][0],p_position[i][m][1]) == 2:
                    check = distance_under_2(maps, i, p_position[i][n][0], p_position[i][n][1], p_position[i][m][0] , p_position[i][m][1])
                    if check == 0:
                        break
                elif mahaton_dist(p_position[i][n][0],p_position[i][n][1], p_position[i][m][0],p_position[i][m][1]) == 1:
                    check = 0
                    break

            if check == 0:
                break
    
        answer.append(check)
        
    return answer

 

📌리뷰

- 11번 문제만 통과 안되어서 시간오래걸렸다

• 맨해튼 거리 계산할 때, 자리 사이에 파티션 존재하는 경우 행 기준, 열 기준으로 따로 검사하기
• 한 번에 검사하면 통과 못함