🐹 거리두기 확인하기

1. 문제 링크

거리두기 확인하기

2. 코드

def solution(places):
    answer = []
    for place in places:
        candidate = find_candidate(place)
        answer.append(check_observe_distance(place, candidate))
    return answer

def find_candidate(place):
    candidate = []
    for i in range(5):
        for j in range(5):
            if place[i][j] == 'P':
                candidate.append((i, j))
    return candidate

def check_observe_distance(place, candidate):
    for i in range(len(candidate)-1):
        now_x, now_y = candidate[i]
        for j in range(i+1, len(candidate)):
            if i == j:
                continue
            next_x, next_y = candidate[j]
            manhattan = abs(now_x - next_x) + abs(now_y - next_y)
            diff_x, diff_y = abs(now_x - next_x), abs(now_y - next_y)
            # 맨허튼 거리가 1이하인 경우
            if manhattan <= 1:
                return 0
            # 맨허튼 거리가 2인 경우
            if manhattan == 2:
                # 가로 길이 2인 경우 중 파티션 없는 경우
                if diff_x == 2 and place[min(now_x, next_x)+1][now_y] != 'X':
                    return 0
                # 세로 길이 2인 경우 중 파티션 없는 경우
                elif diff_y == 2 and place[now_x][min(now_y, next_y)+1] != 'X':
                    return 0
                # 대각선인 경우 중 파티션 없는 경우
                elif place[now_x][next_y] == 'O' or place[next_x][now_y] == 'O':
                    return 0
    return 1

• 실행 결과

테스트 1 〉	통과 (0.16ms, 10.4MB)
테스트 2 〉	통과 (0.05ms, 10.3MB)
테스트 3 〉	통과 (0.04ms, 10.3MB)
테스트 4 〉	통과 (0.06ms, 10.2MB)
테스트 5 〉	통과 (0.04ms, 10.4MB)
테스트 6 〉	통과 (0.03ms, 10.2MB)
테스트 7 〉	통과 (0.03ms, 10.3MB)
테스트 8 〉	통과 (0.08ms, 10.4MB)
테스트 9 〉	통과 (0.04ms, 10.3MB)
테스트 10 〉	통과 (0.03ms, 10.2MB)
테스트 11 〉	통과 (0.04ms, 10.2MB)
테스트 12 〉	통과 (0.03ms, 10.4MB)
테스트 13 〉	통과 (0.04ms, 10.2MB)
테스트 14 〉	통과 (0.03ms, 10.4MB)
테스트 15 〉	통과 (0.03ms, 10.3MB)
테스트 16 〉	통과 (0.06ms, 10.2MB)
테스트 17 〉	통과 (0.05ms, 10.2MB)
테스트 18 〉	통과 (0.06ms, 10.3MB)
테스트 19 〉	통과 (0.07ms, 10.2MB)
테스트 20 〉	통과 (0.08ms, 10.1MB)
테스트 21 〉	통과 (0.07ms, 10.2MB)
테스트 22 〉	통과 (0.06ms, 10.4MB)
테스트 23 〉	통과 (0.02ms, 10.3MB)
테스트 24 〉	통과 (0.03ms, 10.3MB)
테스트 25 〉	통과 (0.03ms, 10.4MB)
테스트 26 〉	통과 (0.02ms, 10.1MB)
테스트 27 〉	통과 (0.03ms, 10.3MB)
테스트 28 〉	통과 (0.03ms, 10.2MB)
테스트 29 〉	통과 (0.05ms, 10.4MB)
테스트 30 〉	통과 (0.03ms, 10.4MB)
테스트 31 〉	통과 (0.05ms, 10.4MB)

3. 해설

강의실 별 응시자들의 위치 정보를 담은 candidate를 생성한 후, 강의실 별 가능한 경우를 파악한다.

불가능한 경우는 다음과 같다.

1. 맨허튼 거리 = 1
2. 맨허튼 거리 = 2
   1. 가로 거리 혹은 세로 거리가 2차이 일때 가운데 위치에 파티션이 없는 경우
   2. 대각선에 응시자들이 위치할 때 반대 방향 대각선 중 파티션이 하나라도 없는 경우

candidate 위치를 비교하여 위의 불가능한 경우일 때는 0을 리턴, 그 이외는 1을 리턴하는 check_observe_distance 함수를 작성한 후 answer에 값을 넣어준다.