forked from ncduy0303/Competitive-Programming
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Traveling Salesman Problem (TSP).cpp
57 lines (47 loc) · 1.63 KB
/
Traveling Salesman Problem (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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
// Given n cities/nodes, find a minimum weight Hamiltonian Cycle/Tour
// Time complexity: O(2^n * n^2)
// Problem link: https://vjudge.net/problem/UVA-10496
#include <bits/stdc++.h>
using namespace std;
#define ar array
#define ll long long
const int MAX_N = 11;
const int MOD = 1e9 + 7;
const int INF = 1e9;
const ll LINF = 1e18;
int n, x[MAX_N], y[MAX_N], adj[MAX_N][MAX_N], dp[1 << MAX_N][MAX_N];
int tsp(int mask, int u) {
if (mask == (1 << n) - 1) return adj[u][0]; // finish all nodes (bitmask == 11...11)
if (dp[mask][u] != -1) return dp[mask][u];
int ans = INF;
// Visit all the unvisited nodes and take the best route
for (int v = 0; v < n; v++) {
if (!(mask & (1 << v))) { // this node is unvisited
int cur = adj[u][v] + tsp(mask | (1 << v), v);
ans = min(ans, cur);
}
}
return dp[mask][u] = ans;
}
void solve() {
int xs, ys; cin >> xs >> ys; // not important
cin >> x[0] >> y[0] >> n; n++;
for (int i = 1; i < n; i++) cin >> x[i] >> y[i];
for (int i = 0; i < n; i++)
for (int j = i; j < n; j++)
adj[i][j] = adj[j][i] = abs(x[i] - x[j]) + abs(y[i] - y[j]);
memset(dp, -1, sizeof dp);
// start from node 0 (bitmask = 00...01)
cout << "The shortest path has length " << tsp(1, 0) << "\n";
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
// freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
int tc; cin >> tc;
for (int t = 1; t <= tc; t++) {
// cout << "Case #" << t << ": ";
solve();
}
}