graph/TSP.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

//nodes[0] has to be the start and end node.
vector<vector<int>> dist;
vector<int> TSP() {
	int n = dist.size(), m = 1 << n;
	vector<vector<ii>> dp(n, vector<ii>(m, ii(MAX_N, -1)));
	
	for(int c = 0; c < n; c++) dp[c][m-1].first = dist[c][0], dp[c][m-1].second = 0;

	for(int v = m - 2; v >= 0; v--) {
		for(int c = n - 1; c >= 0; c--) {
			for(int g = 0; g < n; g++) { 
				if(g != c && (((1 << g) & v) == 0)) {
					if((dp[g][(v | (1 << g))].first + dist[c][g]) < dp[c][v].first) {
						dp[c][v].first = dp[g][(v | (1 << g))].first + dist[c][g];
						dp[c][v].second = g;
					}
				}
			}
		}
	}
	
	vector<int> res; res.push_back(0); int v = 0;
	while(res.back() != 0 || res.size() == 1) {
		res.push_back(dp[res.back()][(v |= (1 << res.back()))].second);
	}
	
	return res;
}