【BZOJ】1975 [Sdoi2010]魔法猪学院 k短路（最短路径+A*）

题目传送门

这题的正解——k短路还是挺好想到的，只不过题目告诉你的是路径长度总和，叫你求k的值。其他的和k短路的思想一模一样。

不过据说这题卡stl中的priority_queue，那么把priority_queue换成push_heap和pop_heap函数就好了。

关于push_heap等函数的介绍，看这里就行了，该博主讲的已经很详细了。

附上AC代码：

#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;

struct note{
	int d;
	double w;
};
vector <note> z[5010],f[5010];
struct node{
	int d;
	double g,f;
	bool operator < (const node lyf) const {
		return f==lyf.f?g>lyf.g:f>lyf.f;
	}
}h[2000010];
queue <int> que;
double sum,w,dis[5010];
int n,m,x,y,len;
bool b[5010];

void spfa(){
	for (int i=1; i<=n; ++i) dis[i]=2e9;
	que.push(n),dis[n]=0,b[n]=1;
	while (!que.empty()){
		int p=que.front();que.pop(),b[p]=0;
		for (int i=0; i<f[p].size(); ++i){
			note q=f[p][i];
			if (dis[q.d]>dis[p]+q.w){
				dis[q.d]=dis[p]+q.w;
				if (!b[q.d]){
					b[q.d]=1;
					que.push(q.d);
				}
			}
		}
	}
	return;
}

int astar(){
	if (dis[1]==2e9) return -1;
	int c=0;
	h[len=1]=(node){1,0,dis[1]};
	while (len){
		pop_heap(h+1,h+1+len);
		node p=h[len--];
		if (p.d==n) {
			sum-=p.g,++c;
			if (sum<=0) return c-1;
		}
		for (int i=0; i<z[p.d].size(); ++i){
			note q=z[p.d][i];
			h[++len]=(node){q.d,p.g+q.w,p.g+q.w+dis[q.d]};
			push_heap(h+1,h+1+len);
		}
	}
	return -1;
}

int main(void){
	scanf("%d%d%lf",&n,&m,&sum);
	for (int i=1; i<=m; ++i){
		scanf("%d%d%lf",&x,&y,&w);
		z[x].push_back((note){y,w});
		f[y].push_back((note){x,w});
	}
	spfa();
	printf("%d",astar());
	return 0;
}