BZOJ 1598 [Usaco2008 Mar]牛跑步 A*

Description

BESSIE准备用从牛棚跑到池塘的方法来锻炼. 但是因为她懒,她只准备沿着下坡的路跑到池塘, 然后走回牛棚. BESSIE也不想跑得太远,所以她想走最短的路经. 农场上一共有M (1 <= M <= 10,000)条路, 每条路连接两个用1..N(1 <= N <= 1000)标号的地点. 更方便的是,如果X>Y,则地点X的高度大于地点Y的高度. 地点N是BESSIE的牛棚;地点1是池塘. 很快, BESSIE厌倦了一直走同一条路.所以她想走不同的路,更明确地讲,她想找出K (1 <= K <= 100)条不同的路经.为了避免过度劳累,她想使这K条路经为最短的K条路经. 请帮助BESSIE找出这K条最短路经的长度.你的程序需要读入农场的地图, 一些从X_i到Y_i 的路经和它们的长度(X_i, Y_i, D_i). 所有(X_i, Y_i, D_i)满足(1 <= Y_i < X_i; Y_i < X_i <= N, 1 <= D_i <= 1,000,000).

Input

* 第1行: 3个数: N, M, 和K

* 第 2..M+1行: 第 i+1 行包含3个数 X_i, Y_i, 和 D_i, 表示一条下坡的路.

Output

* 第1..K行: 第i行包含第i最短路经的长度,或-1如果这样的路经不存在.如果多条路经有同样的长度,请注意将这些长度逐一列出.

Sample Input

5 8 7
5 4 1
5 3 1
5 2 1
5 1 1
4 3 4
3 1 1
3 2 1
2 1 1

Sample Output

1
2
2
3
6
7
-1

HINT

输出解释:

路经分别为(5-1), (5-3-1), (5-2-1), (5-3-2-1), (5-4-3-1),

(5-4-3-2-1).

传送门

K短路裸题……

估价函数是当前搜索到的距离+到终点的最短距离。

先反向边跑最短路，后面A*搜索就好了= =

1A，看上去没什么坑。。

#include<bits/stdc++.h>
using namespace std;
const int 
	N=1005,
	M=10005;
int n,m,K,Ecnt,Ecnto,dist[N];
bool vis[N];
struct Edge{
	int next,to,val;
}E[M],Eopp[M];int head[N],headopp[N];
void add1(int u,int v,int w){
	E[++Ecnt].next=head[u];
	E[Ecnt].to=v;
	E[Ecnt].val=w;
	head[u]=Ecnt;
}
void add2(int u,int v,int w){
	Eopp[++Ecnto].next=headopp[u];
	Eopp[Ecnto].to=v;
	Eopp[Ecnto].val=w;
	headopp[u]=Ecnto;
}

struct Heap{int num,val;};
priority_queue<Heap>Q;
bool operator <(Heap x,Heap y){return x.val>y.val;}
bool operator >(Heap x,Heap y){return x.val<y.val;}

void dijkstra(int s,int t){
	while (!Q.empty()) Q.pop();
	memset(vis,0,sizeof(vis));
	memset(dist,100,sizeof(dist));
	dist[s]=0;
	Q.push((Heap){s,0});
	while ((!Q.empty())){
		Heap mi=Q.top();Q.pop();
		if (vis[mi.num]) continue;
		vis[mi.num]=1;
		for (int j=headopp[mi.num];j;j=Eopp[j].next){
			int q=Eopp[j].to;
			if (vis[q]) continue;
			if (Eopp[j].val+mi.val<dist[q]){
				dist[q]=Eopp[j].val+mi.val;
				Q.push((Heap){q,dist[q]});
			}
		}
	}
}

struct node{int p,h;};
bool operator <(node a,node b){return a.h+dist[a.p]>b.h+dist[b.p];}
bool operator >(node a,node b){return a.h+dist[a.p]<b.h+dist[b.p];}
priority_queue<node>A_Q;
void A_star(int s,int t){
	while (!A_Q.empty()) A_Q.pop();
	A_Q.push((node){s,0});
	while (!A_Q.empty()){
		node u=A_Q.top(),t1;A_Q.pop();
		if (u.p==t){
			printf("%d\n",u.h+dist[u.p]);
			if (!(--K)) return;
			continue;
		}
		for (int i=head[u.p];i;i=E[i].next){
            t1.p=E[i].to;
            t1.h=E[i].val+u.h;
            A_Q.push(t1);
		}
	}
}
int main(){
	scanf("%d%d%d",&n,&m,&K);
	int x,y,z;
	for (int i=1;i<=m;i++){
		scanf("%d%d%d",&x,&y,&z);
		if (x<y) swap(x,y);
		add1(x,y,z),add2(y,x,z);
	}
	dijkstra(1,n);
	A_star(n,1);
	while (K--) printf("-1\n");
	return 0;
}