[BOJ 10971 // C++] 외판원 순회 2

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 10971번 문제인 외판원 순회 2이다.
문제는 아래 링크를 확인하자.

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

10971번: 외판원 순회 2

2098번 문제에서 입력제한이 작아진 문제이다. 해당 문제의 풀이글을 참고하자.

 

입력제한이 작으므로 직접 O(N!)의 모든 가능한 순회경로를 탐색하는 브루트포스 풀이 또한 통과할 수 있다.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <queue>
using namespace std;

int N;
int dist[10][10];
int dp[1024][10];
bool visited[1024];

void bfs() {
	visited[0] = visited[1] = 1;
	dp[1][0] = 1;
	queue<int> que;
	que.push(1);
	while (!que.empty()) {
		int cur = que.front(); que.pop();
		for (int s = 0; s < N; s++) {
			if (dp[cur][s] == 0) continue;
			int sidx = (1 << s);
			if ((cur & sidx) == 0) continue;
			for (int e = 0; e < N; e++) {
				int eidx = (1 << e);
				if (cur & eidx) continue;
				if (dist[s][e] == 0) continue;
				
				if (dp[cur ^ eidx][e] == 0) dp[cur ^ eidx][e] = dp[cur][s] + dist[s][e];
				else dp[cur ^ eidx][e] = min(dp[cur ^ eidx][e], dp[cur][s] + dist[s][e]);
				if (!visited[cur ^ eidx]) {
					visited[cur ^ eidx] = 1;
					que.push(cur ^ eidx);
				}
			}
		}
	}
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N;
	for (int r = 0; r < N; r++) {
		for (int c = 0; c < N; c++) cin >> dist[r][c];
	}

	bfs();

	int ans = 1000000007;
	int MX = (1 << N) - 1;
	for (int i = 0; i < N; i++) {
		if (dp[MX][i] == 0) continue;
		if (dist[i][0] == 0) continue;
		ans = min(ans, dp[MX][i] + dist[i][0]);
	}

	cout << ans - 1;
}