两个节点间有多条时间花费,太恶心了!必须保存时间最少的。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e3+5;
const int INF = 0x3fffffff;
int G[maxn][maxn];
int p[maxn], vis[maxn];
int n,m;
int opt;
void dfs(int s,int fe,int dif){ //节点 费用 相同阵营数
if(fe > opt || dif > 1) return;
if(s == 2){
if(fe < opt) opt=fe;
return;
}
vis[s] = 1;
for(int i=1;i<=n;i++){
if(vis[i] || G[s][i]==INF) continue;
if(p[s] != p[i]) dfs(i, fe+G[s][i], dif+1);
else dfs(i, fe+G[s][i], dif);
}
vis[s] = 0;
}
int main(){
while(cin>>n){
if(n==0) break;
cin>>m;
fill(G[0], G[0] + maxn*maxn, INF);
int u,v,t;
for(int i=0; i < m; i++){
cin>>u>>v>>t;
if(t < G[u][v]){
G[u][v] = t;
G[v][u] = t;
}
}
memset(vis, 0, sizeof(vis));
for(int i=1; i<=n; i++){
cin>>p[i];
}
opt = INF;
dfs(1, 0, 0);
if(opt==INF) cout<<-1<<endl;
else cout<<opt<<endl;
}
return 0;
}