http://ac.jobdu.com/problem.php?pid=1162
// 北大11机试
// 九度:1162
// I wanna go home
//
// 思想:把不同的领导的节点分别看成集合1和集合2,由于始终是由
// 节点1到节点2,所以路径的方向总是朝着由集合1流向集合2.
// 这样集合1与集合2之间的边,保留集合1指向集合2,而去除集合2
// 指向集合1. 既然需要删除部分边,使用邻接矩阵比较方便,直接
// 把值定为-1,即可。
//
#include <stdio.h>
#include <vector>
using namespace std;
#define SIZE 600+5
int edge[SIZE][SIZE];
bool mark[SIZE]; // 是否属于结合K
int Dis[SIZE]; // 1到点i的距离,-1为无穷。
int Leader[SIZE]; // 领导: 1 or 2
void Init(int n){
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
edge[i][j] = -1;//-1代表无穷
}
edge[i][i]=0;//自己到自己为0
}
}
int main()
{
#ifdef ONLINE_JUDGE
#else
freopen("E:\\in.txt", "r", stdin);
#endif
int n, m;
while(scanf("%d", &n) != EOF)
{
if(n == 0)
{
break;
}
scanf("%d", &m);
Init(n);
while(m-->0)
{
int a, b, d;
scanf("%d%d%d", &a, &b, &d);
edge[a][b] = edge[b][a] = d;
}
int i, t;
for(i=1; i<=n; i++)
{
scanf("%d", &Leader[i]);
}
/// 该为单向图,只能由集合1中的点,指向集合2中的点
/// 各集合中的点可以互指
for(i=1; i<=n; i++)
{
int j;
for(j=i; j<= n; j++)
{
if(Leader[i] == 2 && Leader[j] == 1)
{
edge[i][j] = -1;//i 不能指向j
}
}
}
/// Dijstra 核心
for(i=1; i<=n; i++)
{
Dis[i]=-1;
mark[i]=false;
}
Dis[1] = 0;
mark[1] = true;
int newP = 1;
for(i=1; i<n;i++)
{
int j;
for(j=1; j<=n; j++)
{
int t, c;
if(edge[newP][j] != -1)
{
t = j;
c = edge[newP][j];
}
if(mark[t] == true)
{
continue;
}
if(Dis[t] == -1 || Dis[t] > Dis[newP] + c)
{
Dis[t] = Dis[newP] + c;
}
}// 遍历链表
int l = Leader[newP];//记录当前节点的阵营
int min=123123123;
for(j=1; j<=n; j++)
{
if(mark[j] == true){
continue;
}
if(Dis[j] == -1)
{
continue;
}
if(Dis[j] < min)
{
min = Dis[j];
newP = j;
}
}
mark[newP] = true;//加入集合K
}
/// Dijstra 结束
// 现已求出,节点1到所有节点的最短距离
printf("%d\n", Dis[2]);// 不存在时Dis[2]刚好为-1
}
return 0;
}