思路:这个我一开始没想通。后面再讨论区里面发现了一个非常好的思路。
因为要求从点1到点2,且最多只能经过一条边的两端的点处于不同的阵营。如何在求最短路的同时,满足这个限制要求呢?其实我们可以这样,如何某条边连接了两个不同的阵营,那么就限制这条边只能从一个方向走(即规定,只能从阵营1到阵营2)。其实当我们做了这样的规定以后,我们就可以直接用Floyd来做了,因为我们已经满足了题意。
如果不能理解的话,建议直接画一个图。
#include <bits/stdc++.h>
using namespace std;
#define INF 99999999
const int maxn = 610;
const int maxm = 10010;
int n;
int dp[maxn][maxn];
int leader[maxn];
void floyd(){
for(int k = 1; k <= n; ++k){
for(int i = 1; i <= n; ++i){
for(int j = 1; j <= n; ++j){
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
}
}
}
}
int main(){
int m, a, b, c;
while(cin >> n >> m && n != 0){
for(int i = 1; i <= n; ++i){
for(int j = 1; j <= n; ++j){
if(i == j) dp[i][j] = 0;
dp[i][j] = INF;
}
}
for(int i = 0; i < m; ++i){
cin >> a >> b >> c;
if(dp[a][b] > c){
dp[a][b] = dp[b][a] = c;
//一开始边应该先保证是双向的
}
}
for(int i = 1; i <= n; ++i){
cin >> leader[i];
}
for(int i = 1; i <= n; ++i){
for(int j = 1; j <= n; ++j){
//只要把不在一个阵营的边设置成有向边即可:leader1->deader2通,leader2->leader1不通
if(leader[i] == 2 && leader[j] == 1){
dp[i][j] = INF;
}
}
}
floyd();
if(dp[1][2] == INF) cout << "-1" << endl;
else cout << dp[1][2] << endl;
}
return 0;
}