https://www.acmicpc.net/problem/1149
조건
맞닿아 있는 집의 색깔은 달라야 한다.
이전에 선택한 색깔은 다음 선택에 영향을 준다.
현재 집의 최소 비용이 다음 집의 최소 비용을 보장하지 못한다.
결론
가능한 경우에 대한 비용을 모두 구한 후에,
그 중 최소값을 구해야 한다.
아쉽지만 일반화 실패!
첫번째 집의 색깔이 선택되었을때 모든 경우의 수를 딕셔너리에 저장하고, 이 중 가능한 경우(비용)을 따로 저장하여 최소값을 구하는 것까지 구현!
#각 집에 칠하는 색깔을 배열로 구현하기 위해 일반화 필요
each_house_value = [
[26, 40, 83], [49, 60, 57], [13, 89, 99]
]
#색깔의개수와 집의수(N)에 따른 일반화 필요
hash_table = {
1:{1:{},
2:{},
3:{}},
2:{1:{},
2:{},
3:{}},
3:{1:{},
2:{},
3:{}}
}
#결과저장
result_array = []
def get_minimum_value(each_house_value, color, N):
#첫번째집 색깔을 선택한 이후에 나올 수 있는 비용의 경우의 수를 모두 구한 딕셔너리
#해당 딕셔너리를 만드는 함수
hash_table = get_hash_table(N, color, each_house_value)
#위에서 구한 딕셔너리에서 가능한 경우에 대한 비용만 결과배열에 추가하여
#결과배열중 최솟값을 구하는 함수
result = get_result_from_hash_table(N, hash_table)
return result
def get_hash_table(N, color, each_house_value):
if color == 1:
for k in range(N):
if k + 1 < N:
for i in range(N):
for j in range(N):
value = 26 + each_house_value[k][i] + each_house_value[k+1][j]
hash_table[color][i+1][j+1] = value
else:
break
if color == 2:
for k in range(N):
if k + 1 < N:
for i in range(N):
for j in range(N):
value = 40 + each_house_value[k][i] + each_house_value[k+1][j]
hash_table[color][i+1][j+1] = value
else:
break
if color == 3:
for k in range(N):
if k + 1 < N:
for i in range(N):
for j in range(N):
value = 83 + each_house_value[k][i] + each_house_value[k+1][j]
hash_table[color][i+1][j+1] = value
else:
break
#print(hash_table)
return hash_table
def get_result_from_hash_table(N, hash_table):
for k in range(1, 3+1)
for i in range(N):
for j in range(N):
if i + 1 == k or i == j:
hash_table[k][i + 1][j + 1] = 0
continue
else:
result_array.append(hash_table[k][i + 1][j + 1])
continue
result = min(result_array)
return result
get_minimum_value(each_house_value, 1, 3)
print(get_minimum_value(each_house_value, 1, 3))
최대한 일반화하는 코드를 구현해보자
현재 코드는 첫번째 집 색깔을 구했을때 이지만,
이상적인 코드는 모든 경우의 수를 구하는 코드이다!
모든 경우의 수를 구할때 필요한 중첩 반복문 / 중첩 딕셔너리 등 어떤 방식으로 구현할지 더 생각해보기
알고리즘을 구현하는데 너무 시간이 오래걸린다
다른 방향으로 생각하는 것도 중요하지만, 시간을 절약하는 것도 매우 중요하다.
10분내외로 생각하고 알고리즘 구현하는 것까지 가능하도록 꾸준히 연습해보자.
클린코드
하나의 코드에 모든 기능을 담지말고, 기능을 최대한 분산한다.
▶조건을 입력받아 결과를 return하는 로직(메인함수)
▶모든 경우의 수를 구해주고 딕셔너리를 작성하는 로직
▶딕셔너리에서 가능한 경우의 수를 선별하여 비용을 저장하는 로직
가독성이 좋은 코드, 이해가 쉬운 코드의 조건들을 생각해보자.
loop개념
for
while
각 반복문을 사용하기 위해 알맞은 문법과 변수선언 방법 등
딕셔너리
딕셔너리 선언하기, 만들기
중첩딕셔너리 선언하기, 만들기, 접근하기
0/None의 개념
0과 None은 서로 다른 개념!
0은 있는 값, None은 없는 값으로 반복문에 대한 조건을 다룰 때
각 분기별로 과정이 다르다!
코드에 대한 이해가 우선이다. sugar syntax보다는 sugar logic!
안녕하세요^^
저는 튜터는 아니고 함께 스터디에 참여하고 있는 수강생 1인입니다만 궁금해서 코드를 확인해보았습니다.
우선 마지막 함수에서 SyntaxError가 뜹니다.
def get_result_from_hash_table(N, hash_table):
for k in range(1, 3+1)
반복문에 :가 없기 때문입니다.
콜론을 넣고 나서도 프로그램을 돌려보면 마지막 함수에서 KeyError가 뜹니다.
그래서 마지막 함수를 배제하고 두번째 함수까지만 실행을 시키고 모든 경우의 수를 계산한 hash_table을 출력해보면...
{1: {1: {1: 88, 2: 164, 3: 174}, 2: {1: 99, 2: 175, 3: 185}, 3: {1: 96, 2: 172, 3: 182}}, 2: {1: {}, 2: {}, 3: {}}, 3: {1: {}, 2: {}, 3: {}}}
이렇게 출력되는데 각각의 집을 칠하는 최솟값과 무관한 값들입니다.
그래서 코드를 뜯어보았는데요
color=1인 경우만 함수가 돌아가고 있으니
if color==1: 아래만 살펴보면
k가 0에서 N-1까지 반복하고 i가 0에서 N-1까지 반복하고 j가 0에서 N-1까지 반복하는 삼중 반복문이 되어 있습니다.
그러니까 첫번째 반복은 k,i,j가 모두 0인 상태일 것이고
color=1인 경우를 가정했으니까
#hash_table[1][1][1] = 26+each_house_value[0][0]+each_house_value[1][0]
이렇게 저장을 하게 되는데 each_house_value[0][0]이 26인데 26을 두 번 더할 이유가 없습니다. 26은 첫번째 집을 빨간색으로 칠하는 비용일 뿐이니 어떻게 최소비용을 구해도 한 번 나오거나 안 나오거나만 할 수 있으니까요.
따라서 예제의 정답인 96이 아예 출력되지 않는 코드라고 할 수 있는데요. 혹시 백준 사이트 문제 페이지의 파이썬 답안 제출 코너에 제출을 해보셨는지요?