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

M.J — Mon, 10 May 2010 05:02:00 GMT

        这是昨天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==2) break;
        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);
    }    
}

M.J 2010-05-10 13:02 发表评论

POJ 1258 Agri-Net【最小生成树(prim)】

M.J — Sat, 08 May 2010 16:19:00 GMT

今天brent讲座，我竟然忘了！！！郁闷的要死，幸好是去上自习了，不然会更后悔。。。今天他讲的生成树，这个我还会一点，多多少少能自我安慰一下，不过在看课件的时候还是有所收获，发现在实现最小生成树的时候可以有两种方法，差别特别小，但时间还是差一点的，不幸的是我原来的方法正是那个稍微慢一点的。简单来说，有三个差别：
1):在初始化dis数组时一个初始化为map[s][i]，(包括dis[s],其中s为源点)，另一个初始化为无穷大，dis[s]=0；
2):一个将将flag[s]标记为true，另一个标记为false
3):最重要的一点，一个循环n-1次，一个循环n次
我把两个算法贴出来对比一下：
16ms的：

 1#include<stdio.h>
 2#include<string.h>
 3#include<stdlib.h>
 4#define INF 0x1f1f1f1f
 5#define M 102
 6int map[M][M],dis[M];
 7bool flag[M];
 8int prim(int s,int n){
 9    int i,j,md,temp,total=0;
10    memset(flag,false,sizeof(flag));
11    flag[s]=true;
12    for(i=1;i<=n;i++)    
13        dis[i]=map[s][i];
14    for(i=1;i<n;i++){
15        md=INF;
16        for(j=1;j<=n;j++){
17            if(!flag[j]&&dis[j]<md){
18                temp=j;
19                md=dis[j];
20            }
21        }
22        total+=md;
23        flag[temp]=true;
24        for(j=1;j<=n;j++)
25            if(!flag[j]&&map[temp][j]<dis[j])
26                dis[j]=map[temp][j];
27    }    
28    return total;
29}
30int main()
31{
32    int i,j,k,n,cas;
33    while(scanf("%d",&n)!=EOF){
34        for(i=1;i<=n;i++)    
35            for(j=1;j<=n;j++){
36                scanf("%d",&map[i][j]);
37                if(map[i][j]==0) map[i][j]=INF;
38            }
39        printf("%d\n",prim(1,n));
40        //system("PAUSE");
41    }
42}

0ms的：

 1#include<stdio.h>
 2#include<string.h>
 3#include<stdlib.h>
 4#define INF 0x1f1f1f1f
 5#define M 102
 6int map[M][M],dis[M];
 7bool flag[M];
 8int prim(int s,int n){
 9    int i,j,md,temp,total=0;
10    memset(flag,false,sizeof(flag));
11    memset(dis,0x1f,sizeof(dis));
12    dis[s]=0;
13    for(i=1;i<=n;i++){
14        md=INF;
15        for(j=1;j<=n;j++){
16            if(!flag[j]&&dis[j]<md){
17                temp=j;
18                md=dis[j];
19            }
20        }
21        total+=md;
22        flag[temp]=true;
23        for(j=1;j<=n;j++)
24            if(!flag[j]&&map[temp][j]<dis[j])
25                dis[j]=map[temp][j];
26    }    
27    return total;
28}
29int main()
30{
31    int i,j,k,n,cas;
32    while(scanf("%d",&n)!=EOF){
33        for(i=1;i<=n;i++)    
34            for(j=1;j<=n;j++)
35                scanf("%d",&map[i][j]);
36        printf("%d\n",prim(1,n));
37    }
38}
39

以后要学会用这个快的。。。~

M.J 2010-05-09 00:19 发表评论

C++博客-M.J的blog-文章分类-最短路，最小生成树

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

POJ 1258 Agri-Net【最小生成树(prim)】