[백준] 2146번 다리 만들기

HL·2021년 1월 17일
0

백준

목록 보기
13/104
  • 출처 : 다리만들기

  • 풀이방법

    • DFS로 섬의 테두리 좌표를 집합에 저장 (이후 리스트에 append)
    • 이중 for문으로 섬 별 최소 거리 구하기
    • 최소 거리의 최솟값 출력
  • 소감

    • pypy 정답 ㅠ
    • 풀이가 다른 문제에 비해서는 직관적으로 떠올랐다

코드

import sys

def init():

    n = int(input(''))
    board = []
    for i in range(n):
        line = list(map(int, input('').split(' ')))
        board.append(line)

    return n, board


def get_new_position(pos, d):

    if d == 'up':
        return (pos[0]-1,pos[1])

    if d == 'down':
        return (pos[0]+1,pos[1])

    if d == 'right':
        return (pos[0],pos[1]+1)

    if d == 'left':
        return (pos[0],pos[1]-1)


def out(n, pos):

    if pos[0] >= 0 and pos[0] < n:
        if pos[1] >= 0 and pos[1] < n:
            return False
    return True
        

def dfs(n, board, visited, curr_pos, pos_set):

    visited[curr_pos[0]][curr_pos[1]] = True

    for d in ['up','down','right','left']:

        new_pos = get_new_position(curr_pos, d)

        if out(n, new_pos):
            continue
        if board[new_pos[0]][new_pos[1]] == 0:
            pos_set.add(curr_pos)
            continue
        if visited[new_pos[0]][new_pos[1]]:
            continue
        dfs(n, board, visited, new_pos, pos_set)


def get_part_list(n, board):

    visited = [[False for j in range(n)] for i in range(n)]
    
    part_list = []

    for i in range(n):
        for j in range(n):

            if board[i][j] == 0:
                continue
            if visited[i][j]:
                continue

            pos_set = set()
            dfs(n, board, visited, (i,j), pos_set)
            part_list.append(pos_set)

    return part_list


def print_line(line_list):

    for line in line_list:
        print(line)
    print('')


def get_distance(pos1, pos2):

    result = -1
    result += abs(pos1[0]-pos2[0])
    result += abs(pos1[1]-pos2[1])
    return result


def get_min_distance(part1_set, part2_set):

    min_distance = float('inf')

    for pos1 in part1_set:
        for pos2 in part2_set:

            distance = get_distance(pos1, pos2)
            if min_distance > distance:
                min_distance = distance
    return min_distance


sys.setrecursionlimit(99999)

n, board = init()
part_list = get_part_list(n, board)

result = float('inf')
for i in range(len(part_list)):
    for j in range(i+1, len(part_list)):

        min_distance = get_min_distance(part_list[i], part_list[j])
        if result > min_distance:
            result = min_distance

print(result)
profile
Swift, iOS 앱 개발을 공부하고 있습니다

0개의 댓글