例题

JZOJ senior 1163第K短路

题目描述

Bessie 来到一个小农场,有时她想回老家看看她的一位好友。她不想太早地回到老家,因为她喜欢途中的美丽风景。她决定选择K短路径,而不是最短路径。
农村有 R (1≤R≤100,000) 条单向的路,每条路连接 N (1≤N≤10000) 个结点中的两个。结点的编号是 1..N。Bessie 从结点 1出发,她的朋友(目的地)在结点 N。
同一个点可以多次经过。K短路的定义:假设从1出发,有M条长度不同的路径可以到达点N,则K短路就是这M条路径中第K小的路径长度。

输入格式

Line 1: 三个用空格分隔的整数 N,R,K((1≤n≤10000,1≤R≤100000,1≤K≤10000)
Lines 2..R+1: 每行包含三个用空格分隔的整数x,y,len(1≤x,y≤n,1≤len≤10000),表示x到y有一条长度为len的单向道路。

输出格式

输出包括一行,一个整数,第K短路的长度。

样例输入

4 4 2
1 2 100
2 4 200
2 3 250
3 4 100

样例输出

450


做法

考虑用A*算法。
我们知道A*算法有个性质:
设h(i)表示从i到终点的估计距离,设h*(i)表示i到终点的实际距离
1、当h(i) < h*(i)时,程序慢,但是保证找到最优解
2、当h(i) = h*(i)时,程序快,并且保证找到最优解
3、当h(i) > h*(i)时,程序快,但是不保证找到最优解
所以,若有元素t在堆顶弹出,若其为终点,且是第K次出堆,则答案为它走过的距离(t.g)

所以,首先我们预处理出h(i)。怎么求?我们可以考虑以上第二种方案。
因为我们可以先将所有的边的方向反过来,以N为起点,求出每一个点到N的最短路径。
最短路径都会吧?跑一遍SPFA(dijsktra没试过)。
h(i)为i到N的最短路径。

接下来开始A*!开一个堆,每次取出f(1到N估计总距离,f(i)=g(i)+h(i))最小的,然后往与它相连的点拓展。弹出后可以再进堆。若弹出的点为终点且是第K次出堆,则输出答案。

有一种坑了我很久的情况:当两条路径一样时,就当做一条看,加特判就行了。
但是不要乱加特判,要看题目,知道它要我们求什么。


代码实现(C++)
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n,r,k;//如题
struct Edge
{
    int x,y,len;
    Edge* las;//记录上一个和此次的x一样的边
} e1[100001],e2[100001];//e1是正向的,e2是反向的(前向星)
Edge *last1[10001],*last2[10001];//last[i]表示x为i的最后一条边
int h[10001];//见“做法”
const int SIZE=1000000;
int que[SIZE];
bool in_que[10001];
inline void spfa(int);
struct Node
{
    int n,g,f;  //n:在第n个点;g:已走g的路径;f:估计距离
} heap[SIZE];//堆
bool cmp(const Node& a,const Node& b)//比较函数
{
    return a.f>b.f;
}
inline void A_star(int,int);
int main()
{
    scanf("%d%d%d",&n,&r,&k);
    int i,j=0,x,y,len;
    for (i=1;i<=r;i++)
    {
        scanf("%d%d%d",&x,&y,&len);
        while(getchar()!='\n');//这句话是因为数据有问题才加上去的,不用管 
        j++;
        e1[j]={x,y,len,last1[x]};//新增一条边
        last1[x]=e1+j;
        e2[j]={y,x,len,last2[y]};
        last2[y]=e2+j;
    }
    spfa(n);
    A_star(1,n);
}
inline void spfa(int u)
{
    memset(h,127,sizeof h);
    h[u]=0;
    int head=0,tail=1;
    Edge* to;
    que[1]=u;
    in_que[u]=1;
    do
    {
        if (++head==SIZE)
            head=0;
        for (to=last2[que[head]];to;to=to->las)
        {
            if (h[to->x]+to->len>=h[to->y])
                continue;
            h[to->y]=h[to->x]+to->len;
            if (in_que[to->y])
                continue;
            if (++tail==SIZE)
                tail=0;
            que[tail]=to->y;
            in_que[to->y]=1;
        }
        in_que[que[head]]=0;
    }
    while (head!=tail);
}
inline void A_star(int u,int v)
{
    int nh=1,cnt=0,ans;
    Edge* to;
    *heap={u,0,h[u]};
    while (nh)
    {
        if (heap->n==v && heap->g!=ans)//特判相等的情况
        {
            ans=heap->g;
            if (++cnt==k)
            {
                printf("%d\n",heap->g);
                return;
            }
        }
        for (to=last1[heap->n];to;to=to->las)
        {
            heap[nh].n=to->y;
            heap[nh].g=heap->g+to->len;
            heap[nh].f=heap[nh].g+h[to->y];
            nh++;
            push_heap(heap,heap+nh,cmp);//加入堆
        }
        pop_heap(heap,heap+nh,cmp);
        nh--;//弹出堆
    }
}