Floyd+DP
最开始打算直接最短路径做的,后来觉得思路不对~网上参考了下大神的博客才知道这个题是Dp;
转移方程为dp[i][j]=min(dp[k-1][j],dp[k][j]+map[k][i]),j表示的是使用鞋的次数,k表示的是枚举使用鞋的地点。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=101;
const int inf=1<<29;
int a,b,m,l,K,n;
int map[maxn][maxn],dp[maxn][maxn];
bool fly[maxn][maxn];
void Floyd()
{
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(map[i][j]>map[i][k]+map[k][j])
{
map[i][j]=map[i][k]+map[k][j];
if(k<=a&&map[i][j]<=l)
fly[i][j]=fly[j][i]=1;
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(fly,0,sizeof(fly));
scanf("%d%d%d%d%d",&a,&b,&m,&l,&K);
n=a+b;
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
map[i][j]=map[j][i]=inf;
for(int i=0;i<m;i++)
{
int ita,itb,itc;
scanf("%d%d%d",&ita,&itb,&itc);
map[ita][itb]=map[itb][ita]=itc;
if(map[ita][itb]<=l)
fly[ita][itb]=fly[itb][ita]=1;
}
Floyd();
for(int i=1;i<=n;i++)
dp[i][0]=map[1][i];
dp[1][0]=0;
for(int i=2;i<=n;i++)
for(int j=1;j<=K;j++)
{
int mini=inf;
for(int k=1;k<i;k++)
if(fly[k][i])
mini=min(mini,min(dp[k][j-1],dp[k][j]+map[k][i]));
else
mini=min(mini,dp[k][j]+map[k][i]);
dp[i][j]=mini;
}
printf("%d\n",dp[n][K]);
}
return 0;
}