POJ 2449 A*k短路

题意：给出一个有向图，起点终点以及k，求k短路

思路：使用A*算法，首先预处理出每个点到终点的距离h[i]，搜索时使用优先队列，关键字为（f + h[i]），其中f 是已经走的距离，i是当前走到的节点

注意：此题坑点巨多：

1）起点可能与终点相同，此时0不算最短路，故k++

2）k短路不存在时，输出-1

3）由于是有向图，从Dijkstra时只能用反边，A*时只能用正边，否则MLE

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
using namespace std;
const int maxn=1010;
const int INF=1e9;
vector<int>g[maxn],w[maxn];
vector<bool>P[maxn];
struct node{
int id,w;
bool operator <(const node &A)const{
return w>A.w;
}
};
int vis[maxn],dis[maxn];
int n,m;
void add(int x,int y,int z,int p){
g[x].push_back(y);
w[x].push_back(z);
P[x].push_back(p);
}
void dijkstra(int st){
for(int i=1;i<=n;i++) vis[i]=0,dis[i]=INF;
priority_queue<node>q;
dis[st]=0;
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(int i=0;i<g[s].size();i++){
int v=g[s][i];
if(!P[s][i] && dis[v]>G+w[s][i]){
dis[v]=G+w[s][i];
q.push((node){v,dis[v]});
}
}
}
}
struct data{
int pos,d;
bool operator <(const data &A)const{
return d+dis[pos]>A.d+dis[A.pos];
}
};
int A_star(int k,int st,int ed){
memset(vis,0,sizeof(vis));
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.d;
if(vis[u]>k) continue;
for(int i=0;i<g[u].size();i++){
int v=g[u][i];
if(P[u][i]) q.push((data){v,s.d+w[u][i]});
}
}
return -1;
}
int main(){
int st,ed,k;
int i,j;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z,1);
add(y,x,z,0);
}
scanf("%d%d%d",&st,&ed,&k);
if(st==ed) k++;
dijkstra(ed);
printf("%d\n",A_star(k,st,ed));
return 0;
}

^_^

by：dds

http://blog.csdn.net/dy0607/article/details/52423809