백준 21278번 호석이 두 마리 치킨 C++

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

21278번: 호석이 두 마리 치킨

 

플로이드 워셜로 각 건물과의 거리를 전부 구하고 백트래킹을 통해 두개의 치킨집과의 왕복 거리의 합이 가장 적은 두 건물을 골랐다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, m;
vector<vector<int>> graph;

vector<bool> visited;
vector<int> idxes;

int node1, node2;
int minDist = 0x7FFFFFFF;

void BT(int idx, int cnt) {
    if (cnt == 2) {
        vector<int> tmp(n + 1, 0x7FFFFFFF);
        for (auto a : idxes) {
            for (int i = 1; i <= n; i++) {
                if (a == i) tmp[i] = 0;
                tmp[i] = min(tmp[i], graph[a][i] * 2);
            }
        }
        int sum = 0;
        for (int i = 1; i <= n; i++) sum += tmp[i];
        if (sum < minDist) {
            node1 = idxes[0];
            node2 = idxes[1];
            minDist = sum;
        }
        return;
    }
    for (int i = idx; i <= n; i++) {
        if (visited[i]) continue;
        visited[i] = true;
        idxes.push_back(i);
        BT(i + 1, cnt + 1);
        visited[i] = false;
        idxes.pop_back();
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> m;
    graph = vector<vector<int>>(n + 1, vector<int>(n + 1, -1));
    visited = vector<bool>(n + 1, false);
    for (int i = 0; i < m; i++) {
        int s, e;
        cin >> s >> e;
        graph[s][e] = 1;
        graph[e][s] = 1;
    }
    for (int k = 1; k <= n; k++) {
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                if (graph[i][k] == -1 || graph[k][j] == -1) continue;
                if (graph[i][j] == -1) graph[i][j] = graph[i][k] + graph[k][j];
                else graph[i][j] = min(graph[i][j], graph[i][k] + graph[k][j]);
            }
        }
    }
    BT(1, 0);
    cout << node1 << " " << node2 << " " << minDist;
}