圖的第k短路

【問題描述】

給你一個有向圖，求從1到n的第k短路。

【解法】

SPFA+A*搜索。

1 A*算法

A*算法在人工智能中是一種典型的啟發式搜索算法，啟發中的估價是用估價函數表示的：

h(n)=f(n)+g(n)

其中f(n)是節點n的估價函數，g(n)表示實際狀態空間中從初始節點到n節點的實際代價，h(n)是從n到目標節點最佳路徑的估計代價。另外定義h'(n)為n到目標節點最佳路徑的實際值。如果h'(n)≥h(n)則如果存在從初始狀態走到目標狀態的最小代價的解，那么用該估價函數搜索的算法就叫A*算法。

2 第K最短路的算法

我們設源點為s，終點為t，我們設狀態f(i)的g(i)為從s走到節點i的實際距離，h(i)為從節點i到t的最短距離，從而滿足A*算法的要求，當第K次走到f(n-1)時表示此時的g(n-1)為第K最短路長度。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<queue>
#define MAXN 20100
#define INF 1000100000
#define pii pair<int,int> 
using namespace std;
struct node{int y,next,v;}e1[MAXN],e2[MAXN];
struct node2
{
    int f,g,id;    
    friend bool operator < (node2 a,node2 b)
    {
        return a.f>b.f;
    }
};
int n,m,k,len1,len2,Link1[MAXN],Link2[MAXN],dis[MAXN],vis[MAXN],tim[MAXN],ans[MAXN];
priority_queue<node2>Q;
inline int read()
{
    int x=0,f=1;  char ch=getchar();
    while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getchar();}
    while(isdigit(ch))  {x=x*10+ch-'0';  ch=getchar();}
    return x*f;
}
void insert(int xx,int yy,int vv)
{
    e1[++len1].next=Link1[xx];
    Link1[xx]=len1;
    e1[len1].y=yy;
    e1[len1].v=vv;              
    e2[++len2].next=Link2[yy];
    Link2[yy]=len2;
    e2[len2].y=xx;
    e2[len2].v=vv;              
}
void SPFA()
{
    priority_queue<int,vector< int >,greater< int > >q;
    memset(dis,50,sizeof(dis));
    dis[1]=0,vis[1]=1,q.push(1);
    while(!q.empty())
    {
        int tn=q.top();
        vis[tn]=0;
        q.pop();
        for(int i=Link2[tn];i;i=e2[i].next)
        {
            int tmp=e2[i].y;
            if(dis[tmp]>dis[tn]+e2[i].v)
            {
                dis[tmp]=dis[tn]+e2[i].v;
                if(!vis[tmp])
                {
                    vis[tmp]=1;
                    q.push(tmp);
                }
            }
        }
    }
}
void Astar()
{
    if(dis[n]==INF)  return;
    node2 temp;  temp.g=0,temp.f=dis[n],temp.id=n;
    Q.push(temp);
    while(!Q.empty())
    {
        temp=Q.top(); Q.pop();
        if(temp.id==1)  {ans[++ans[0]]=temp.f;  if(ans[0]>k)  return;}  
        for(int i=Link1[temp.id];i;i=e1[i].next)
        {
            node2 opt;
            opt.g=temp.g+e1[i].v;
            opt.f=opt.g+dis[e1[i].y];
            opt.id=e1[i].y;
            Q.push(opt);
        }
    }
}
int main()
{
    n=read();  m=read();  k=read();
    for(int i=1;i<=m;i++)
    {
        int x=read(),y=read(),v=read();
        insert(x,y,v);
    }
    SPFA();
    Astar();
    for(int i=1;i<=k;i++)
    {
        if(!ans[i])  printf("-1\n");
        else printf("%d\n",ans[i]);
    }
    return 0;
}