UESTC

传送门：

http://www.qscoj.cn/#/problem/show/1987

童心未泯的帆宝和乐爷 Edit

Time Limit: 10000 MS Memory Limit: 256 MB

6·1即将来临，游乐园推出了新的主题活动，雨过天晴，帆宝和乐爷童心未泯，准备一探究竟。

兴奋的他们一入园便和孩子们打成一片，不知不觉便走散了。

当他们意识到的时候，只能通过手机来确认对方的位置。

他们当然想尽快找到对方，然而由于孩子们实在是太多，只能选择距离稍远的但是游客稀少的路会合。

帆宝希望找到第

帆宝迫切地想知道该条路径的长度，而乐于助人的你也一定会帮助她的。

Input

第一行三个整数

第二行两个整数

接下来

Output

第一行一个整数

无解输出

Sample input and output

Sample Input	Sample Output
3 3 2 1 2 1 2 2 1 3 4 3 2 1	5

Hint

Source

2018 UESTC ACM Training for Search Algorithm and String

分析：

第k短路，A*+spfa解决

属于模板题

但是我A*+优化的迪杰斯特拉超时。。。

mmp

还拿着在poj交了一道题，可以过啊

应该是测试数据不适合迪杰斯特拉吧

。。。。。。。。。。（自我安慰）

code：

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int AX = 1e5+66;
const int MAXN = 1e3+66;
int n,m,k;
int s,t;
int tot;
int retot;
struct edge
{
    int to,w;
    int next1;
} G[AX],RG[AX];

struct Node
{
    int v;
    int f,h,g;
    bool operator < (const Node &a) const
    {
        return f==a.f? g>a.g : f>a.f;
    }
};

int dis[MAXN];
int head[MAXN];
int rehead[AX];
int vis[MAXN];

void add_edge(int u,int v,int c)
{
    G[tot].to=v;
    G[tot].w=c;
    G[tot].next1=head[u];
    head[u]=tot++;

    RG[retot].to=u;
    RG[retot].w=c;
    RG[retot].next1=rehead[v];
    rehead[v]=retot++;
}
void SPFA()
{
    for(int i=1; i<=n; i++) dis[i]=INF;
    dis[t]=0;
    queue<int> Q;
    Q.push(t);
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
        for(int i=rehead[u]; i!=-1; i=RG[i].next1)
        {
            int v=RG[i].to ;
            int w=RG[i].w ;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                Q.push(v);
            }
        }
    }
}

int Astar(Node a)//A*算法
{
    memset(vis,0,sizeof(vis));
    if(dis[s]==INF) return -1;//如果没有与S相连的点
    if(s==t) k++;
    priority_queue<Node> Q;
    Q.push(a);
    while(!Q.empty())
    {
        Node tmp=Q.top();
        Q.pop();
        int v=tmp.v;
        vis[v]++;
        if(vis[t]==k) return tmp.g;
        for(int i=head[v]; i!=-1; i=G[i].next1)
        {
            Node p;
            p.v=G[i].to;
            p.h=dis[G[i].to];
            p.g=tmp.g+G[i].w;
            p.f=p.g+p.h;
            Q.push(p);
        }
    }
    return -1;
}

int main()
{
    tot=0;
    retot=0;
    memset(head,-1,sizeof head);
    memset(rehead,-1,sizeof rehead);
    scanf("%d%d%d",&n,&m,&k);
    scanf("%d%d",&s,&t);
    int x,y,w;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d%d",&x,&y,&w);
        add_edge(x,y,w);
    }
    SPFA();
    Node a;
    a.v=s;
    a.g=0;
    a.h=dis[s];
    a.f=a.g+a.h;
    int g=Astar(a);
    printf("%d
",g);
    return 0 ;
}

也贴一下超时的A*+优化的迪杰斯特拉

#include <iostream>
#include <cstdio>
#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<math.h>
#include<memory>
#include<queue>
#include<vector>
using namespace std;
typedef long long LL;
#define max_v 10010
#define INF 99999999
struct node
{
    int to,val;
    node(){}
    node(int a,int b)
    {
        to=a;
        val=b;
    }
};
vector<node> e[max_v],ee[max_v];
int n,m,k;
int vis[max_v];
int dis[max_v];
void addEdge(int x,int y,int val)
{
    e[x].push_back(node(y,val));
    ee[y].push_back(node(x,val));//把图反向
}
void Dijkstra(int s,int t)
{
    priority_queue<int,vector<int>,greater<int> > q;
    while(!q.empty())
        q.pop();

    for(int i=1;i<=n;i++)
        vis[i]=0,dis[i]=INF;

    vis[t]=1,dis[t]=0,q.push(t);

    int u,len;
    while(!q.empty())
    {
        u=q.top();
        q.pop();

        len=ee[u].size();
        for(int i=0;i<len;i++)
        {
            node v=ee[u][i];
            if(dis[v.to]>dis[u]+v.val)
            {
                dis[v.to]=dis[u]+v.val;
                if(!vis[v.to])
                {
                    q.push(v.to);
                    vis[v.to]=1;
                }
            }
        }
        vis[u]=0;
    }
}
struct Anode
{
    int h,g,id;
    Anode(int a,int b,int c){h=a;g=b;id=c;}
    bool operator<(Anode a) const
    {
        return h+g>a.h+a.g;
    }
};
priority_queue<Anode> Q;

int Astar(int s,int t)//A*算法
{
    while(!Q.empty())
        Q.pop();
    Q.push(Anode(0,dis[s],s));

    int len,num;
    num=0;
    while(!Q.empty())
    {
        Anode u=Q.top();
        Q.pop();

        if(u.id==t)
            ++num;
        if(num>=k)
            return u.h;

        len=e[u.id].size();
        for(int i=0;i<len;i++)
        {
            node v=e[u.id][i];
            Q.push(Anode(u.h+v.val,dis[v.to],v.to));
        }
    }
    return -1;//不能连通或者没有k短路
}
int main()
{
    int x,y,v,s,t;
    while(~scanf("%d %d %d",&n,&m,&k))
    {
        scanf("%d %d",&s,&t);
        for(int i=0;i<max_v;i++) e[i].clear(),ee[i].clear();
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&v);
            addEdge(x,y,v);
        }
        if(s==t) k++;
        Dijkstra(s,t);
        printf("%d
",Astar(s,t));
    }
    return 0;
}
//有向图 第k短路模板