hdu6181 Two Paths

一道A*算法求第k短路的模板题，但是！我有话要说。

可能因为学校老师比较水，我们讲迪杰斯特拉算法的时候老师讲了个n^2算法实现的，教科书上也他妈n^2，我也没细想

后来比赛的时候我自己优化了，结果一百度发现直接优先级队列，elogn。。。。。伤心，错过一道水题。

自己做了个模板，以后直接用。对了，A*算法确实有复杂的用法，但是我们这里只要利用他求第k短就行了。

至于估值函数 F=G+H 这里求H用了迪杰斯特拉算法。

#include<iostream>#include<queue>#include<cstdio>#include<vector>#include<cstring>#include<algorithm>using namespace std;typedef long long ll;int T,n,m,a,b,w;ll inf=1e18;typedef struct node { //dij and mpint at;ll len;bool operator < (const node &te) const { return len>te.len;}}node;typedef struct _node { //Astarint at;ll len;ll tot;bool operator < (const _node &te) const { return tot>te.tot;}}_node;vector<node> mp[100005];ll dis[100005];ll dij(int go) {priority_queue<node> q;for(int i=1;i<=n;i++) {dis[i]=inf;}dis[go]=0;q.push(node{go,0});while(!q.empty()) {node e=q.top(); q.pop();//这里加跳出判断可以求出最短路,也就是单纯的dij for(int i=0;i<mp[e.at].size();i++) {node xe=mp[e.at][i];if(e.len+xe.len<dis[xe.at]) {dis[xe.at]=e.len+xe.len;q.push(node{xe.at,e.len+xe.len});}}}return 0;//返回最短路径,这个题目不需要 }ll Astar(int Start,int End,int Knum) {    priority_queue<_node> q;    q.push(_node{Start,0,dis[Start]});    while(!q.empty()) {    _node e=q.top(); q.pop();    if(e.at==End) {    Knum--;    if(Knum==0)         return e.tot;}    for(int i=0;i<mp[e.at].size();i++) {    node xe=mp[e.at][i];    q.push(_node{xe.at,e.len+xe.len,e.len+xe.len+dis[xe.at]});}}}int main() {scanf("%d",&T);while(T--) {scanf("%d%d",&n,&m);for(int i=1;i<=n;i++) {mp[i].clear();}for(int i=1;i<=m;i++) {scanf("%d%d%d",&a,&b,&w);mp[a].push_back(node{b,w});mp[b].push_back(node{a,w});}int Start=1,End=n,Knum=2;//A*求第Knum段路 ll ans1=dij(n);ll ansKnum=Astar(Start,End,Knum);printf("%lld\n",ansKnum);}return 0;}