BZOJ 1975 [Sdoi2010]魔法猪学院 - k短路(手写堆)

大概就是一道k短路的裸题，然后恶心得要手写堆，学习了一下写法，发现好简单233

注意就是当搜索到终点时不可continue，终点存在自环，可以从终点出发再次搜到终点。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<queue>
#include<algorithm>
#define INF 1e20

using namespace std;

const int maxn=5005;
const int maxm=200005;

int n,m,cnt,fcnt,ans,top;
int head[maxn],fhead[maxn];
bool in[maxn];
double sum,dist[maxn];

struct edge
{
int to,next;
double val;
}e[maxm],fe[maxm];

struct node
{
int to;
double val;
bool operator < (const node &a)const
{
return dist[to]+val<dist[a.to]+a.val;
}
bool operator > (const node &a)const
{
return dist[to]+val>dist[a.to]+a.val;
}
}q[maxm<<3];

void insert(int a,int b,double val)
{
e[++cnt].to=b;e[cnt].next=head[a];e[cnt].val=val;head[a]=cnt;
}
void finsert(int a,int b,double val)
{
fe[++fcnt].to=b;fe[fcnt].next=fhead[a];fe[fcnt].val=val;fhead[a]=fcnt;
}
void push(node x)
{
q[++top]=x;
int t=top;
while(t>1)
{
if(q[t]<q[t>>1])
{
swap(q[t],q[t>>1]);
t>>=1;
}
else break;
}
}
void pop()
{
q[1]=q[top--];
int x=2;
while(x<=top)
{
if(q[x]>q[x+1]&&x+1<=top)x++;
if(q[x]<q[x>>1])
{
swap(q[x],q[x>>1]);
x<<=1;
}
else break;
}
}
node front()
{
return q[1];
}
void spfa()
{
for(int i=1;i<=n;i++)dist[i]=INF;
dist[1]=0;
in[1]=true;
queue<int>qu;
qu.push(1);
while(!qu.empty())
{
int u=qu.front();
qu.pop();
in[u]=false;
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].to;
if(dist[v]>dist[u]+e[i].val)
{
dist[v]=dist[u]+e[i].val;
if(!in[v])qu.push(v),in[v]=true;
}
}
}
}
void kth_path()
{
push((node){n,0});
while(top)
{
int u=front().to;
double now=front().val;
pop();
if(u==1)
{
if(sum<now)break;
sum-=now;
ans++;
//continue;
}
for(int i=fhead[u];i;i=fe[i].next)
{
int v=fe[i].to;
push((node){v,now+fe[i].val});
}
}

}
int main()
{
scanf("%d%d%lf",&n,&m,&sum);
for(int i=1;i<=m;i++)
{
int a,b;double val;
scanf("%d%d%lf",&a,&b,&val);
insert(a,b,val);
finsert(b,a,val);
}
spfa();
kth_path();
printf("%d",ans);
}