【poj2449】 Remmarguts' Date

http://poj.org/problem?id=2449 (题目链接)

题意

　　求有向图K短路。

Solution

　　A*。g(x)为当前节点到起点的步数，h(x)为当前节点到终点的最短距离（也就是估价函数）。

细节

　　dijkstra求终点到各点最短路时要把边反向。原来起点和终点可以是同一个点，坑死了。。。

代码

// poj2499
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define LL long long
#define inf 2147483640
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;

const int maxn=100010;
struct edge {int to,next,w;}e[maxn<<1];
int head[maxn],cnts[maxn],dis[maxn],vis[maxn];
int n,m,S,T,K,cnt,U[maxn],V[maxn],W[maxn];

struct data {
	int num,w;
	friend bool operator < (const data a,const data b) {
		return a.w>b.w;
	}
};
struct node {
	int num,g,h;
	friend bool operator < (const node a,const node b) {
		return a.g+a.h>b.g+b.h;
	}
};
void link(int u,int v,int w) {
	e[++cnt]=(edge){v,head[u],w};head[u]=cnt;
}
void Dijkstra() {
	for (int i=1;i<=n;i++) dis[i]=inf;dis[T]=0;
	priority_queue<data> q;
	data x=(data){T,0};q.push(x);
	while (!q.empty()) {
		x=q.top();q.pop();
		if (vis[x.num]) continue;
		vis[x.num]=1;
		for (int i=head[x.num];i;i=e[i].next) if (dis[e[i].to]>x.w+e[i].w) {
				dis[e[i].to]=e[i].w+x.w;
				q.push((data){e[i].to,dis[e[i].to]});
			}
	}
}
int Astar() {
	node x=(node){S,0,dis[S]};
	priority_queue<node> q;q.push(x);
	while (!q.empty()) {
		x=q.top();q.pop();
		cnts[x.num]++;
		if (cnts[x.num]>K) continue;
		if (cnts[T]==K) return x.g;
		for (int i=head[x.num];i;i=e[i].next) q.push((node){e[i].to,x.g+e[i].w,dis[e[i].to]});
	}
	return -1;
}
int main() {
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;i++) {
		scanf("%d%d%d",&U[i],&V[i],&W[i]);
		link(V[i],U[i],W[i]);
	}
	scanf("%d%d%d",&S,&T,&K);
	if (S==T) K++;
	Dijkstra();
	memset(head,0,sizeof(head));cnt=0;
	for (int i=1;i<=m;i++) link(U[i],V[i],W[i]);
	int ans=Astar();
	printf("%d",ans);
	return 0;
}