【AcWing 178】第K短路 A*算法

给定一张N个点（编号1,2…N），M条边的有向图，求从起点S到终点T的第K短路的长度，路径允许重复经过点或边。

注意： 每条最短路中至少要包含一条边。

输入格式
第一行包含两个整数N和M。

接下来M行，每行包含三个整数A,B和L，表示点A与点B之间存在有向边，且边长为L。

最后一行包含三个整数S,T和K，分别表示起点S，终点T和第K短路。

输出格式
输出占一行，包含一个整数，表示第K短路的长度，如果第K短路不存在，则输出“-1”。

数据范围
1≤S,T≤N≤1000,
0≤M≤105,
1≤K≤1000,
1≤L≤100
输入样例：
2 2
1 2 5
2 1 4
1 2 2
输出样例：
14

题意：如题

思路：

A*优雅的暴力算法，还不是特别清楚过程，先记录一下。

AC代码：

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include <bitset>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
typedef pair<ll,PII> PIII;
const int maxn = 1e5+200;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'|ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0',  ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };

const int V = 2e3, E=2e5+500;
ll d[V],cost[E];
ll n,m,s,t;
ll k;
ll head[V],tail[V], pnt[E],nxt[E],e=1;      //head引导正向边，tail引导反向边
ll f[V];
ll vis[V], sta[V];

void addedge(ll *head, ll u,ll v,ll c)
{
    pnt[e]=v;       //当前以u为顶点，c为边长的，到v的一条边
    cost[e]=c;      //存入当前边权值
    nxt[e]=head[u];     //下一个其实是前一个
    head[u]=e++;        //当前边编号
}

void Dijkstra()     //先用一次dijkstra求出t到各个点的最短路，作为估计值
{
    rep(i,0,n+1) d[i] = inf;
    priority_queue<PII, vector<PII>, greater<PII> > q;
    d[t] = 0;
    q.push(mp(0LL,t));
    while(!q.empty())
    {
        ll x = q.top().se; q.pop();
        if(vis[x]) continue;
        vis[x] = 1;
        for(int i=tail[x]; i ;i = nxt[i])
        {
            ll v = pnt[i];
            if(d[v]>d[x]+cost[i]&&!vis[v])
            {
                d[v] = d[x] + cost[i];
                q.push(mp(d[v], v));
            }
        }
    }
    memcpy(f,d,sizeof(d));  //丢给f数组
}

ll A_star()     
{
    priority_queue<PIII, vector<PIII>, greater<PIII> > q;   //pair< 估计值+ 到当前最短路，pair<到当前最短路，结点> >
    q.push(mp(f[s], mp(0LL,s) ) );  //初始
    while(!q.empty())
    {
        ll x = q.top().se.se; ll dis = q.top().se.fi; q.pop();      
        if(sta[x]>=k) continue; //对于某个点如果遇到了k次以上，就不用考虑了。
        sta[x] ++;  //计数++
        if(x==t&&sta[x]==k) return dis;     //到终点，且为第k次出队，就是答案了。
        for(int i=head[x];i ;i = nxt[i])
        {
            ll v = pnt[i];
            if(sta[v]<k)    //还没到k次就进队
            {
                ll tmp = dis + cost[i];
                q.push(mp(tmp + f[v], mp(tmp,v) ));
            }
        }
    }
    return -1;
}


int main()
{
    n = read(), m = read();
    rep(i,1,m)
    {
        ll x = read(), y = read(), z = read();
        addedge(head,x,y,z), addedge(tail,y,x,z);   //正反边
    }
    s = read() , t = read() , k = read();
    if(s==t) k++;   //起点终点重合的时候k++确保有解的时候不会错
    Dijkstra();
    cout<<A_star()<<'\n';
    return 0;
}