题意:两个人住在城市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;
}