POJ 2449 Dijstra + A* K短路

最新推荐文章于 2018-08-03 16:09:43 发布

转载最新推荐文章于 2018-08-03 16:09:43 发布 · 555 阅读

ACM 同时被 3 个专栏收录

144 篇文章

订阅专栏

图论

57 篇文章

订阅专栏

搜索

8 篇文章

订阅专栏

本文介绍如何使用A*算法高效求解第K条最短路径问题。首先通过Dijkstra算法预计算出各节点到目标节点的最短距离作为启发函数h(x)，然后利用优先队列加速搜索过程，确保每次扩展的是当前最佳节点。

这题一开始的思路应该是直接从源点进行BFS搜索K短路。

但这样的复杂度在点数和K的值增大后将会变得很大。

而A*算法则构造一个h（x），在进行BFS时，每次都抛出最小的h(x)从而使汇点的出队速度加快。

这题则是先进行一次Dijstra求出每个点到汇点的最短路h(x)从而在优先队列中抛出最小的f（x）

A*算法就是启发式搜索，基本形式就是这样:f(x)=g(x)+h(x);其中f(x)代表在x点所需要的总代价，而g(x)代表：从源点到x点已经耗费的实际代价，h(x)代表从x到终点需要的估计代价，这个函数是一个估计值.而从x到终点真正需要的代价为h*(x)，在整个启发式搜索中我们必须保证h(x)<=h*(x);不然的话会由于对当前的估价值过高，则会引起答案的错误。构建A*的关键在于准确的规划一个h(x)函数，使得接近h*(x),这样的搜索会使得答案又快又准。可以想象h(x)过小会使得解空间过大，这样搜索出来的结果会很准确但是速度太慢，而对h(x)的过高估计，即估计代价太大会使得结果不准确。

代码：

#include<iostream>
#include<cstdio>
#include<queue>
#define MAXN 1005
#define MAXM 200100
using namespace std;

struct Node{
       int v,c,nxt;
}Edge[MAXM];

int head[MAXN];
int tail[MAXN];
int h[MAXN];

struct Statement
{
       int v,d,h;
       bool operator <( Statement a )const
       {    return a.d+a.h<d+h;   }
};

void addEdge( int u,int v,int c,int e )
{
     Edge[e<<1].v=v;
     Edge[e<<1].c=c;
     Edge[e<<1].nxt=head[u];
     head[u]=e<<1;
     
     Edge[e<<1|1].v=u;
     Edge[e<<1|1].c=c;
     Edge[e<<1|1].nxt=tail[v];
     tail[v]=e<<1|1;
     return ;
}

void Dijstra( int n,int s,int t )
{
     bool vis[MAXN];
     memset( vis,0,sizeof(vis) );
     memset( h,0x7F,sizeof(h) );
     h[t]=0;
     for( int i=1;i<=n;i++ )
     {
          int min=0x7FFF;
          int k=-1;
          for( int j=1;j<=n;j++ )
          {
               if( vis[j]==false && min>h[j] )
                   min=h[j],k=j;
          }
          if( k==-1 )break;
          vis[k]=true;
          for( int temp=tail[k];temp!=-1;temp=Edge[temp].nxt )
          {
               int v=Edge[temp].v;
               if( h[v]>h[k]+Edge[temp].c )
                   h[v]=h[k]+Edge[temp].c;
          }
     }
}

int Astar_Kth( int n,int s,int t,int K )
{
    Statement cur,nxt;
    //priority_queue<Q>q;
    priority_queue<Statement>FstQ;
    
    int cnt[MAXN];
    memset( cnt,0,sizeof(cnt) );
    cur.v=s;
    cur.d=0;
    cur.h=h[s];
    
    FstQ.push(cur);
    
    while( !FstQ.empty() )
    {
           cur=FstQ.top();
           FstQ.pop();
           
           cnt[cur.v]++;
           if( cnt[cur.v]>K ) continue;
           if( cnt[t]==K )return cur.d;
           
           for( int temp=head[cur.v];temp!=-1;temp=Edge[temp].nxt )
           {
                int v=Edge[temp].v;
                nxt.d=cur.d+Edge[temp].c;
                nxt.v=v;
                nxt.h=h[v];
                FstQ.push(nxt);
           }
    }
    return -1;
}

int main()
{
    int n,m;
    while( scanf( "%d %d",&n,&m )!=EOF )
    {
           int u,v,c;
           memset( head,0xFF,sizeof(head) );
           memset( tail,0xFF,sizeof(tail) );
           
           for( int i=0;i<m;i++ )
           {
                scanf( "%d %d %d",&u,&v,&c );
                addEdge( u,v,c,i );
           }
           int s,t,k;
           scanf( "%d %d %d",&s,&t,&k );
           if( s==t ) k++;
           Dijstra( n,s,t );
           printf( "%d\n",Astar_Kth( n,s,t,k ) );
    }
    return 0;
}