7-kokeunho

[kokeunho]

🔗 문제 링크

[BOJ] 치킨 배달(https://www.acmicpc.net/problem/15686)

✔️ 소요된 시간

1h30m

✨ 수도 코드

브루트포스로 풀었습니다. 혼자서 푼 건 아니고요...

1. 도시의 집과 치킨집을 입력 받으면서 집과 치킨집의 좌표를 각각 리스트에 넣어줍니다.
2. M개의 치킨집이 선택되는 모든 조합을 확인합니다.
3. 조합마다 각 집에서 가까운 치킨집까지의 거리의 합(total_chicken_distance)를 구하고
   기존의 최단 거리(min_distance)와 비교합니다.
5. 모든 조합에 대하여 거리 비교가 끝나면 결과를 출력합니다.

📚 새롭게 알게된 내용

브루트포스 알고리즘을 사용하는 법을 익힌 것 같습니다.
재귀적으로 메서드를 호출해서 조합을 만들어내는게 단박에 이해가 가지 않았는데
어떻게 이해가 됐습니다..

중요한 내용은 아닙니다만

distance = |hx-sx| + |hy-sy| 

처럼 좌표 1, 2의 거리를 맨해튼 거리라고 한다고 합니다.

Manhattan Distance = |x1-x2| + |y1-y2|

math 라이브러리에서 제공하는 abs는 값의 절대값을 구해주어서
편하게 거리를 계산할 수 있었습니다.

[kangrae-jo]

구현 완전 빡세네요..

한 50분 걸려서 완성했는데 계속 테스트를 통과 못하길래 근호님 코드 봤습니다.

다른 부분은 다 비슷한데 다른점은
저는 house를 (제 코드에서는 home) 큐로 뒀었는데,
이걸 좀 이상하게 처리했었나봐요.

그리고 dfs돌리는 부분도 2중 for문으로 좀 복잡했는데,
근호님 처럼 바꾸니 잘 동작하는군요.,.

고생하셨습니다~

CPP CODE

#include <algorithm>
#include <climits>
#include <iostream>
#include <queue>
#include <vector>

#define EMPTY 0
#define HOME 1
#define CHICKEN 2

using namespace std;

int N, M, result = INT_MAX;
vector<pair<int, int>> chicken;
queue<pair<int, int>> home;

int getDistance(int x, int y, int x_, int y_) {
    return abs(x - x_) + abs(y - y_);
}

int calculateDistance(vector<pair<int, int>>& selected) {
    int sumDistance = 0;
    queue<pair<int, int>> q = home;

    while (!q.empty()) {
        int y_ = q.front().first;
        int x_ = q.front().second;
        q.pop();

        int minDistance = INT_MAX;
        for (auto& c : selected) {
            int distance = getDistance(x_, y_, c.second, c.first);
            minDistance = min(minDistance, distance);
        }
        sumDistance += minDistance;
    }

    return sumDistance;
}

void dfs(vector<pair<int, int>>& selected, int index, int count) {
    if (count == M) result = min(result, calculateDistance(selected));

    for (int i = index; i < chicken.size(); i++) {
        selected.push_back(chicken[i]);
        dfs(selected, i + 1, count + 1);
        selected.pop_back();
    }
}

int main() {
    cin >> N >> M;

    for (int y = 0; y < N; y++) {
        for (int x = 0; x < N; x++) {
            int temp;
            cin >> temp;
            if (temp == HOME) home.push({y, x});
            if (temp == CHICKEN) chicken.push_back({y, x});
        }
    }

    vector<pair<int, int>> selected;
    dfs(selected, 0, 0);

    cout << result;

    return 0;
}

[kangrae-jo]

여기서 city 2차원 배열은 안써도 됐을 것 같아요.
city의 쓰임은 입력말고는 없으니까, int로 대체해도 될 듯 합니다!

[g0rnn]

아 그렇네요! 저도 무의식적으로 2차원 배열을 사용했는데 정말 한번도 안썼어요..ㅋㅋ

[g0rnn]

이 문제를 풀면서 이전에 제가 올린 연구소? 문제가 생각나더군요. 그래서 구현이 빨리 끝날줄 알았으나.. 시간초과로 인해 좀 더 썼습니다.

문제에서 메모리 제한이 512이길래 너무 여유로워서 city도 2차원 배열로 설정하고 할 수 있는건 전부 배열로 설정해버렸네요..ㅋㅋㅋ 그래서 dfs 내부에서 이중 for문을 사용하게 되었고 시간초과가 났습니다.

이후엔 치킨 가게들을 배열에 넣어서 for문을 사용해서 구현했는데 dfs를 실행할 때마다 0부터 끝까지 치킨 가게를 순회하게 되어 중복되는 조합이 많아져 시간초과가 한번 더 떴습니다.

최종적으로 dfs할 때 index를 넘겨주어 선택하는 치킨가게의 조합이 중복되지 않도록 구헌했습니다. (이 과정에서 survivedMap은 지워도 되었는데 까먹고 수정안했습니다..)

제가 구현한 코드입니다

#include <iostream>
#include <cmath>
#include <vector>
#include <climits>
using namespace std;

#define MAX 50
#define HOUSE 1
#define BHC 2
#define SURVIVED 3
#define NONE 0

int n, m;
int city[MAX + 5][MAX + 5] = { 0, };
int survivedMap[MAX + 5][MAX + 5] = { 0, };
int minimum = INT_MAX;
vector<pair<int, int>> survived;
vector<pair<int, int>> houses;
vector<pair<int, int>> bhcs;

int getChickenLength(pair<int, int> house, pair<int, int> bhc) {
	return abs(house.first - bhc.first) + abs(house.second - bhc.second);
}

void init() {
	cin >> n >> m;
	for (int y = 0; y < n; y++) {
		for (int x = 0; x < n; x++) {
			int k;
			cin >> k;
			city[y][x] = k;
			if (k == HOUSE) houses.push_back({ x, y });
			if (k == BHC) bhcs.push_back({ x, y });
		}
	}
}

void setMinimum() {
	int total = 0;
	for (auto house : houses) {
		int len = INT_MAX;
		for (auto s : survived) {
			//치킨 거리(집과 가장 가까운 치킨집 사이의 거리)를 구함
			len = min(len, getChickenLength(house, s));
		}
		total += len;
	}
	minimum = min(total, minimum);
}

void dfs(int index) {
	if (survived.size() == m) {
		setMinimum();
		return;
	}
	for (int i = index; i < bhcs.size(); i++) {
		//survived에 포함되어 있지 않을 때
		if (survivedMap[bhcs[i].second][bhcs[i].first] == NONE) {
			survived.push_back(bhcs[i]);
			survivedMap[bhcs[i].second][bhcs[i].first] = SURVIVED;
			dfs(i + 1);
			survived.pop_back();
			survivedMap[bhcs[i].second][bhcs[i].first] = NONE;
		}
	}
}

int main() {
	init();
	dfs(0);
	cout << minimum;
	return 0;
}

[g0rnn]

아 그렇네요! 저도 무의식적으로 2차원 배열을 사용했는데 정말 한번도 안썼어요..ㅋㅋ

[g0rnn]

여기서 치킨집을 선택할 때 index로 접근하도록 했네요. 저는 그냥 pair를 때려 박았는데 되게 참신한 접근인 것 같습니다 👍