13-froglike6

[froglike6]

🔗 문제 링크

치킨 배달

✔️ 소요된 시간

30분

✨ 수도 코드

수도 코드

    N, M ← 정수 입력                    ▸ 그리드 크기 N, 선택할 치킨집 수 M

    houses ← 빈 리스트                   ▸ 집 좌표 저장용
    chickens ← 빈 리스트                 ▸ 치킨집 좌표 저장용

    for r from 0 to N−1                  ▸ 각 행 순회
        row ← 정수 리스트 입력          ▸ r번째 행 읽기
        for 각 i, v in enumerate(row)    ▸ 각 열 값 검사
            만약 v = 1이면
                houses에 (r, i) 추가    ▸ 집 위치 기록
            아니면 만약 v = 2이면
                chickens에 (r, i) 추가  ▸ 치킨집 위치 기록

    min_city_dist ← ∞                    ▸ 최소 도시 치킨 거리 초기화

    for comb in combinations(chickens, M) ▸ 치킨집 M개 조합 생성
        city_dist ← 0                   ▸ 이 조합의 거리 합 초기화
        for 각 (hr, hc) in houses       ▸ 모든 집에 대해
            dist ← min(                ▸ 집→치킨집 최단 거리 계산
                |hr−cr| + |hc−cc| 
                for 각 (cr, cc) in comb
            )
            city_dist ← city_dist + dist
            만약 city_dist ≥ min_city_dist이면
                break

        만약 city_dist < min_city_dist이면
            min_city_dist ← city_dist   ▸ 최솟값 갱신

    min_city_dist 출력                   ▸ 결과 출력

이 문제에서 치킨 거리는, 집에서부터 가까운 치킨집까지의 맨해튼 거리입니다. 도시에 여러 집과 여러 치킨집이 있는데, 여기서 만들어질 수 있는 치킨 거리의 총 합이 도시의 치킨 거리입니다.
따라서 문제에서 구하고자 하는 것은, 도시에서 치킨집을 최대 $M$개 없앨 때의 최소 도시의 치킨 거리를 구하라는 문제입니다.

저는 일단 좌표들을 입력 받고, enumerate를 이용해 각 집과 치킨집의 좌표를 저장했습니다. 그런 다음 combination을 사용해 모든 가능한 "남은 치킨집에 대한 치킨 거리"를 구하여 최소 도시의 치킨 거리를 구했습니다.

📚 새롭게 알게된 내용

Enumerate를 새롭게 알게 되었습니다. 다음과 같이 사용하면, 요소의 값과 함께 그 인덱스도 같이 반환해주는 역할입니다.

>>> seasons = ['Spring', 'Summer', 'Fall', 'Winter']
list(enumerate(seasons))
[(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]
>>> list(enumerate(seasons, start=1))
[(1, 'Spring'), (2, 'Summer'), (3, 'Fall'), (4, 'Winter')]

[dohyeondol1]

후....2시간 정도 고민한 것 같은데
집, 치킨집의 좌표를 받아내고 써먹는 부분에서 막혀서 결국 풀이를 보고 나서야 이해했네요..

백트래킹으로 풀어야겠다는 생각은 단번에 못했습니다.
조금 아쉽네요... 백트래킹 발상을 빨리 했더라면..

C++

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

int N, M;
vector<pair<int, int>> houses, chickens;
bool selected[14];
int answer = 1000000000;

void calculateChickenDistance() {
    int total = 0;
    for(auto [hx, hy] : houses) {
        int distance = 1000000000;
        for(int i = 0; i < chickens.size(); i++) {
            if(selected[i]) {
                int cx = chickens[i].first;
                int cy = chickens[i].second;
                int d = abs(hx - cx) + abs(hy - cy);
                if(d < distance) distance = d;
            }
        }
        total += distance;
    }
    if(total < answer) answer = total;
}

void dfs(int idx, int cnt) {
    if(cnt == M) {
        calculateChickenDistance();
        return;
    }

    for(int i = idx; i < chickens.size(); i++) {
        selected[i] = true;
        dfs(i + 1, cnt + 1);
        selected[i] = false;
    }
}

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);

    cin >> N >> M;
    for(int i = 0; i < N; i++) {
        for(int j = 0; j < N; j++) {
            int val;
            cin >> val;
            if(val == 1) houses.push_back({i, j});
            else if(val == 2) chickens.push_back({i, j});
        }
    }

    dfs(0, 0);
    cout << answer << '\n';

    return 0;
}

풀이를 보고 나니 백트래킹의 익숙한 얼굴이 보이더군요..
백트래킹 유형을 손에 익혀야 할 것 같습니다.

이 문제는 묵혀뒀다가 풀이를 까먹을 쯤에 다시 도전해봐야겠네요.

[hadongun]

음.. 문제를 보고 어떻게 구현해야 할까.. 계속 생각해봤어요

집과 치킨집의 좌표를 저장하고, 치킨집을 m개 뽑은 후 각 치킨 거리를 구하여 최솟값을 구한다면?
이라고 생각했는데 구현을 하기가 쉽지 않더군요.. 코드를 쓰는데 자꾸 막히는?
도현님의 수도 코드를 보니까 방향은 맞았던 것 같네요

조금 더 수련해서 다시 치킨 배달에 도전해보겠습니다..!

enumerate는 매우 유용하네요!! 기억했다가 잘 써보겠습니다

[caucsejunseo]

좌표만 보면 머리가 띵해져서 어떻게 풀어야할 지 감이 안오네요.
맨날 이것 저것 머릿속으로 생각만 하고 손이 안움직여지네요

경우의 수가 너무 많아보여서 전부다 해보는 건 빠르게 포기했었는데 생각보다 시간복잡도가 크지 않은 문제였네요. 시간 복잡도 계산이 익숙하지 않아서 그냥 머리로 되겠다, 안되겠다 생각했었는데 손으로 써보며 시간복잡도 계산에 조금 더 익숙해져야 할 거 같습니다.

dfs, 백트래킹, 좌표... 전부 더 익숙하게 만든 뒤 다시 풀어보겠습니다...!