I Wanna Go Home【北京大学】

题目链接

两个节点间有多条时间花费，太恶心了！必须保存时间最少的。

#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;
}