POJ 2449 K短路 A*搜索

题目链接


题意:
求一个有向图的 k k k短路长度。


思路:
经典问题的经典算法—— A ∗ A* A搜索。
感觉 A ∗ A* A搜索和 D i j k s t r a Dijkstra Dijkstra堆优化写法有很多类似的地方。

首先以每个点到终点的最短距离作为估计函数 g g g,故先需要反向建图,随后对反图从终点 T T T做一遍 D i j k s t r a Dijkstra Dijkstra

随后从起点开始 A ∗ A* A搜索,第 k k k个搜索到的值即为第 k k k短路长度。

另外注意当 S = = T S == T S==T时, k k k需要加 1 1 1,因为必须经过至少一条路径。

此题得解。


代码:

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;

const int INF = 1e8 + 10;
const int A = 2e3 + 10;
const int B = 2e5 + 10;
class Gra{
public:
    int v,w,next;
}G[B];

class Gra_Inv{
public:
    int v,w,next;
}IG[B];

int n,m,S,T,K;
int head[A],Ihead[A],g[A],dis[A],tot,Itot;
bool vis[A];

class P{
public:
    int id,W;

    P(int _id = 0,int _w = 0){
        id = _id;
        W = _w;
    }

    bool operator<(const P& rhs)const{
        return W+g[id] > rhs.W+g[rhs.id];
    }
};
priority_queue<P> que;

void init(){
    memset(head,-1,sizeof(head));
    memset(Ihead,-1,sizeof(Ihead));
    memset(g,0,sizeof(g));
    tot = Itot = 0;
}

void add(int u,int v,int w){
    G[tot].v = v;
    G[tot].w = w;
    G[tot].next = head[u];
    head[u] = tot++;
}

void Iadd(int u,int v,int w){
    IG[Itot].v = v;
    IG[Itot].w = w;
    IG[Itot].next = Ihead[u];
    Ihead[u] = Itot++;
}

void Dijkstra(int st){
    while(que.size()) que.pop();
    for(int i=0 ;i<=n ;i++){
        dis[i] = INF;
        vis[i] = 0;
    }

    que.push(P(st,0));
    dis[st] = 0;

    while(que.size()){
        P x = que.top();que.pop();

        int u = x.id;
        if(vis[u]) continue;
        vis[u] = 1;
       
        for(int i=Ihead[u] ;i!=-1 ;i=IG[i].next){
            int v = IG[i].v,w = IG[i].w;
            if(dis[v] > dis[u] + w){
                dis[v] = dis[u] + w;
                que.push(P(v,dis[v]));
            }
        }
    }

    for(int i=1 ;i<=n ;i++){
        g[i] = dis[i];
    }
}

int a_star(int st){
    while(que.size()) que.pop();

    que.push(P(st,0));
    int cnt = 0;
    while(que.size()){
        P x = que.top();que.pop();
        int u = x.id;

        if(u == T){
            cnt++;
            if(cnt == K) return x.W;
        }

        for(int i=head[u] ;i!=-1 ;i=G[i].next){
            que.push(P(G[i].v,G[i].w + x.W));
        }
    }
    return -1;
}

int main(){
    scanf("%d%d",&n,&m);
    init();

    for(int i=1 ;i<=m ;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);Iadd(v,u,w);
    }

    scanf("%d%d%d",&S,&T,&K);
    if(S == T) K++;

    Dijkstra(T);
    printf("%d\n",a_star(S));
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值