M.J的blog

algorithm,ACM-ICPC
随笔 - 39, 文章 - 11, 评论 - 20, 引用 - 0
数据加载中……

POJ.3767 I Wanna Go Home【最短路】

        这是昨天POJ校赛的一道题,昨天中午从12点到1点多一直纠结于这道题,后来有社团的乒乓球比赛就没有做了,今天总算把它搞定了Orz..
        题目很简单,在一个国家有两个group,记做1和2,N个city,每个city属于1或者2。每两个city间有一定的距离,现在要从city1去city2,问最短
的距离是多少,要求至多只有一次穿越时跨过分属不同group的city。city1总是属于group1,city2总属于group2。
        很明显的最短路,但关键是如何建图。可以看到,一共只有两种走法,一种是从city1出发,一直走属于group1的city直到city2,或者一直走属
于group2的city直到city2。我昨天建了两个图来表示两种情况,分别求最短路,然后取小的,结果越写越乱。今天发现其实一次就可以,只要在建
图的时候,如果两个city属于一个group,就建双向图,否则就建从属于group1的city到属于group2的city的单向图。原因很简单,因为两种情况都不会
出现从2到1的情况。建好图然后就简单了,注意无解的情况。
#include<stdio.h>
#include
<string.h>
#define INF 0x1f1f1f1f
#define M 602
int map[M][M],dis[M],team[M];
bool flag[M];
void dijkstra(int s,int n){
    
int i,j,k,md,temp;
    memset(dis,
0x1f,sizeof(dis));
    memset(flag,
false,sizeof(flag));
    dis[s]
=0;
    
for(i=1;i<n;i++){
        md
=INF;
        
for(j=1;j<=n;j++){
            
if(!flag[j]&&dis[j]<md){
                md
=dis[j];
                temp
=j;
            }

        }

        
if(temp==2break;
        flag[temp]
=true;
        
for(j=1;j<=n;j++)
            
if(!flag[j]&&map[temp][j]+md<dis[j])
                dis[j]
=md+map[temp][j];
    }

    
if(dis[2]<map[1][1])printf("%d\n",dis[2]);
    
else printf("-1\n");
}

int main()
{
    
int i,j,k,n,m,p,q;
    
while(scanf("%d",&n),n){
        scanf(
"%d",&m);
        memset(map,
0x1f,sizeof(map));
        
for(j=1;j<=m;j++){
            scanf(
"%d%d%d",&p,&q,&k);
            map[p][q]
=map[q][p]=k;
        }

        
for(j=1;j<=n;j++)
            scanf(
"%d",&team[j]);
        
for(i=1;i<n;i++)
            
for(j=i+1;j<=n;j++){
                
if(team[i]==team[j])                   //如果是一个group,不用管 
                    continue;
                
else if(team[i]==1&&team[j]==2)         //如果一个是1一个是2,则建单向边 
                    map[j][i]=map[1][1];
                
else if(team[i]==2&&team[j]==1)         //同上 
                    map[i][j]=map[1][1];
            }

        dijkstra(
1,n);
    }
    
}

posted on 2010-05-10 13:02 M.J 阅读(1045) 评论(0)  编辑 收藏 引用 所属分类: 最短路,最小生成树


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理