BOJ: #15686 - 치킨 배달

ㅋ..ㅋㅋㅋ… 제발!!!!!!!! 지문 좀 잘 읽자!!!!!!!!!!!!!!

임의의 두 칸 (r1, c1)과 (r2, c2) 사이의 거리는 |r1-r2| + |c1-c2|로 구한다.

(r1, c1), (r2, c2)를 (r1, r2), (c1, c2)로 읽어서 삽질 오지게 했다.

문제

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

내가 작성한 코드

Swift

let input = readLine()!.split(separator: " ").map { Int(String($0))! }
let n = input[0]
let m = input[1]
var board = [[Int]]()
var chicken = [(Int, Int)]()
var house = [(Int, Int)]()
for _ in 0..<n {
    let line = readLine()!.split(separator: " ").map { Int(String($0))! }
    board.append(line)
}
for i in 0..<n {
    for j in 0..<n {
        if board[i][j] == 1 {
            house.append((i + 1, j + 1))
        } else if board[i][j] == 2 {
            chicken.append((i + 1, j + 1))
        }
    }
}
 
// 조합 구하기
var results = [[(Int, Int)]]()
var picked = [(Int, Int)]()
var visited = Array(repeating: false, count: chicken.count)
func pick(toPick: Int, start: Int) {
    if toPick == 0 {
        results.append(picked)
        return
    }
 
    for i in start..<chicken.count {
        if visited[i] {
            continue
        }
        visited[i] = true
        picked.append(chicken[i])
        pick(toPick: toPick - 1, start: i + 1)
        picked.removeLast()
        visited[i] = false
    }
}
 
pick(toPick: m, start: 0)
 
var ans = 987654321
for result in results {
    var sum = 0
    for h in house {
        var distance = 987654321
        for c in result { // 한 집의 치킨 거리 구하기
            let dist = abs(c.0 - h.0) + abs(c.1 - h.1)
            distance = min(distance, dist)
        }
        sum += distance
    }
    ans = min(sum, ans)
}
print(ans)