[K短路]P2901

PS：本题只是K短路的部分正解，不能保证时间复杂度足够优秀（即，过不了 P2483的hack数据）

给出有向图，求前 $k$ 短路长度，若不满 $k$ 条，则输出-1。

---

以本题为例写一下用 $A^{\ast }$ 算法解决K短路问题的思路。

首先对原图的反图进行一次求解最短路（$SPFA$ yyds），求出数组 $di{s}_i$ 表示从 $i$ 点走向终点的最短路径。

然后执行 $A^{\ast }$ 算法，以实际路径长 $le{n}_i$ 加上估价函数 $di{s}_i$ 进行小根堆排序。可以保证第 $i$ 次取出走到终点的即为第 $K$ 短路。

#include<bits/stdc++.h>
using namespace std;
#define N 1005
#define M 10005
#define mp make_pair
struct edge{
	int to,nxt,w;
}e[M],ef[M];
struct node{
	int pos,len;
};
int head[N],headf[N],ecnt,dis[N],bj[N];
bool operator <(node a,node b){
	return a.len+dis[a.pos]>b.len+dis[b.pos];
}

void adde(int u,int v,int w){
	e[++ecnt].nxt=head[u];
	ef[ecnt].nxt=headf[v];
	head[u]=ecnt;
	headf[v]=ecnt;
	e[ecnt].to=v;
	ef[ecnt].to=u;
	e[ecnt].w=w;
	ef[ecnt].w=w;
}
int n,m,k;
queue<int> Q;
priority_queue< node > q;
void SPFA(){
	dis[1]=0;
	Q.push(1);
	while(!Q.empty()){
		int u=Q.front();
		Q.pop();
		bj[u]=0;
		for(int i=headf[u];i;i=ef[i].nxt){
			int v=ef[i].to;
			if(dis[u]+ef[i].w<dis[v]){
				dis[v]=dis[u]+ef[i].w;
				if(bj[v]==0){
					bj[v]=1;
					Q.push(v);
				}
			}
		}
	}
}

void Astar(){
	q.push((node){n,0});
	while(!q.empty()){
		int pos=q.top().pos;
		int len=q.top().len;
		q.pop();
		if(pos==1){
			printf("%d\n",len);
			--k;
			if(k==0)
				return;
		}
		for(int i=head[pos];i;i=e[i].nxt){
			int v=e[i].to;
			q.push((node){v,len+e[i].w});
		}
	}
	return;
}
int main(){
	scanf("%d %d %d",&n,&m,&k);
	for(int i=1;i<=m;++i){
		int x,y,d;
		scanf("%d %d %d",&x,&y,&d);
		adde(x,y,d);
	}
	memset(dis,0x3f,sizeof(dis));
	SPFA();
	Astar();
	while(k--){
		printf("-1\n");
	}
	return 0;
}

顺便放出P2483的非正解

#include<bits/stdc++.h>
using namespace std;
#define N 5005
#define M 200005
struct edge{
	int to,nxt;
	double w;
}e[M],ef[M];
struct node{
	int pos;
	double len;
};
int head[N],headf[N],ecnt,bj[N];
double dis[N];
bool operator <(node a,node b){
	return a.len+dis[a.pos]>b.len+dis[b.pos];
}

void adde(int u,int v,double w){
	e[++ecnt].nxt=head[u];
	ef[ecnt].nxt=headf[v];
	head[u]=ecnt;
	headf[v]=ecnt;
	e[ecnt].to=v;
	ef[ecnt].to=u;
	e[ecnt].w=w;
	ef[ecnt].w=w;
}
int n,m,ans;
double k;
queue<int> Q;
priority_queue< node > q;
void SPFA(){
	dis[n]=0.0;
	Q.push(n);
	while(!Q.empty()){
		int u=Q.front();
		Q.pop();
		bj[u]=0;
		for(int i=headf[u];i;i=ef[i].nxt){
			int v=ef[i].to;
			if(dis[u]+ef[i].w<dis[v]){
				dis[v]=dis[u]+ef[i].w;
				if(bj[v]==0){
					bj[v]=1;
					Q.push(v);
				}
			}
		}
	}
}

void Astar(){
	q.push((node){1,0});
	while(!q.empty()){
		int pos=q.top().pos;
		double len=q.top().len;
		q.pop();
		if(pos==n){
			if(k>=len){
				k-=len;
				ans++;
			}
			else{
				return;
			}
		}
		for(int i=head[pos];i;i=e[i].nxt){
			int v=e[i].to;
			q.push((node){v,len+e[i].w});
		}
	}
	return;
}
int main(){
	scanf("%d %d %lf",&n,&m,&k);
	for(int i=1;i<=m;++i){
		int x,y;
		double d;
		scanf("%d %d %lf",&x,&y,&d);
		adde(x,y,d);
	}
	for(int i=1;i<=n;++i){
		dis[i]=10000000.0;
	}
	SPFA();
	Astar();
	printf("%d",ans);
	return 0;
}