F-Moving On

Moving On

题意：两个人住在城市u，每天要去城市v,但是要经过一些城市和街道，而且每个城市都有抢劫等的概率，要求找一条路的从城市u到城市v而且抢劫率不超过w的最短路径。

#include<bits/stdc++.h>
#define EPS 1e-9
using namespace std;

typedef long long ll;
const int INF=0x3f3f3f3f;
const int MAXN=211;

int rob[211];
int dp[211][211][211];
int id[211];

bool cmp(int i,int j)
{
    return rob[i]<rob[j];
}

int main()
{
    int Tests;
    scanf("%d",&Tests);
    for(int cntT=1; cntT<=Tests; ++cntT)
    {
        memset(dp,INF,sizeof(dp));
        int N,Q;
        scanf("%d%d",&N,&Q);
        for(int i=1; i<=N; ++i)
        {
            id[i]=i;
            scanf("%d",&rob[i]);
        }
        sort(id+1,id+N+1,cmp);//按危险值从小到大排列城市序号

        for(int i=1; i<=N; ++i)
            for(int j=1; j<=N; ++j)
                scanf("%d",&dp[0][i][j]);
        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][id[k]]+dp[k-1][id[k]][j]);//dp[k][i][j]表示 添加(第1~k小的危险值的城市后)的 i->j 最短路
                }
        }
        printf("Case #%d:\n",cntT);
        for(int q=1; q<=Q; ++q)
        {
            int u,v,dan;
            scanf("%d%d%d",&u,&v,&dan);
            int k=0;
            for(int i=1; i<=N; ++i)
                if(rob[id[i]]<=dan) k=i;
            printf("%d\n",dp[k][u][v]);
        }
    }
    return 0;
}