poj 2449 Remmarguts' Date K短路+A*

题目链接：http://poj.org/problem?id=2449

"Good man never makes girls wait or breaks an appointment!" said the mandarin duck father. Softly touching his little ducks' head, he told them a story.

"Prince Remmarguts lives in his kingdom UDF – United Delta of Freedom. One day their neighboring country sent them Princess Uyuw on a diplomatic mission."

"Erenow, the princess sent Remmarguts a letter, informing him that she would come to the hall and hold commercial talks with UDF if and only if the prince go and meet her via the K-th shortest path. (in fact, Uyuw does not want to come at all)"

Being interested in the trade development and such a lovely girl, Prince Remmarguts really became enamored. He needs you - the prime minister's help!

DETAILS: UDF's capital consists of N stations. The hall is numbered S, while the station numbered T denotes prince' current place. M muddy directed sideways connect some of the stations. Remmarguts' path to welcome the princess might include the same station twice or more than twice, even it is the station with number S or T. Different paths with same length will be considered disparate.

题意描述：王子和喜欢的女孩儿在不同的城堡里，王子为了见女孩儿，必须从自己的城堡走第K条最短的路径到达女孩儿所在的城堡里。求第K条最短路径的长度。

算法分析：K短路的模板题，一般运用A*算法求解。

说明：这道题的K达1000之多，应该是POJ上面这道题的数据不强吧，数据极限的时候估计A*TLE吧。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 #include<queue>
 #define inf 0x7fffffff
 using namespace std;
 const int maxn=+;
 const int M = +;

 int n,m,from,to,K;
 struct Edge
 {
     int to,w;
     int next;
 }edge[M*],edge2[M*];
 int head[maxn],edgenum;
 int head2[maxn],edgenum2;

 void add(int u,int v,int w)
 {
     edge[edgenum].to=v ;edge[edgenum].w=w;
     edge[edgenum].next=head[u] ;head[u]=edgenum++ ;
 }
 void add2(int u,int v,int w)
 {
     edge2[edgenum2].to=v ;edge2[edgenum2].w=w;
     edge2[edgenum2].next=head2[u] ;head2[u]=edgenum2++ ;
 }

 int dis[maxn],vis[maxn];
 void spfa()
 {
     for (int i= ;i<=n ;i++) {dis[i]=inf ;vis[i]= ; }
     queue<int> que;
     que.push(to);
     vis[to]=;
     dis[to]=;
     while (!que.empty())
     {
         int u=que.front() ;que.pop() ;
         vis[u]=;
         for (int i=head[u] ;i!=- ;i=edge[i].next)
         {
             int v=edge[i].to;
             if (dis[v]>dis[u]+edge[i].w)
             {
                 dis[v]=dis[u]+edge[i].w;
                 if (!vis[v])
                 {
                     vis[v]=;
                     que.push(v);
                 }
             }
         }
     }
     return ;
 }

 struct node
 {
     int to,g,f;///评估函数: f=g+h;
     friend bool operator < (node a,node b)
     {
         if (a.f != b.f) return a.f > b.f;
         return a.g > b.g;
     }
 }cur,tail;

 int A_star()
 {
     if (from==to) K++;
     if (dis[from]==inf) return -;
     priority_queue<node> Q;
     cur.to=from ;cur.g= ;cur.f=cur.g+dis[from];
     Q.push(cur);
     int cnt=;
     while (!Q.empty())
     {
         cur=Q.top() ;Q.pop() ;
         int u=cur.to;
         if (u==to) cnt++;
         if (cnt==K) return cur.g;
         for (int i=head2[u] ;i!=- ;i=edge2[i].next)
         {
             tail.to=edge2[i].to;
             tail.g=cur.g+edge2[i].w;
             tail.f=tail.g+dis[edge2[i].to ];
             Q.push(tail);
         }
     }
     return -;
 }

 //int flag[maxn][maxn];
 int main()
 {
     while (scanf("%d%d",&n,&m)!=EOF)
     {
         memset(head,-,sizeof(head));
         memset(head2,-,sizeof(head2));
         edgenum=;
         edgenum2=;
         //memset(flag,0,sizeof(flag));
         int a,b,c;
         for (int i= ;i<m ;i++)
         {
             scanf("%d%d%d",&a,&b,&c);
             //if (flag[a][b]) continue;
             //flag[a][b]=1;
             add(b,a,c);
             add2(a,b,c);
         }
         scanf("%d%d%d",&from,&to,&K);
         spfa();
         int ans=A_star();
         printf("%d\n",ans);
     }
     return ;
 }