백준-15686(파이썬)

https://www.acmicpc.net/problem/15686

15686번: 치킨 배달

 

 

문제생각

문제 그대로를 코드로 옮겼습니다. 

일단 집과 치킨집의 위치를 모두 뽑습니다.

그리고 M개를 제외한 다른 치킨집은 폐업이기에 combinations를 사용하여 나올 수 있는 치킨집 리스트를 뽑았습니다.

마지막으로 3중 for문을 이용하여 집과 m개 조합의 치킨집의 치킨거리를 구하고 이때 최소값도 구해줍니다.

 

문제코드

from itertools import combinations

n, m=map(int, input().split())
info=[list(map(int, input().split())) for _ in range(n)]
# 0:빈 칸, 1:집, 2:치킨집

home=[]
chicken=[]

#집 위치와 치킨집 위치 뽑아내기
for i in range(n):
    for j in range(n):
        if info[i][j]==1:
            home.append([i, j])
        elif info[i][j]==2:
            chicken.append([i, j])

#치킨집 조합 만들기
chic_info=list(combinations(chicken, m))

ans=1e9 #정답

#최소치킨거리 구하기
for chi in chic_info:
    Sum=0
    for h in home:
        temp=1e9
        for c in chi:
            result=(abs(c[0]-h[0])+abs(c[1]-h[1]))
            if temp>result:
                temp=result
        Sum+=temp
        if Sum>ans:
            break
    if Sum<ans:
        ans=Sum
        
print(ans)

1초의 시간제한 때문에 시간초과를 받지 않을까 생각도 했지만 입력받는 n, m의 범위가 그렇게 크지않아 통과가 되었는 것 같습니다.