poj 3767 I Wanna Go Home

题目分析：

             这个题属于典型的dijkstra 算法的题；

             n个村庄，m 条路 和两村庄之间所用时间

              最后是各个城市所属阵营；

题意分析：       

              分为两个阵营，1,2城市分别属于1,2阵营（题意简单化题目）求从城市1到城市2 的最短时间

重点：

            可以从城市1到2，反则不行：因此，在这加上判断条件，套用dijkstra算法就ok了

#include<stdio.h>

#define max 0x7fffffff

int grah[601][601];

int root[601];

int n;

void dij()

{

	int d[601];

    bool vis[601]={false};   

	int i,j,pos=0;

	for(i = 0;i < n; i++)

		d[i] = grah[0][i];

	vis[0] = true ;

	for( i= 1; i <= n; i++)

	{

		int min=max;



		for(j = 0; j < n; j++)

		{

			if( ! vis[j] && min > d[j])

			

			{

				

			    min=d[j];

				pos = j;

			}

		}

		vis[pos] = true;

		for( j = 0;j < n ;j++)

		{

			if( !vis[j] && grah[pos][j] < max && !(root[pos] == 2 && root[j] == 1))

			{

				if(d[j] > d[pos]+ grah[pos][j])

					d[j] = d[pos] + grah[pos][j];

			}

		}

		

	}

	    if(d[1] < max) printf("%d\n",d[1]);

		else  printf("-1\n");

}

int main()

{

//	freopen("e://1.txt","r",stdin);

	int m,x,y,z,i,j;

	while(scanf("%d",&n),n)

	{

		for(i = 0;i < n;i++)

			for(j = 0;j < n;j++)

			{

				if(i==j) grah[i][j]=0;

				else grah[i][j]=max;

			}

		scanf("%d",&m);

		while(m--)

		{

			scanf("%d%d%d",&x,&y,&z);

			grah[x-1][y-1]=grah[y-1][x-1]=z;

		}

		for(i = 0;i < n;i++)

			scanf("%d",root+i);

         dij();

	}

  return 0;

}