kingsubin

• 치킨 거리를 구하는 것은 O(1)로 가능
• 집에서 치킨집 까지의 거리를 모두 구한 리스트를 만들고 선택한 M 개의 치킨집까지의 거리 중에서 최솟값의 합을 구하면 nCm 개의 도시의 치킨거리가 나올텐데 그 중에 최솟값이 ans 라고 생각
• 최악의 경우를 고려해도 시간초과가 나지 않아 보였음.
• 13Cm 인데 여기서 m은 1~13, Combination 연산 최댓값 생각해도 13C7
• 대략 O(집 100 * 치킨집 7 * 13C7)
• sol 1 로 제출하고 다른 모범 답안 참조하여 sol 2로 수정함.

 

# sol 1
import sys
from itertools import combinations
 
N, M = map(int, input().split())
board = [list(map(int, input().split())) for _ in range(N)] # 여기서 board는 필요없었음.
 
home = []
chicken = []
for i in range(N):
    for j in range(N): # enumerate 사용 생각 했어야함.
        if board[i][j] == 1:
            home.append([i, j])
        elif board[i][j] == 2:
            chicken.append([i, j])
 
# 굳이 모든 dist 를 먼저 구할 필요가 없긴 했음.
dist = []
for i in range(len(home)):
    d = []
    for j in range(len(chicken)):
        d.append(abs(home[i][0] - chicken[j][0]) + abs(home[i][1] - chicken[j][1]))
    dist.append(d)
 
ans = sys.maxsize
for selected in combinations([_ for _ in range(len(chicken))], M):
    total = 0
 
        # total 구하는 과정이 많이 구렸음. sol2 처럼 하는게 좋은듯.
    for d in dist:
        chicken_dist = sys.maxsize
        for j in selected:
            chicken_dist = min(chicken_dist, d[j])
 
        total += chicken_dist
 
    ans = min(ans, total)
 
print(ans)

# sol 2
import sys
from itertools import combinations
 
N, M = map(int, input().split())
 
homes = []
chickens = []
for i in range(N):
    for j, v in enumerate(map(int, input().split())):
        if v == 1:
            homes.append((i, j))
        elif v == 2:
            chickens.append((i, j))
 
 
def get_dist(a, b):
    return abs(a[0] - b[0]) + abs(a[1] - b[1])
 
 
ans = sys.maxsize
for selected in combinations(chickens, M):
    total = 0
 
    for house in homes:
        total += min(get_dist(house, chicken) for chicken in selected)
 
    ans = min(ans, total)
 
print(ans)