第k短路【bfs】【a_star算法】

题目传送门

暴力朴素做法：暴力宽搜，每次取队头，把每次出边加入队，每个点第几次出队就是第几短路

用a_star算法优化：

#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<string.h>
using namespace std;
const int N=2e5+10;
typedef pair<int,int>pll;
typedef pair<int,pll>plll;
int h[N],hr[N],w[N],e[N],ne[N],idx;
int dis[N],f[N],cnt[N];
int s,t,k;
int n,m;
bool st[N];
void add(int *h,int a,int b,int c)
{
    e[idx]=b;
    w[idx]=c;
    ne[idx]=h[a];
    h[a]=idx++;
}

void dijistra()
{
    priority_queue<pll,vector<pll>,greater<pll> >q;
    for(int i=1;i<=n;i++) dis[i]=0x3f3f3f3f;
    dis[t]=0;
    q.push(make_pair(0,t));
    while(q.size())
    {
        auto t=q.top();
        q.pop();
        int ver=t.second;
        int distance = t.first;
        if(st[ver]) continue;
        st[ver]=true;
        for(int i=hr[ver];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>dis[ver]+w[i]){
                dis[j]=dis[ver]+w[i];
                q.push(make_pair(dis[j],j));
            }
        }
    }
    memcpy(f,dis,sizeof dis);
}

//求第k短路，玩命扩展，迪杰斯特拉求最短路，就扩展一次，需要标记；
int a_star()
{
    priority_queue<plll,vector<plll>,greater<plll> >q;
    q.push(make_pair(f[s],make_pair(0,s)));
    while(q.size())
    {
        auto tt=q.top();
        q.pop();
        int ver=tt.second.second,distance=tt.second.first;
        if(cnt[ver]>=k) continue;//中途点超过k+1,没必要扩展了
        cnt[ver]++;
        //到终点；
        if(ver==t &&cnt[ver]==k) return distance;//正在试图扩展k+1，等到答案
        for(int i=h[ver];~i;i=ne[i])
        {
            int j=e[i];
            if(cnt[j]<k)  // 没有扩展k+1次
                q.push(make_pair(f[j]+w[i]+distance,make_pair(distance+w[i],j)));
        }
    }
    return -1;
}

int main()
{
    memset(h,-1,sizeof h);
    memset(hr,-1,sizeof hr);
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b,l;
        cin>>a>>b>>l;
        add(h,a,b,l);
        add(hr,b,a,l);
    }
    cin>>s>>t>>k;
    if(s==t) k++;
    dijistra();
    cout<<a_star()<<endl;
    return 0;
}