BZOJ 1975 魔法猪学院(A*求K短路)
显然每次贪心的走最少消耗的路径即可。那么也就是找出最短路,次短路,,,K短路之后消耗E的能量的最多的路径条数。
也就是裸的A*算法。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include <bits/stdc++.h> using namespace std; typedef double lf; const int N=5005, M=200005; lf d[N], En; typedef pair<lf, int> pr; #define mkpr(x, y) make_pair<lf, int> (x, y) priority_queue<pr, vector<pr>, greater<pr> >q; struct Gr { struct E { int next, to; lf w; }e[M]; int n, ihead[N], cnt; void add(int x, int y, lf w) { e[++cnt]=(E){ihead[x], y, w}; ihead[x]=cnt; } void dij() { static bool vis[N]; memset(vis, 0, sizeof(bool)*(n+1)); for(int i=1; i<=n; ++i) { d[i]=1e150; } d[n]=0; q.push(mkpr((lf)0, n)); while(q.size()) { int x=q.top().second; q.pop(); if(vis[x]) { continue; } vis[x]=1; for(int i=ihead[x]; i; i=e[i].next) { int y=e[i].to; if(d[y]>d[x]+e[i].w) { d[y]=d[x]+e[i].w; q.push(mkpr(d[y], y)); } } } } int getans() { int ans=0; q.push(mkpr(d[1], 1)); while(En>0 && q.size()) { int x=q.top().second; lf g=q.top().first-d[x]; q.pop(); if(x==n) { if(En>=g) { En-=g; ++ans; continue; } else { break; } } for(int i=ihead[x]; i; i=e[i].next) { int y=e[i].to; lf w=e[i].w; // 有精度误差啊,不能乱减枝啊 q.push(mkpr(g+w+d[y], y)); } } return ans; } }g, G; int main() { int n, m; scanf("%d%d%lf", &n, &m, &En); g.n=G.n=n; for(int i=1; i<=m; ++i) { int x, y; lf e; scanf("%d%d%lf", &x, &y, &e); g.add(x, y, e); G.add(y, x, e); } G.dij(); printf("%d\n", g.getans()); return 0; }