给定n个城市(从0到n-1),3元组[A, B, C]表示城市A和城市B之间存在道路,且成本为C。计算从0号城市出发,旅行完每个城市一遍,最后回到0号城市的最小成本与路径。如果不存在最优方案,输出-1.用c++代码实现
时间: 2023-11-22 07:56:01 浏览: 90
用c++语言实现三元组
以下是使用C++语言实现的代码:
```cpp
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
const int INF = 1e9; // 定义正无穷
int n; // 城市数
vector<vector<int>> graph; // 邻接矩阵
int dp[1 << 16][16]; // dp数组,dp[S][i]表示集合S中包含i点时,从i点出发遍历完集合S中所有点后回到0点的最小成本
// 注意:由于n <= 15,因此可以用1 << 16来表示所有的状态
int tsp() {
memset(dp, 0x3f, sizeof(dp)); // 初始化dp数组为正无穷
dp[1][0] = 0; // dp[1][0]表示只包含0点的集合,成本为0
// 枚举所有的集合S
for (int S = 1; S < (1 << n); S++) {
// 枚举所有的点i
for (int i = 0; i < n; i++) {
// 如果集合S不包含点i,则跳过
if (!(S & (1 << i))) {
continue;
}
// 枚举集合S中除去点i之外的所有点j
for (int j = 0; j < n; j++) {
if (i == j || !(S & (1 << j))) {
continue;
}
dp[S][i] = min(dp[S][i], dp[S ^ (1 << i)][j] + graph[j][i]);
}
}
}
// 遍历完所有点后回到0点的最小成本
int res = INF;
for (int i = 1; i < n; i++) {
res = min(res, dp[(1 << n) - 1][i] + graph[i][0]);
}
if (res == INF) { // 如果不存在最优方案
return -1;
}
return res;
}
int main() {
int m;
cin >> n >> m;
// 初始化邻接矩阵
graph.resize(n);
for (int i = 0; i < n; i++) {
graph[i].resize(n, INF);
}
for (int i = 0; i < m; i++) {
int a, b, c;
cin >> a >> b >> c;
graph[a][b] = c;
}
cout << tsp() << endl;
return 0;
}
```
代码思路:
1. 使用邻接矩阵存储图信息;
2. 使用dp[S][i]表示集合S中包含i点时,从i点出发遍历完集合S中所有点后回到0点的最小成本;
3. 初始化dp数组为正无穷,dp[1][0]表示只包含0点的集合,成本为0;
4. 枚举所有的集合S,枚举所有的点i,枚举集合S中除去点i之外的所有点j,更新dp[S][i];
5. 遍历完所有点后回到0点的最小成本为:min(dp[(1 << n) - 1][i] + graph[i][0]),其中i不等于0;
6. 如果不存在最优方案,输出-1。
阅读全文