bzoj 1598: [Usaco2008 Mar]牛跑步 （k短路）

1598: [Usaco2008 Mar]牛跑步

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 503   Solved: 293
[ Submit][ Status][ Discuss]

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).

Source

Gold

[ Submit][ Status][ Discuss] 

题解：k短路

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#define N 10003
#define LL long long
using namespace std;
int point[N],v[N],nxt[N],n,m,k;
int point1[N],v1[N],nxt1[N],tot,tot1,can[N],cnt[N];
LL len[N],len1[N],dis[N],ans[N];
const LL inf=1e13;
struct data{
	LL g,h; int x;
	bool operator <(data a) const{
     return a.g+a.h<g+h;
	}
};
void add(int x,int y,LL z)
{
	tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; len[tot]=z;
	tot1++; nxt1[tot1]=point1[y]; point1[y]=tot1; v1[tot1]=x; len1[tot]=z;
}
void spfa()
{
	queue<int> p; 
	for (int i=1;i<=n;i++) dis[i]=inf;
	dis[1]=0; can[1]=1; p.push(1);
	while (!p.empty()) {
		int now=p.front(); p.pop();
		for (int i=point1[now];i;i=nxt1[i])
		 if (dis[v1[i]]>dis[now]+len1[i]) {
		 	dis[v1[i]]=dis[now]+len1[i];
		 	if (!can[v1[i]]) {
		 		can[v1[i]]=1;
		 		p.push(v1[i]);
			 }
		 }
		can[now]=0;
	}
}
void astar()
{
	priority_queue<data> p;
	data cur,nt;
	cur.x=n; cur.g=0; cur.h=dis[n]; p.push(cur);
	int size=0;
	while (!p.empty()) {
		cur=p.top(); p.pop();
		cnt[cur.x]++;
		if (cnt[cur.x]>k) continue;
		if (cur.x==1) ans[++size]=cur.g;
		if (cnt[1]==k) return;
		for (int i=point[cur.x];i;i=nxt[i]){
			nt.x=v[i]; 
			nt.g=cur.g+len[i]; nt.h=dis[v[i]];
			p.push(nt);
		}
	}
}
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d%d%d",&n,&m,&k);
	for (int i=1;i<=m;i++) {
		int x,y,z; scanf("%d%d%d",&x,&y,&z);
		if (x<y) swap(x,y);
		add(x,y,(LL)z);
	}
	spfa();
	memset(ans,-1,sizeof(ans));
	astar(); 
	for (int i=1;i<=k;i++) printf("%d\n",ans[i]);
}