poj2449/acwing178 第k短路

A*的算法核心：选择一个合适的估计函数f(x)，且必须估计值小于等于实际值（否则是错的）实际值表示为g(x),则f(x)<=g(x)

本题因为是一张图 所以想到每个点的估计函数作为该点到终点的最短路 。这样符合要求，且能顺应g(x)的实际变化趋势

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>

using namespace std;
int n,m;
int st,ed,k;
int h[2][1005],tot[2];
int dis[1005];
struct edge
{
	int to,nxt,v;
}e[2][100005];
void add_edge(int x,int y,int v,int p)
{
	e[p][++tot[p]].to=y;
	e[p][tot[p]].nxt=h[p][x];
	e[p][tot[p]].v=v;
	h[p][x]=tot[p];
}
priority_queue<pair<int,int> >pq;
bool vis[1005];
void dijkstra()
{
	while(!pq.empty())pq.pop();
	memset(dis,0x7f,sizeof(dis));
	dis[ed]=0;
	pq.push(make_pair(0,ed));
	while(!pq.empty())
	{
		int x=pq.top().second;
		pq.pop();
		if(vis[x])continue;
		vis[x]=true;
		for(int i=h[0][x];i;i=e[0][i].nxt)
		{
			if(dis[x]+e[0][i].v<dis[e[0][i].to])
			{
				dis[e[0][i].to]=dis[x]+e[0][i].v;
				pq.push(make_pair(-dis[e[0][i].to],e[0][i].to));
			}
		}
	}
}

priority_queue<pair<int,int> >q;

int cnt[1005];
int bfs()
{
	q.push(make_pair(-0-dis[st],st));
	while(!q.empty())
	{
		int x=q.top().second,y=-q.top().first-dis[x];
		q.pop();
		++cnt[x];
		if(cnt[x]>k)continue;
		if(x==ed&&cnt[x]==k)return y;
		for(int i=h[1][x];i;i=e[1][i].nxt)
		{
			q.push(make_pair(-y-e[1][i].v-dis[e[1][i].to],e[1][i].to));
		}
	}
	return -1;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;++i)
	{
		int x,y,v;
		scanf("%d%d%d",&x,&y,&v);
		add_edge(y,x,v,0);
	}
	scanf("%d%d%d",&st,&ed,&k);
	if(st==ed)++k;
	dijkstra();
//	for(int i=1;i<=n;++i)printf("%d\n",dis[i]);
//	return 0;
	for(int i=1;i<=n;++i)
	{
		for(int j=h[0][i];j;j=e[0][j].nxt)
		{
			add_edge(e[0][j].to,i,e[0][j].v,1);
		}
	}
	
	printf("%d\n",bfs());
	return 0;
}