14-H0ngJu

[H0ngJu]

🔗 문제 링크

치킨 게임

✔️ 소요된 시간

2H

✨ 수도 코드

백트래킹이 개념적으로 뭔지는 알겠는데 문제 적용이 힘든 것 같아서 요 문제를 골랐습니다 허허

문제 요약
문제에서 제시된 치킨 집 개수를 M개만 살려두었을 때,
전체 치킨 거리가 가장 작은 M개의 조합을 구하여 치킨 거리의 합을 반환하자@!

이 문제를 풀면서 가장 처음 풀어야하는 문제는

어떻게 가능한 M개의 치킨 집 조합을 찾을 것인가? 입니다.

처음에는 M개가 될 때까지 1개씩 줄여나가면서 최소 치킨 거리를 찾을까? 했는데..

뭔가 배보다 배꼽이 커지는 느낌이라 빠르게 pass 했습니다.

근데 조합에 대해서 찾아보니까 파이썬에서 제공하는 itertools가 있길래 import해서 사용했습니다

* combinations(list, N)을 하게되면 반환 값은 tuple이 됩니다.

그렇게 치킨 집을 저장해둔 list에 대하여 combinations를 하면 손쉽게 모든 조합의 경우의 수를 구할 수 있습니다.

이후에는 모든 집들을 돌며, 모든 치킨 집에 대한 치킨 거리를 구합니다.

저는 여기서 굳이 모든 치킨집에 대한 거리를 구할 필요가 있나? 어느 순간 확인하지 않아도 되는 치킨집이 있을 것 같은데.. 라고 생각해서 뺑뺑 둘러갔지만,,,

결론부터 말하면 그냥 한 집에 대하여 모든 치킨 집에 대한 치킨 거리를 구하고, min을 반환하는 것이 가장 간단했습니다.

그렇게 각 조합에 대한 치킨 거리 합을 구하고, min을 반환해주면 문제를 풀 수 있습니다!

로직 짜는데 몇번 삽질하고 tmp 초기화를 엉뚱한 곳에 해서 시간을 많이 썼네요 허허 😅😅

분명 어제는 쪼매 어려웠는데 오늘 보니까 그정도는 아닌 것 같기도? 하네요?

import sys
from itertools import combinations

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

N, M = map(int, input().split())
house = [[] for _ in range(N)]
chicken_house = []
result = 99999

for i in range(N):
    house[i].extend(map(int, input().split()))

for r in range(N):
    for c in range(N):
        if house[r][c] == 2:
            chicken_house.append((r, c))

for chickens in combinations(chicken_house, M):
    total = 0
    for r in range(N):
        for c in range(N):
            if house[r][c] == 1:
                tmp = 9999
                for chicken in chickens:
                    tmp = min(tmp, abs(r - chicken[0]) + abs(c - chicken[1]))
                total += tmp
    
    result = min(result, total)

print(result)

📚 새롭게 알게된 내용

[tgyuuAn]

백트래킹으로 구현하다가 어디서 잘못됐는지....

홍주르 PR 보고 combinations 보고 이마 탁쳤씁니다.........................

리마인드 시켜주셔서 감사합니다 ............ ㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠ 😢

[tgyuuAn]

와 진짜 combinations 안떠올라가지고 백트래킹으로 구현하고 있었는데..

시간초과나서 화딱지나서 PR봤는데 진짜 허무하네요.................

진짜 감 다죽었따........... ㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠ

[H0ngJu]

저어도 combinations 알게되기 전까지 많은 삽질이 있었다죠 .. ㅋ..ㅋ.ㅋ..