BZOJ 1975: [Sdoi2010]魔法猪学院 大水题 第k短路 spfa

https://www.lydsy.com/JudgeOnline/problem.php?id=1975

我好像到现在了第k短路都不会写，mdzz。

先spfa求出最短路，然后扫点存各种前置路径已经决定的最短路，小根堆暴力即可。

有向图要存反向边，写完才发现的，临时添成两种了，丑也没办法

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define pa pair<double,int>
const int maxn=5010;
const double inf=1e17;
int n,m;double ene;
struct nod{
    int y,next;double v;
}e[2][maxn*40];
int head[2][maxn]={},tot[2]={};
double dis[maxn]={};
bool vis[maxn]={};
queue<int>q;
priority_queue< pa , vector< pa > , greater< pa > >q1;
void init(int x,int y,double v,int op){
    e[op][++tot[op]].v=v;e[op][tot[op]].y=y;e[op][tot[op]].next=head[op][x];head[op][x]=tot[op];
}
void spfa(){
    for(int i=1;i<=n;i++)dis[i]=inf;
    dis[n]=0;vis[n]=1;q.push(n);
    int x,y;double v;
    while(!q.empty()){
        x=q.front();q.pop();
        for(int i=head[0][x];i;i=e[0][i].next){
            y=e[0][i].y;v=e[0][i].v;
            //cout<<x<<y<<v<<dis[y]<<endl;
            if(dis[y]>dis[x]+v){
                dis[y]=dis[x]+v;
                if(!vis[y]){
                    vis[y]=1;
                    q.push(y);
                }
            }
        }vis[x]=0;
    }
}
int getit(){
    q1.push(make_pair(dis[1],1));
    int ans=0;double vq,v;int x;
    while(!q1.empty()){
        x=q1.top().second; v=q1.top().first; q1.pop();
        vq=v-dis[x];
        if(x==n){
            if(ene-v>=0){ene-=v;ans++;}
            else break;
        }
        for(int i=head[1][x];i;i=e[1][i].next){
            q1.push(make_pair(vq+e[1][i].v+dis[e[1][i].y],e[1][i].y));
        }
    }
    return ans;
}
int main(){
    scanf("%d%d%lf",&n,&m,&ene);
    int x,y;double v;
    for(int i=1;i<=m;i++){scanf("%d%d%lf",&x,&y,&v);init(x,y,v,1);init(y,x,v,0);}
    spfa();
    printf("%d\n",getit());
    return 0;
}