题目:$\textsf{\textcolor{#52C41A}{P4568 [JLOI2011] 飞行路线}}$。
提交记录:
-
错误记录:R237826896,55pts。
-
正确记录:R237829107,100pts。
错误原因:城市范围 $0 \sim n - 1$,代码中写 $1 \sim n$。
#include <bits/stdc++.h>
using namespace std;
struct node
{
int id, dis, cnt;
bool operator<(const node &x) const
{
return dis > x.dis;
}
};
struct edge
{
int to, w;
};
const int LEN = 1e4 + 5;
int dist[LEN][11], n, m, k, s, t, ans = INT_MAX;
bool vis[LEN][11];
vector<edge> G[LEN];
priority_queue<node> q;
void dij(int s)
{
// for (int i = 1; i <= n; i++) <----------- !!!!
for (int i = 0; i < n; i++)
for (int j = 0; j <= k; j++)
dist[i][j] = INT_MAX;
dist[s][0] = 0;
q.push({s, 0, 0});
while (!q.empty())
{
node u = q.top();
q.pop();
// printf("q.pop: u.id = %d, u.dis = %d,u.cnt = %d\n", u.id, u.dis, u.cnt);
if (vis[u.id][u.cnt])
{
// printf("continue: u.id = %d, u.cnt = %d\n", u.id, u.cnt);
continue;
}
vis[u.id][u.cnt] = 1;
for (int i = 0; i < G[u.id].size(); i++)
{
edge e = G[u.id][i];
if (dist[u.id][u.cnt] + e.w < dist[e.to][u.cnt])
{
dist[e.to][u.cnt] = dist[u.id][u.cnt] + e.w;
q.push({e.to, dist[e.to][u.cnt], u.cnt});
// printf("q.push: e.to = %d, dist[e.to][u.cnt] = %d, u.cnt = %d\n", e.to, dist[e.to][u.cnt], u.cnt);
}
if (u.cnt < k && dist[u.id][u.cnt] < dist[e.to][u.cnt + 1])
{
dist[e.to][u.cnt + 1] = dist[u.id][u.cnt];
q.push({e.to, dist[e.to][u.cnt + 1], u.cnt + 1});
// printf("q.push: e.to = %d, dist[e.to][u.cnt + 1] = %d, u.cnt + 1 = %d\n", e.to, dist[e.to][u.cnt + 1], u.cnt + 1);
}
}
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin >> n >> m >> k >> s >> t;
for (int i = 1; i <= m; i++)
{
int a, b, c;
cin >> a >> b >> c;
G[a].push_back({b, c});
G[b].push_back({a, c});
}
dij(s);
for (int i = 0; i <= k; i++)
ans = min(ans, dist[t][i]);
cout << ans;
// for (int i = 0; i <= k; i++)
// {
// for (int j = 0; j < n; j++)
// cout << dist[j][i] << " ";
// cout << "\n";
// }
return 0;
}