題意:給你一個圖,圖中每個點有對應的危險值,q個詢問,每個詢問給出起點,終點,限制值,需要你計算出從起點走到終點不走那些危險值超過限制值的最短路;(起點和終點的危險值不算)
解題思路:一開始看題目給了10s,想的是用d每次詢問跑一遍dij,在dij里面+一個限制,但超時了(講道理,感覺10s夠啊)。。。然后看別人的解法使用三維floyd解決的,dp[k][i][j],k表示用完危險值排名為k的點松弛后的當前最短路的值
代碼:
#include<bits/stdc++.h> using namespace std; struct node { int id; int val; }a[250]; const int inf=0x3f3f3f3f; int dp[250][250][250]; int n,m,t,cnt; int x,y,w; bool cmp(node a,node b) { return a.val<b.val; } int main() { scanf("%d",&t); while(t--) { memset(dp,inf,sizeof(dp)); scanf("%d%d",&n,&m);cnt++; for(int i=1;i<=n;i++) a[i].id=i; for(int i=1;i<=n;i++) scanf("%d",&a[i].val); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) scanf("%d",&dp[0][i][j]); sort(a+1,a+1+n,cmp);//按危險值排序 for(int k=1;k<=n;k++) { for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { dp[k][i][j]=min(dp[k-1][i][j],dp[k-1][i][a[k].id]+dp[k-1][a[k].id][j]); } } }//floyd printf("Case #%d:\n",cnt); while(m--) { scanf("%d%d%d",&x,&y,&w); int pos=0; for(int i=1;i<=n;i++) { if(a[i].val<=w) pos=i;//找到對應在哪個危險值的點結束 } printf("%d\n",dp[pos][x][y]); } } }
