PKU 3613 Cow Relays （指定路徑條數的最短路）

標籤：style   class   blog   code   2014   string   

題意：N，T，S，E：給你T條邊，每條邊兩端都有編號和權值，問從S走到E允許走N條邊，求最短路。

foyld加矩陣快速冪思想。

注意要把邊離散

#include <iostream>#include <fstream>#include <string.h>#include <algorithm>using namespace std;#define M 303#define inf 0x3fffffffstruct node{int a[M][M];node(){for(int i=0;i<M;i++){for(int j=0;j<M;j++)a[i][j]=inf;}}};int n,t,s,e;int mp[1010],cnt;node ma;void foyld(node &a,node &b,node &c){node temp;for(int k=1;k<=cnt;k++){for(int i=1;i<=cnt;i++){for(int j=1;j<=cnt;j++)temp.a[i][j]=min(a.a[i][k]+b.a[k][j],temp.a[i][j]);}}for(int i=0;i<M;i++)for(int j=0;j<M;j++)c.a[i][j]=temp.a[i][j];}void pow(){node ans;for(int i=0;i<M;i++) ans.a[i][i]=0;while(n){if(n&1)foyld(ma,ans,ans);foyld(ma,ma,ma);n>>=1;}printf("%d\n",ans.a[mp[s]][mp[e]]);}int main(){while(~scanf("%d%d%d%d",&n,&t,&s,&e)){int a,b,c;cnt=0;memset(mp,0,sizeof(mp));while(t--){scanf("%d%d%d",&c,&a,&b);if(!mp[a]) mp[a]=++cnt;if(!mp[b]) mp[b]=++cnt;ma.a[mp[a]][mp[b]]=ma.a[mp[b]][mp[a]]=c;}pow();}return 0;}