A*模板

求第k短路。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int INF = 1e9+7;
int n,m,ts,te,k,tot1,tot2,ru,rv,rw;
int first1[100010],nxt1[100010],first2[100010],nxt2[100010],dis[100010];
bool used[100010];
struct edge
{
    int u,v,w;
}l1[100010],l2[100010];
struct st
{
    int num,d;
    bool operator < (st a)const
    {
        return d>a.d;
    }
}p1[100010];
priority_queue<st>q2;
struct heap
{
    int g,h,num;
    bool operator < (heap a)const
    {
        return (g+h)>(a.g+a.h);
    }
}p2[100010];
priority_queue<heap>q1;
void build_1(int f,int t,int c)
{
    l1[++tot1]=(edge){f,t,c};
    nxt1[tot1]=first1[f];
    first1[f]=tot1;
}
void build_2(int f,int t,int c)
{
    l2[++tot2]=(edge){f,t,c};
    nxt2[tot2]=first2[f];
    first2[f]=tot2;
}
void geth()
{
    for(int i=1;i<=n;i++)
    dis[i]=INF;
    dis[te]=0;
    q2.push((st){te,0});
    while(!q2.empty())
    {
        st h=q2.top();
        q2.pop();
        if(used[h.num]==1)
        continue;
        used[h.num]=1;
        for(int i=first2[h.num];i!=-1;i=nxt2[i])
        {
            int x=l2[i].v;
            if(dis[x]>dis[h.num]+l2[i].w)
            {
                dis[x]=dis[h.num]+l2[i].w;
                q2.push((st){x,dis[x]});
            }
        }
    }
}
void Astar()
{
    geth();
    q1.push((heap){0,dis[ts],ts});
    while(!q1.empty())
    {
        heap h=q1.top();
        q1.pop();
        if(h.num==te)
        {
            tot1++;
            if(tot1>=k)
            {
                printf("%d",h.g);
                return;
            }
        }
        for(int i=first1[h.num];i!=-1;i=nxt1[i])
        {
            int x=l1[i].v;
            q1.push((heap){h.g+l1[i].w,dis[x],x});
        }
    }
    printf("-1");
}
int main()
{
    memset(first2,-1,sizeof(first2));
    memset(first1,-1,sizeof(first1));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&ru,&rv,&rw);
        build_1(ru,rv,rw);
        build_2(rv,ru,rw);
    }
    scanf("%d%d%d",&ts,&te,&k);
    if(ts==te)
    k++;
    tot1=0;
    Astar();
    return 0;
}

注意估价函数设置，对于求第k短路而言，估价函数的影响使得Dijkstra过程中只会依次严格经过第1,2…k短路上的边到达终点。

求第k短路也可以使用Dijkstra，第k次找到终点的路径即为由起点到终点的第k短路。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
int n,m,ts,te,k,ru,rv,rw,tot;
int first[100010],nxt[100010],cnt[100010];
bool flag;
ll ans;
struct edge
{
    int u,v,w;
}l[100010];
struct st
{
    int num;
    ll d;
}p;
bool operator < (st a,st b)
{
    return  a.d>b.d;
}
priority_queue<st>q;
void build(int f,int t,int c)
{
    l[++tot]=(edge){f,t,c};
    nxt[tot]=first[f];
    first[f]=tot;
}
void Dijkstra()
{
    q.push((st){ts,0});
    while(!q.empty())
    {
        st h=q.top();
        q.pop();
        if(cnt[h.num]>=k)//十分重要的剪枝
        continue;
        cnt[h.num]++;
        if(h.num==te&&cnt[h.num]==k)
        {
            flag=1;
            ans=h.d;
            break;
        }
        for(int i=first[h.num];i!=-1;i=nxt[i])
        {
            int x=l[i].v;
            q.push((st){x,h.d+l[i].w});
        }
    }
}
int main()
{
    memset(first,-1,sizeof(first));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&ru,&rv,&rw);
        build(ru,rv,rw);
    }
    scanf("%d%d%d",&ts,&te,&k);
    tot=0;
    Dijkstra();
    if(flag)
    printf("%I64d",ans);
    else printf("-1");
    return 0;
}