poj 2449 k短路

 

poj 2449 k短路

分类： 最短路 2012-11-09 20:41  327人阅读  评论(0)  收藏  举报

一道给定图，求S->T的k短路，想了半天还是没想法，看了别人题解才懂得，先求反向图源点为t的最短路，我是用dij的优先队列写的。然后用对于正向图，利用ASTAR算法，

利用优先队列，估价函数为f[i]=w+d[i]，意思就是，我走了一个距离w,还有最短d[i]距离再到k，满足d[i]<=实际代价，ASTAR算法成立，这样对于每个节点，把所有与之相邻的点，压入优先队列，这样访问到节点t第k次时为k短路，返回此时的w,里面一个剪纸，对于一个点访问了大于k次就直接continue，因为这样走下去，肯定为>k短路。

code:

[cpp]  view plain copy

1. #include<cstdio>  
2. #include<cstring>  
3. #include<queue>  
4. #include<vector>  
5. using namespace std;  
6. int n,m,s,t,k,d[1010],find[1010],cnt[1010];  
7. const int inf=999999999;  
8. struct node  
9. {  
10.     int y,l;  
11. };  
12. struct dij  
13. {  
14.     int v,w;  
15.     friend bool operator<(dij a,dij b)  
16.     {  
17.         return a.w>b.w;  
18.     }  
19. };  
20. struct aastar  
21. {  
22.     int v,w;  
23.     friend bool operator<(aastar a,aastar b)  
24.     {   return a.w+d[a.v]>b.w+d[b.v];}  
25. };  
26. vector<node>vt[1010];  
27. vector<node>rvt[1010];  
28. void dijstra()  
29. {  
30.     for(int i=1;i<=n;i++)d[i]=inf;  
31.     d[t]=0;  
32.     memset(find,0,sizeof(find));  
33.     priority_queue<dij>q;  
34.     dij ss,tt;  
35.     ss.v=t;  
36.     ss.w=0;  
37.     q.push(ss);  
38.     while(!q.empty())  
39.     {  
40.         ss=q.top();  
41.         q.pop();  
42.         int v=ss.v;  
43.         int w=ss.w;  
44.         find[v]=1;  
45.         for(int i=0;i<rvt[v].size();i++)  
46.         {  
47.             int y=rvt[v][i].y;    
48.             int l=rvt[v][i].l;  
49.             if(!find[y]&&d[y]>d[v]+l)  
50.             {  
51.                 d[y]=d[v]+l;  
52.                 tt.v=y;  
53.                 tt.w=d[y];  
54.                 q.push(tt);  
55.             }  
56.         }  
57.     }  
58. }  
59. int Astar()  
60. {  
61.     memset(cnt,0,sizeof(cnt));  
62.     aastar ss,tt;  
63.     ss.v=s,ss.w=0;  
64.     if(d[s]==inf)return -1;  
65.     if(s==t)k++;  
66.     priority_queue<aastar>q;  
67.     q.push(ss);  
68.     while(!q.empty())  
69.     {  
70.         ss=q.top();  
71.         q.pop();  
72.         int v=ss.v;  
73.         int w=ss.w;  
74.         if(v==t)  
75.         {  
76.             cnt[t]++;  
77.             if(cnt[t]==k)  
78.                 return w;  
79.         }  
80.         else  
81.         cnt[v]++;  
82.         if(cnt[v]>k)continue;  
83.         for(int i=0;i<vt[v].size();i++)  
84.         {  
85.             int y=vt[v][i].y;  
86.             int l=vt[v][i].l;  
87.             tt.v=y,tt.w=w+l;  
88.             q.push(tt);  
89.         }  
90.     }  
91.     return -1;  
92. }  
93. int main()  
94. {  
95.     while(2==scanf("%d%d",&n,&m))  
96.     {  
97.         int a,b,c;  
98.         for(int i=1;i<=n;i++)  
99.         {  
100.             vt[i].clear();  
101.             rvt[i].clear();  
102.         }  
103.         for(int i=0;i<m;i++)  
104.         {  
105.             scanf("%d%d%d",&a,&b,&c);  
106.             node s;  
107.             s.y=b,s.l=c;  
108.             vt[a].push_back(s);  
109.             s.y=a;  
110.             rvt[b].push_back(s);  
111.         }  
112.         scanf("%d%d%d",&s,&t,&k);  
113.         dijstra();  
114.         printf("%d\n",Astar());  
115.     }  
116.     return 0;  
117. }