现在的位置: 首页 > 综合 > 正文

POJ 2449 A* K短路

2017年12月13日 ⁄ 综合 ⁄ 共 5561字 ⁄ 字号 评论关闭

 

很久以前便想做做K短路了。只是无奈鄙人这苦逼的小菜啊~。

首先讲讲A*算法吧。众所周知,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)的过高估计,即估计代价太大会使得结果不准确。

这样我们可以理解了BFS的搜索过程,BFS的搜索过程中没有考虑到h(x)的估计代价,也就是说h(x)=0,只考虑g(x)的实际代价。这样根据实际代价来进行搜索,虽然可以说是很恶心的A*,同样地我们可以知道,BFS的解空间确实很大。

第一次写A*,目前只会应用在K短路上。不过也有点感觉了,关键在于h(x)的设计!

谈具体的实现:

首先我们在解空间取出的就是f(x)最小的,这样我们就要运用到优先队列了。这里提供一个使用C++系统优先队列的方法:

  1. #include<queue>   
  2. struct Q{  
  3.     int g,h;  
  4.     bool operator<( Q a )const  
  5.     { return a.g+a.h<g+h; }  
  6. }  
  7. priority_queue<Q>queue;  
  8. Q b;  
  9. queue.push(b);  

C++的STL中自带了优先队列,通过重载运算法"<",可以实现我们需要的对f(x)的自动维护。

描述一下怎样用启发式搜索来解决K短路。

首先我们知道A*的基础公式:f(x)=g(x)+h(x);对h(x)进行设计,根据定义h(x)为当前的x点到目标点t所需要的实际距离。也就是说x->t距离,由于有很多的节点都是到t的距离,为了计算这个估计值,当然必须先算出x->t的最短路径长度。显然x的值很多而t的值只有一个,对每个x去求单源点最短路径当然不划算!于是反过来做,从t点出发到其他点的单源点最短路径,这样吧估价函数h(x)都求出来,注意这样求出来的h(x)=h*(x);

然后就可以对构造完的h(x)开始启发式搜索了。

首先的点当然就是定义头结点了,头结点的已消耗代价为0,估计代价为h[s],下一个点为v;进入队列,开始for循环。每次取出队头的f(x)最小的节点对其他节点进行拓展。对当前节点的拓展次数++,若当前节点的拓展次数超过K,显然不符合要求,则不进行拓展。若对t节点的拓展次数恰好为K,则找到了所需要的。对当前节点的拓展次数即为到当前节点的第几短路。找到需要节点的K短路后,返回g(t)即可,也就是通过K次拓展的实际消耗的长度。

在for循环中的入队情况:当前节点的可拓展所有边,的所有状态都入队,当前节点到拓展节点的实际代价为当前节点的实际代价+两节点之间的边长。下个节点就是拓展节点,估计函数的值则为拓展节点到目标节点的距离h(x);

下面是解决的Code:

  1. #include<iostream>   
  2. #include<cstdio>   
  3. #include<queue>   
  4. #define MAXN 1005   
  5. #define MAXM 200100
      
  6. using namespace std;  
  7.   
  8. struct Node{  
  9.        int v,c,nxt;  
  10. }Edge[MAXM];  
  11.   
  12. int head[MAXN];  
  13. int tail[MAXN];  
  14. int h[MAXN];  
  15.   
  16. struct Statement  
  17. {  
  18.        int v,d,h;  
  19.        bool operator <( Statement a )const  
  20.        {    return a.d+a.h<d+h;   }  
  21. };  
  22.   
  23. void addEdge( int u,int v,int c,int e )  
  24. {  
  25.      Edge[e<<1].v=v;  
  26.      Edge[e<<1].c=c;  
  27.      Edge[e<<1].nxt=head[u];  
  28.      head[u]=e<<1;  
  29.        
  30.      Edge[e<<1|1].v=u;  
  31.      Edge[e<<1|1].c=c;  
  32.      Edge[e<<1|1].nxt=tail[v];  
  33.      tail[v]=e<<1|1;  
  34.      return ;  
  35. }  
  36.   
  37. void Dijstra( int n,int s,int t )  
  38. {  
  39.      bool vis[MAXN];  
  40.      memset( vis,0,sizeof(vis) );  
  41.      memset( h,0x7F,sizeof(h) );  
  42.      h[t]=0;  
  43.      forint i=1;i<=n;i++ )  
  44.      {  
  45.           int min=0x7FFF;  
  46.           int k=-1;  
  47.           forint j=1;j<=n;j++ )  
  48.           {  
  49.                if( vis[j]==false && min>h[j] )  
  50.                    min=h[j],k=j;  
  51.           }  
  52.           if( k==-1 )break;  
  53.           vis[k]=true;  
  54.           forint temp=tail[k];temp!=-1;temp=Edge[temp].nxt )  
  55.           {  
  56.                int v=Edge[temp].v;  
  57.                if( h[v]>h[k]+Edge[temp].c )  
  58.                    h[v]=h[k]+Edge[temp].c;  
  59.           }  
  60.      }  
  61. }  
  62.   
  63. int Astar_Kth( int n,int s,int t,int K )  
  64. {  
  65.     Statement cur,nxt;  
  66.     //priority_queue<Q>q;   
  67.     priority_queue<Statement>FstQ;  
  68.       
  69.     int cnt[MAXN];  
  70.     memset( cnt,0,sizeof(cnt) );  
  71.     cur.v=s;  
  72.     cur.d=0;  
  73.     cur.h=h[s];  
  74.       
  75.     FstQ.push(cur);  
  76.       
  77.     while( !FstQ.empty() )  
  78.     {  
  79.            cur=FstQ.top();  
  80.            FstQ.pop();  
  81.              
  82.            cnt[cur.v]++;  
  83.            if( cnt[cur.v]>K ) continue;  
  84.            if( cnt[t]==K )return cur.d;  
  85.              
  86.            forint temp=head[cur.v];temp!=-1;temp=Edge[temp].nxt )  
  87.            {  
  88.                 int v=Edge[temp].v;  
  89.                 nxt.d=cur.d+Edge[temp].c;  
  90.                 nxt.v=v;  
  91.                 nxt.h=h[v];  
  92.                 FstQ.push(nxt);  
  93.            }  
  94.     }  
  95.     return -1;  
  96. }  
  97.   
  98. int main()  
  99. {  
  100.     int n,m;  
  101.     while( scanf( "%d %d",&n,&m )!=EOF )  
  102.     {  
  103.            int u,v,c;  
  104.            memset( head,0xFF,sizeof(head) );  
  105.            memset( tail,0xFF,sizeof(tail) );  
  106.              
  107.            forint i=0;i<m;i++ )  
  108.            {  
  109.                 scanf( "%d %d %d",&u,&v,&c );  
  110.                 addEdge( u,v,c,i );  
  111.            }  
  112.            int s,t,k;  
  113.            scanf( "%d %d %d",&s,&t,&k );  
  114.            if( s==t ) k++;  
  115.            Dijstra( n,s,t );  
  116.            printf( "%d\n",Astar_Kth( n,s,t,k ) );  
  117.     }  
  118.     return 0;  
  119. }  


A完这题后

1.学会了用数组来定义链表,

2.使用STL的优先队列

3.A*搜索的初步理解

4.K短路的一种求解方法。

:)

抱歉!评论已关闭.