题意:给出一个有向图,起点终点以及k,求k短路
思路:使用A*算法,首先预处理出每个点到终点的距离h[i],搜索时使用优先队列,关键字为(f + h[i]),其中f 是已经走的距离,i是当前走到的节点
注意:此题坑点巨多:
1)起点可能与终点相同,此时0不算最短路,故k++
2)k短路不存在时,输出-1
3)由于是有向图,从Dijkstra时只能用反边,A*时只能用正边,否则MLE
代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
#define For(i,j,k) for(int i = j;i <= k;i ++)
#define Set(A, val) memset(A, val, sizeof(A))
const int N = 1010;
using namespace std;
struct Node{
int id, w;
bool operator < (const Node& a)const{
return w > a.w;
}
};
int d[N], vis[N], n, m;
vector<int> G[N], w[N];
vector<bool> P[N];
void Add(int x, int y, int c, int p){
G[x].push_back(y);
w[x].push_back(c);
P[x].push_back(p);
}
void Dijkstra(int st){
For(i,1,n) vis[i] = 0, d[i] = 1e9;
d[st] = 0;
priority_queue<Node> q;
q.push((Node){st, 0});
while(!q.empty()){
int s = q.top().id, g = q.top().w;
q.pop();
if(vis[s])continue;
vis[s] = 1;
For(i,0,(int)G[s].size() - 1){
int v = G[s][i];
if(!P[s][i] && d[v] > g + w[s][i])
d[v] = g + w[s][i], q.push((Node){v, d[v]});
}
}
}
struct Data{
int pos, dis;
bool operator < (const Data &a) const{
return dis + d[pos] > a.dis + d[a.pos];
}
};
int A_Star(int k, int st, int ed){
Set(vis, 0);
priority_queue<Data> q;
q.push((Data){st, 0});
while(!q.empty()){
Data s = q.top();q.pop();
int u = s.pos;
vis[u]++;
if(vis[ed] == k) return s.dis;
if(vis[u] > k) continue;
For(i,0,(int)G[u].size() - 1){
int v = G[u][i];
if(P[u][i])
q.push((Data){v, s.dis + w[u][i]});
}
}
return -1;
}
int main(){
int st, ed, k;
scanf("%d%d", &n, &m);
For(i,1,m){
int x, y, c;
scanf("%d%d%d", &x, &y, &c);
Add(x, y, c, 1);
Add(y, x, c, 0);
}
scanf("%d%d%d", &st, &ed, &k);
if(st == ed) ++k;
Dijkstra(ed);
printf("%d\n", A_Star(k, st, ed));
return 0;
}