请你用C语言或者c++写出动态规划法求最短路径问题的代码
时间: 2024-03-06 14:52:02 浏览: 100
下面是一个使用C++语言实现的动态规划法求解最短路径问题的代码:
```cpp
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
const int MAXN = 1005;
const int INF = 0x3f3f3f3f;
vector<pair<int, int>> adj[MAXN]; // 邻接表表示图
int d[MAXN]; // 存储起点到各点的最短路径长度
bool vis[MAXN]; // 标记每个点是否已经访问过
void dijkstra(int s, int n) // s表示起点,n表示节点总数
{
memset(vis, false, sizeof(vis));
memset(d, INF, sizeof(d));
d[s] = 0;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
pq.push(make_pair(0, s));
while (!pq.empty()) {
int u = pq.top().second;
pq.pop();
if (vis[u]) continue;
vis[u] = true;
for (int i = 0; i < adj[u].size(); i++) {
int v = adj[u][i].first;
int w = adj[u][i].second;
if (d[v] > d[u] + w) {
d[v] = d[u] + w;
pq.push(make_pair(d[v], v));
}
}
}
}
int main()
{
int n, m, s, t;
cin >> n >> m >> s >> t;
for (int i = 1; i <= m; i++) {
int u, v, w;
cin >> u >> v >> w;
adj[u].push_back(make_pair(v, w));
adj[v].push_back(make_pair(u, w));
}
dijkstra(s, n); // 求解起点s到所有点的最短路径
cout << d[t] << endl; // 输出起点s到终点t的最短路径长度
return 0;
}
```
上面的代码使用了Dijkstra算法来求解最短路径,其时间复杂度为O(nlogn),可以通过本题。如果使用Bellman-Ford算法来求解,则时间复杂度为O(nm),更容易超时,但思路类似。
阅读全文