POJ2449 A*+第K短路

题意：s到t的第K短路

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 1005;
const int maxm = 200005;
const int inf = 9999999;
struct node{
    int u,val,next;
}edge[ maxm ];
struct node2{
    int u,h,d;
    bool operator <( node2 a )const 
    {
        return a.d+a.h<d+h;
    }
};
int cnt,head[ maxn ],tail[ maxn ];
int n;
int h[ maxn ],vis[ maxn ];

void init(){
    cnt=0;
    memset( head,-1,sizeof( head ));
    memset( tail,-1,sizeof( tail ));
}
/*
void addedge( int u,int v,int c,int e )  
{  
     edge[e<<1].u=v;  
     edge[e<<1].val=c;  
     edge[e<<1].next=head[u];  
     head[u]=e<<1;  
       
     edge[e<<1|1].u=u;  
     edge[e<<1|1].val=c;  
     edge[e<<1|1].next=tail[v];  
     tail[v]=e<<1|1;  
     return ;  
}  
*/

void addedge( int a,int b,int c ){
    edge[ cnt ].u=b;
    edge[ cnt ].val=c;
    edge[ cnt ].next=head[ a ];
    head[ a ]=cnt++;

    edge[ cnt ].u=a;
    edge[ cnt ].val=c;
    edge[ cnt ].next=tail[b];
    tail[ b ]=cnt++;

}


void Dij( int s,int t ){
    for( int i=1;i<=n;i++ ){
        h[ i ]=inf;
        vis[ i ]=0;
    }
    h[ s ]=0;
    for( int i=1;i<=n;i++ ){
        int min,k;
        min=inf;
        for( int j=1;j<=n;j++ ){
            if( min>h[ j ]&&vis[j]==0 ){
                min=h[j];
                k=j;
            }
        }
        if( min==inf ) break;
        vis[ k ]=1;
        for( int j=tail[ k ];j!=-1;j=edge[ j ].next ){
            int v=edge[ j ].u;
            if( vis[ v ]==0 && h[ v ]>h[ k ]+edge[ j ].val ){
                h[ v ]=h[ k ]+edge[ j ].val;
            }
        }
    }
    return ;
}

int A_star( int s,int t,int k ){
    node2 cur,next;
    priority_queue<node2>q;
    while( !q.empty() )
        q.pop();
    int cnt[ maxn ];
    memset( cnt,0,sizeof( cnt ));
    cur.u=s;
    cur.h=h[ s ];
    cur.d=0;
    q.push( cur );
    while( !q.empty() ){
        cur=q.top();
        q.pop();
        cnt[ cur.u ]++;
        if( cnt[ cur.u ]>k ) continue;
        if( cnt[ t ]==k )return cur.d;
        for( int i=head[ cur.u ];i!=-1;i=edge[ i ].next ){
            next.u=edge[ i ].u;
            next.d=cur.d+edge[ i ].val;
            next.h=h[ next.u ];
            q.push( next );
        }
    }
    return -1;
}

int main(){
    int m;
    while( scanf("%d%d",&n,&m)!=EOF ){
        int a,b,c;
        init();
        for( int i=0;i<m;i++ ){
            scanf("%d%d%d",&a,&b,&c);
            addedge( a,b,c );
        }
        int s,t,k;
        scanf("%d%d%d",&s,&t,&k);
        if( s==t )k++;
        Dij( t,s );
        printf("%d\n",A_star( s,t,k ));
    }
    return 0;
}