말랑말랑한 개발자 이야기

[백준 10971번] 외판원 순회2 본문

알고리즘/백준

[백준 10971번] 외판원 순회2

말랑너구리 2021. 6. 22. 00:07

[백준 10971번] 외판원 순회2

www.acmicpc.net/problem/10971

 

10971번: 외판원 순회 2

첫째 줄에 도시의 수 N이 주어진다. (2 ≤ N ≤ 10) 다음 N개의 줄에는 비용 행렬이 주어진다. 각 행렬의 성분은 1,000,000 이하의 양의 정수이며, 갈 수 없는 경우는 0이 주어진다. W[i][j]는 도시 i에서 j

www.acmicpc.net

 

 

 문제

 외판원 순회 문제는 영어로 Traveling Salesman problem (TSP) 라고 불리는 문제로 computer science 분야에서 가장 중요하게 취급되는 문제 중 하나이다. 여러 가지 변종 문제가 있으나, 여기서는 가장 일반적인 형태의 문제를 살펴보자.

 1번부터 N번까지 번호가 매겨져 있는 도시들이 있고, 도시들 사이에는 길이 있다. (길이 없을 수도 있다) 이제 한 외판원이 어느 한 도시에서 출발해 N개의 도시를 모두 거쳐 다시 원래의 도시로 돌아오는 순회 여행 경로를 계획하려고 한다. 단, 한 번 갔던 도시로는 다시 갈 수 없다. (맨 마지막에 여행을 출발했던 도시로 돌아오는 것은 예외) 이런 여행 경로는 여러 가지가 있을 수 있는데, 가장 적은 비용을 들이는 여행 계획을 세우고자 한다.

 각 도시간에 이동하는데 드는 비용은 행렬 W[i][j]형태로 주어진다. W[i][j]는 도시 i에서 도시 j로 가기 위한 비용을 나타낸다. 비용은 대칭적이지 않다. 즉, W[i][j] 는 W[j][i]와 다를 수 있다. 모든 도시간의 비용은 양의 정수이다. W[i][i]는 항상 0이다. 경우에 따라서 도시 i에서 도시 j로 갈 수 없는 경우도 있으며 이럴 경우 W[i][j]=0이라고 하자.

 N과 비용 행렬이 주어졌을 때, 가장 적은 비용을 들이는 외판원의 순회 여행 경로를 구하는 프로그램을 작성하시오.

 

 입력

 첫째 줄에 도시의 수 N이 주어진다. (2 ≤ N ≤ 10) 다음 N개의 줄에는 비용 행렬이 주어진다. 각 행렬의 성분은 1,000,000 이하의 양의 정수이며, 갈 수 없는 경우는 0이 주어진다. W[i][j]는 도시 i에서 j로 가기 위한 비용을 나타낸다.

 항상 순회할 수 있는 경우만 입력으로 주어진다.

 

 출력

 첫째 줄에 외판원의 순회에 필요한 최소 비용을 출력한다.

 

 

 풀이

#include <iostream>
using namespace std;

int N, arr[10][10];
int target = 2147483647;
bool check[10];
int cnt = 0;

void ff(int now, int cost){
	if(cnt == N && now == 0){
		target = min(target, cost);
		return;
	}
	
	for(int i=0;i<N;i++){
		if(arr[now][i] == 0) continue;
		if(check[i] == false){
			check[i] = true; cnt++;
			ff(i, cost + arr[now][i]);
			check[i] = false; cnt--;
		}
	}
	
	return;
}
int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
	
	cin >> N;
	for(int i=0;i<N;i++){
		for(int j=0;j<N;j++){
			cin >> arr[i][j];
		}
	}
	ff(0, 0);
	cout << target;
	return 0;
}

 외판원 순회, TSP 문제라고 불리는 꽤나 유명한 문제이다. 도시의 수(N)의 범위에 따라 푸는 방법이 다르다던데 이 문제는 N의 범위가 최대 10까지라 DFS로도 충분히 풀리는 문제이다. DFS를 할 때에는 재귀를 이용하되, check와 같은 배열을 이용하여 for문( true -> 재귀 -> false)의 구조로 작성하면 된다. 이는 조합문제에서 사용되는 방법과 같다.

 DFS를 이용하여 풀면 같은 과정을 반복하게 되는데 이를 메모이제이션하여 DP로 푸는 것이 N의 범위가 더 클때 사용하는 방법인 것 같다. 시간될 때 풀어봐야겠다.