gym-102222-F(moving on)-floyd


題意:給你一個圖,圖中每個點有對應的危險值,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]);
        }
    }
}

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM