圖論---第k短路

poj2249

A*算法

來自百度百科
A* 算法,A* (A-Star)算法是一種靜態(tài)路網(wǎng)中求解最短路徑最有效的直接搜索方法,也是解決許多搜索問題的有效算法。算法中的距離估算值與實(shí)際值越接近,最終搜索速度越快。

公式表示為: f(n) = g(n) + h(n),

其中, f(n) 是從初始狀態(tài)經(jīng)由狀態(tài)n到目標(biāo)狀態(tài)的代價(jià)估計(jì),
g(n) 是在狀態(tài)空間中從初始狀態(tài)到狀態(tài)n的實(shí)際代價(jià),
h(n) 是從狀態(tài)n到目標(biāo)狀態(tài)的最佳路徑的估計(jì)代價(jià)。
(對(duì)于路徑搜索問題,狀態(tài)就是圖中的節(jié)點(diǎn),代價(jià)就是距離)

h(n)的選取:
保證找到最短路徑(最優(yōu)解的)條件,關(guān)鍵在于估價(jià)函數(shù)f(n)的選取(或者說h(n)的選?。?/p>

我們以d(n)表達(dá)狀態(tài)n到目標(biāo)狀態(tài)的距離,那么h(n)的選取大致有如下三種情況:

  1. 如果h(n)< d(n)到目標(biāo)狀態(tài)的實(shí)際距離,這種情況下,搜索的點(diǎn)數(shù)多,搜索范圍大,效率低。但能得到最優(yōu)解。
  2. 如果h(n)=d(n),即距離估計(jì)h(n)等于最短距離,那么搜索將嚴(yán)格沿著最短路徑進(jìn)行, 此時(shí)的搜索效率是最高的。
  3. 如果 h(n)>d(n),搜索的點(diǎn)數(shù)少,搜索范圍小,效率高,但不能保證得到最優(yōu)解。
    所以取h(n) \le d(n)

模板

求一個(gè)圖中從源點(diǎn) s 到終點(diǎn) t 的第 k 短路

定義數(shù)組d[n],表示從 nt 的距離,以 t 為源點(diǎn)做一次最短路可以得到每個(gè)點(diǎn)到 t 的距離,所以要存一個(gè)反向圖,用來預(yù)處理d[n],直接以d[n]當(dāng)作h(v)

有了d[n]之后,維護(hù)一個(gè)優(yōu)先隊(duì)列,隊(duì)列中的每個(gè)節(jié)點(diǎn)都存放{v,sum},v表示當(dāng)前點(diǎn),sum表示到達(dá)點(diǎn)v一共走了多少距離。優(yōu)先隊(duì)列按照公式f(v) = g(v) + h(v)確定優(yōu)先級(jí),其中g(n)對(duì)應(yīng)sum,h(v)對(duì)應(yīng)d[v],通過優(yōu)先隊(duì)列一次次的擴(kuò)展,如果第k此取到t,那么此時(shí)的sum就是stk短路。

f(v)其實(shí)就是s-v的距離加上v-t的距離,這就是一條路,我們可以知道每條路長(zhǎng)度 ,所以可以找到第k短的那一條,每次取到t,就是找到了一個(gè)比上次 次短 的路.
例題 POJ2249

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <queue>
using namespace std;
#define MAXN 100009

int n,m,s,t,k; 
int d[MAXN];// 每個(gè)點(diǎn)到t的距離
const int inf = 0x3f3f3f3f;
struct  edge{
    int to,len,next;
}e1[MAXN], e2[MAXN];

struct  node{//優(yōu)先隊(duì)列中的節(jié)點(diǎn)
    int v,sum;
    node(int v,int w):v(v),sum(w){}
    bool operator < (const node & o)const{
        return  this->sum+d[this->v] > o.sum+d[o.v];  // 這里就是估值小的放前面,f(v) = g(v) + h(v),sum就是g(v),d[]做h(v)
    }
};
int head1[MAXN], head2[MAXN] , js1,js2; //分別存 原圖,反向圖
bool vst[MAXN];
void init(){
    js1 = 0;js2 = 0;
    memset(head1,-1,sizeof(head1));
    memset(head2,-1,sizeof(head2));
}
void add1(int a, int b, int len){//建原圖
    e1[js1].to = b;
    e1[js1].len = len;
    e1[js1].next = head1[a];
    head1[a] = js1;
    js1++;
}
void add2(int a, int b, int len){//建反向圖
    e2[js2].to = b;
    e2[js2].len = len;
    e2[js2].next = head2[a];
    head2[a] = js2;
    js2++;
}
void spfa(){//用反向圖初始化d[]
    queue<int> que;
    que.push(t);
    memset(vst,0,sizeof(vst));
    memset(d,inf,sizeof(d));
    d[t]=0;
    vst[t]=1;
    while(!que.empty()){
        int u=que.front(); que.pop();
        vst[u]=0;
        for(int i=head2[u]; i!=-1; i=e2[i].next){
            int v=e2[i].to;
            int w=e2[i].len;
            if(d[u]+w < d[v]){
                d[v] = d[u]+w;
                //cout<<d[v]<<endl;
                if(!vst[v]){
                    vst[v]=1;
                    que.push(v);
                }
            }
        }
    }
}
int astar(){//A* 算法
    if(d[s]==inf) return -1;//說明沒有s到t的通路
    priority_queue<node> que;
    que.push(node(s,0)); //從s出發(fā)
    int cnt=0; //記錄到達(dá)了多少次
    while(!que.empty()){
        node u = que.top(); que.pop();
        if(u.v == t){
            cnt++;
            if(cnt == k)return u.sum;
        }
        for(int i=head1[u.v]; i!=-1; i=e1[i].next){
            int v=e1[i].to;
            int w=e1[i].len;
            que.push( node(v, u.sum+w) );
        }
    }
    return  -1; //不存在k短路
}
int main(){
    
    scanf("%d %d",&n, &m);
    init();
    for(int i=0;i<m;i++){
        int a,b,len;
        scanf("%d %d %d",&a,&b,&len);
        add1(a,b,len);
        add2(b,a,len); //反向圖
    }
    scanf("%d %d %d",&s,&t,&k);
    if(s==t)k++;  //poj2449這道題中,如果st重合,那么第一短結(jié)果肯定是0,原地不動(dòng)不當(dāng)作一條路看
    spfa();//初始化估計(jì)值 d[]
    int ac = astar();
    printf("%d\n",ac);
    return 0;
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容