33-H0ngJu

[H0ngJu]

🔗 문제 링크

병원 거리 최소화

✔️ 소요된 시간

50M

✨ 수도 코드

이번에는 삼성 SW 기출 문제 중에 하나를 선택했습니다!

이 문제를 어떻게 풀까... 하다가 N개 중에 M개 뽑아서 그냥 최소 거리 갱신하면 되겠다 해서 아래 흐름으로 풀었습니다.

1. 각 사람마다 각 병원의 거리를 계산한다.
2. 병원 K개 중에서 M개를 뽑는 조합 생성
3. 조합에서 하나씩 꺼내서 최소거리 갱신하기

2에서 저는 itertools의 combinations 함수를 사용했어요

근데 뭔가 문제에서 요구하는게 이게 맞나? 싶어서 다른 풀이를 찾아보던 중에
삼성 코테에서는 itertools import하는 것을 막는다고 하더라고요?

그래서 combinations 함수를 따로 구현해서 한번 더 풀었습니다.

itertools.combinations(combbination_arr, m)

이랬는데

all_combinations = []

def combination(start, depth, combination_arr):
    if depth == m: # m만큼 찾으면 return
        all_combinations.append(combination_arr[:])
        return
    for i in range(start, len(hospital_arr)):
        combination_arr.append(hospital_arr[i]) # 원소 일단 넣고
        combination(i + 1, depth + 1, combination_arr) # 재귀 돌림
        combination_arr.pop() # 다시 뽑은거 복귀 -> [1,2,3]을 구하면, 3빼고 다시 다음 반복문 넣어서 [1,2,4] 가능

combination(0, 0, [])

요래 풀어서...

📚 새롭게 알게된 내용

slicing은 깊은 복사일까요 얕은 복사일까요?

저는 slicing이 deepcopy라고 생각했습니다.

첫번째 케이스 )

arr = [0,1,2,3,4,5,6]
new_arr = arr[3:]

arr[5] = 777

print(arr, new_arr)
print(id(arr[5]), '=', id(new_arr[2]))

# 결과 
# [0, 1, 2, 3, 4, 777, 6] [3, 4, 5, 6]
# 1939064674800 = 1939063636336 

이니 new_arr과 arr이 메모리 주소가 다르니까요

두번째 케이스 )
slicing은 얕은 복사다라는 블로그 글을 보게 되었는데요

글의 예제인 print(id(a[1]), '=', id(b[1]))를 보면 결국 메모리 주소가 같게 나와서 slicing은 얕은 복사라고 말하더라고요

a = [[1, 2], [3, 4, 5]]
b = a[:]

print(a, '=', b) # 같은 값 출력
print(id(a), id(b)) # 바깥 list의 메모리 주소는 다름
print(id(a[0]), id(b[0])) # 내부 list의 메모리 주소는 같음

# 결과
# [[1, 2], [3, 4, 5]] = [[1, 2], [3, 4, 5]]
# 2586361266304 2586361266368
# 2586360892928 2586360892928

2차원 배열을 slicing으로 복사를 하는 경우에,
바깥 리스트에만 주소가 달라지는 것을 알 수 있습니다.


내부 리스트는 같은 곳을 참조하는 거죠.

저는 slicing을 얕은 복사라고 생각하기로 했습니다. (근데 깊은 복사인척 하는...)

다른 분들은 slicing이 얕은 복사, 깊은 복사 중 무엇이라고 생각하시나요?

[9kyo-hwang]

문제가 눈에 익다 했더니... 백준에 15686 치킨 배달 문제랑 완벽하게 동일하네요 ㅋㅋ
요거로 점수 달달하게 챙기시죠 :>

[9kyo-hwang]

추가로 리스트 슬라이싱은 말씀하신 것처럼 "얕은 복사"입니다.
a라는 리스트를 슬라이싱해서 b에 대입하게 되면, a와 b는 서로 다른 주소를 가지나 각각의 원소는 같은 주소를 가리킵니다. list 안에 mutable한 개체인 list가 또 있는 구조이기 때문입니다.
해당 게시글이 많이 도움될 것 같습니다.

[tgyuuAn]

슬라이싱은 얇은 복사에요!
파이썬에서 깊은 복사는 import copy 에 copy.deepcopy() 밖에 없는걸로압니다!!! (맞나요 ?)

[tgyuuAn]

from itertools import combinations
import sys

def input(): return sys.stdin.readline().rstrip()

N, M = map(int, input().split())
hospital = []
people = []
for row in range(N):
    _input = list(map(int, input().split()))
    for col in range(N):
        if _input[col] == 1:
            people.append((row, col))
        elif _input[col] == 2:
            hospital.append((row, col))      

answer = int(1e9)
for items in combinations(hospital, M):
    dedicate = 0
    for person in people:        
        _min = int(1e9)

        for item in items:
            temp = abs(item[0] - person[0]) + abs(item[1] - person[1])
            _min = min(_min, temp)

        dedicate += _min
    answer = min(answer, dedicate)

print(answer)

itertools... 개사기긴하죠....
백준에 N과 M 시리즈 문제집 itertools 안쓰고 풀면 효과적이에요!
문제 제대로 안읽고 40분 날림..

[H0ngJu]

추가로 리스트 슬라이싱은 말씀하신 것처럼 "얕은 복사"입니다. a라는 리스트를 슬라이싱해서 b에 대입하게 되면, a와 b는 서로 다른 주소를 가지나 각각의 원소는 같은 주소를 가리킵니다. list 안에 mutable한 개체인 list가 또 있는 구조이기 때문입니다. 해당 게시글이 많이 도움될 것 같습니다.

저도 올려주신 게시글 보고 알게되었습니다

[H0ngJu]

슬라이싱은 얇은 복사에요! 파이썬에서 깊은 복사는 import copy 에 copy.deepcopy() 밖에 없는걸로압니다!!! (맞나요 ?)

옙 저도 그렇게 알고 있습니다 ~. ~