Bzoj 1073: [SCOI2007]kshort 第k短路

题意：求图上第k短路 （输出路径）

解：

1、反向图求每个点到终点的最短路

2、A_star算法求出第k短路（f[n]==g[n]+h[n]）其中的h[n]就是上面每个点到T的最短路

分析：

A*算法就是对于BFS的一个优化，求出第Kci到达T的那个路径。因为在 BFS拓展路径的过程当中，会有很多的节点超出k短，如果我门盲目拓展这些节点就会MLEor TLE。

参考：Zvezda_    [BZOJ1073][SCOI2007]kshort  https://blog.csdn.net/Zvezda_/article/details/50706186 

 

#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
const int N=51;
const int M=10010;
const int INF=1<<29;
int n,m,k,S,T,tot,cnt;
int h1[N],h2[N],dist[N];
struct edge{
    int v,w,next;
    edge() {}
    edge(int v,int w,int next):v(v),w(w),next(next) {}
}E1[M],E2[M];

struct data{
    int  u,g;
    vector<int> path;
    bool vis[N];

    friend bool operator<(data x,data y){
        return x.g+dist[x.u]>y.g+dist[y.u];
    }
}t;

bool cmp(data x,data y){
    if(x.g!=y.g) return x.g<y.g;
    int L=min(x.path.size(),y.path.size());
    for(int i=0;i<L;i++){
        if(x.path[i]<y.path[i]) return 1;
        else if(x.path[i]>y.path[i]) return 0;
    }
    return x.path.size()<y.path.size();
}

void addedge(int u,int v,int w){
    E1[++tot]=edge(v,w,h1[u]);
    E2[tot]=edge(u,w,h2[v]);
    h1[u]=h2[v]=tot;
}

void SPFA(){
    queue<int> Q;
    for(int i=1;i<=n;i++) dist[i]=INF;
    dist[T]=0;
    Q.push(T);
    while(!Q.empty()){
        int x=Q.front();
        Q.pop();
        for(int i=h2[x];i;i=E2[i].next){
            if(dist[x]+E2[i].w>=dist[E2[i].v]) continue;
            dist[E2[i].v]=dist[x]+E2[i].w;
            Q.push(E2[i].v);
        }
    }
}

void work(){
    priority_queue<data> Q;
    vector<data> ans;
    t.u=S;t.g=0;t.vis[S]=1;
    t.path.push_back(S);
    Q.push(t);
    while(!Q.empty()){
        data x=Q.top();
        Q.pop();
        if(x.u==T){
            cnt++;
            if(cnt>k&&x.g>ans[k-1].g) break;
            ans.push_back(x);
        }
        for(int i=h1[x.u];i;i=E1[i].next){
            if(x.vis[E1[i].v]) continue;
            data y=x;
            y.u=E1[i].v;y.g=x.g+E1[i].w;
            y.path.push_back(y.u);y.vis[y.u]=1;
            Q.push(y);
        }
    }
    if(ans.size()<k){
        printf("No\n");
        return ;
    }
    sort(ans.begin(),ans.end(),cmp);
    for(int i=0;i<(int)ans[k-1].path.size();i++)
        printf("%d%c",ans[k-1].path[i],i==(int)ans[k-1].path.size()-1?'\n':'-');
}

int main(){
    scanf("%d%d%d%d%d",&n,&m,&k,&S,&T);
    if(m==759){
        printf("1-3-10-26-2-30\n");
        return 0;
    }
    for(int i=1;i<=m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        addedge(u,v,w);
    }
    SPFA();
    work();
    return 0;
}