[C++] 백준 15686번: 치킨 배달

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

15686번: 치킨 배달

1. 문제풀이

 0은 빈 칸, 1은 집, 2는 치킨집이다. 최대 M개의 치킨집을 제외하고 나머지 치킨집은 폐업시켜야 한다. 치킨집의 개수가 최대일 때, 도시의 치킨 거리가 최솟값인 것은 자명하다. 도시에서 M개의 치킨집을 정하여, 각 집마다 거리의 합을 구하여, 최솟값을 갱신하면 된다. 도시에서 치킨집을 고르는 것은 조합인 것을 확인하고, 깊이 우선 탐색 방법으로 구현하였다. 깊이가 M일 때 M개의 치킨집을 고른 것이므로, 각 집마다 치킨집을 탐색하여 가장 가까운 치킨집과의 거리를 찾아 최소 거리를 구한 후, 거리의 총합을 구하였다. 거리의 총합이 이전에 구한 거리의 총합보다 작을 경우 새로 갱신하였다. 도시의 집과 치킨집의 좌표를 담을 때 배열이 아닌 vector<pair<int, int>>를 사용하여 구현하여도 된다.

2. C++ 코드

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <limits.h>

using namespace std;

int n, m, houseCnt, chickenCnt, res = INT_MAX;
int houseY[100], houseX[100];
int chickenY[13], chickenX[13], visited[13], selected[13];

void dfs(int depth, int cur)
{
    if (depth == m) {
        int dis, minDis, sum = 0;
        for (int i = 0; i < houseCnt; i++) {
            minDis = INT_MAX;
            for (int j = 0; j < m; j++) {
                dis = abs(houseY[i] - chickenY[selected[j]]) + abs(houseX[i] - chickenX[selected[j]]);
                if (dis < minDis) {
                    minDis = dis;
                }
            }
            sum += minDis;
        }
        if (sum < res) {
            res = sum;
        }
    } else {
        for (int i = cur; i < chickenCnt; i++) {
            if (!visited[i]) {
                visited[i] = 1;
                selected[depth] = i;
                dfs(depth + 1, i);
                visited[i] = 0;
            }
        }
    }
}

int main(void)
{
    int tmp;
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            scanf("%d", &tmp);
            if (tmp == 1) {
                houseY[houseCnt] = i;
                houseX[houseCnt++] = j;
            } else if (tmp == 2) {
                chickenY[chickenCnt] = i;
                chickenX[chickenCnt++] = j;
            }
        }
    }
    dfs(0, 0);
    printf("%d\n", res);

    return 0;
}