poj2449 Remmarguts' Date(K短路,A*算法)

求S->T的第K短路。估值函数f(x)=g(x)+h(x),g(x)表示从起始搜索点到当前点的代价(即本题中的从S到x的已走距离)。h(n),它表示启发式搜索中最为重要的一部分,即当前结点到目标结点的估值,h(n)设计的好坏,直接影响着具有此种启发式函数的启发式算法的是否能称为A*算法。

而本题中的h(x)就是x点到T的距离,就是实际值,所以复杂度是有保证的。

我们用一个优先队列,每次取出f(x)最小的点,更新邻接的点的g,扔进优先队列即可。当T第K次出现时就是答案。

一个剪枝:当一个点出队次数>k时,就可以不再去用它扩展了。(因为此时已经至少有K条路径小于它了)

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1010
#define M 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,dis[N],h[N],h1[N],num=0,S,T,K,cnt[N];
bool inq[N];
struct edge{
    int to,next,val;
}data[M],data1[M];
struct node{
    int x,g;//g--从S到x的已走距离
    node(int _x,int _g){x=_x;g=_g;}
    friend bool operator<(node a,node b){return a.g+dis[a.x]>b.g+dis[b.x];}
};
inline void spfa(){
    queue<int>q;memset(dis,inf,sizeof(dis));
    q.push(T);inq[T]=1;dis[T]=0;
    while(!q.empty()){
        int x=q.front();q.pop();inq[x]=0;
        for(int i=h1[x];i;i=data1[i].next){
            int y=data1[i].to,val=data1[i].val;
            if(dis[x]+val<dis[y]){
                dis[y]=dis[x]+val;if(!inq[y]) q.push(y),inq[y]=1;
            }
        }
    }
}
inline int A_Star(){
    if(S==T) ++K;priority_queue<node>q;
    q.push(node(S,0));
    while(!q.empty()){
        int x=q.top().x,g=q.top().g;q.pop();
        if(++cnt[x]>K) continue;if(cnt[T]==K) return g;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;q.push(node(y,g+data[i].val));
        }
    }return -1;
}
int main(){
    freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=m;++i){
        int x=read(),y=read(),val=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
        data1[num].to=x;data1[num].next=h1[y];h1[y]=num;data1[num].val=val;
    }S=read();T=read();K=read();spfa();printf("%d\n",A_Star());
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值