poj3767

分析:由于城市1与城市2所属的集合固定,故在路径中必定有一条而且只能有一条路从属于集合 1的城市出发进入城市2,而定不会出现从集合2走向集合1的城市。

如此,M条路中凡是连接属于不同集合的城市的路为单向路,只能从集合1中的城市走向集合2, 在此基础上用Dijkstra算法求最短路即可

#include
#include
#define INF 1000000
int map[606][606];
int visit[601];
int dist[601];
int leap[601];
int n,m,p;
void dij()
{
    int i,j,min,flag;
    for(i=1;imap[p][j]+dist[p]))
            dist[j]=map[p][j]+dist[p];
            if(!visit[j]&&min>dist[j])
            {
                min=dist[j];
                flag=j;
            }            
        }
        p=flag;
    }
    if(dist[2]==INF)
    printf("%d\n",-1);
    else
    printf("%d\n",dist[2]);
}
int main()
{
    int i,j,x,y,val;
    while(scanf("%d",&n),n)
    {
        scanf("%d",&m);
        for(i=1;i<=n;i++)
        {
            visit[i]=0;
            for(j=1;j<=i;j++)
            map[i][j]=map[j][i]=INF;
        }
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&val);
            map[x][y]=map[y][x]=val;
        }
        for(i=1;i<=n;i++)
        scanf("%d",&leap[i]);
        for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
        {
            if(leap[i]!=leap[j])
            {
                if(leap[i]==1)
                map[j][i]=INF;
                else
                map[i][j]=INF;
            }
        }
        for(i=1;i<=n;i++)
        dist[i]=map[1][i];
        dist[1]=0;
        p=1;
        dij();
    }
    return 0;
}


 

 

 

 


 

 

你可能感兴趣的:(ACM_OJ)