POJ 3268 - Silver Cow Party dijkstra+转置矩阵

来源:互联网 发布:清华五道口考研知乎 编辑:程序博客网 时间:2024/05/18 01:41

One cow from each of N farms (1 ≤ N ≤ 1000) conveniently numbered 1..N is going to attend the big cow party to be held at farm #X (1 ≤ X ≤ N). A total of M (1 ≤ M ≤ 100,000) unidirectional (one-way roads connects pairs of farms; road i requires Ti (1 ≤ Ti ≤ 100) units of time to traverse.

Each cow must walk to the party and, when the party is over, return to her farm. Each cow is lazy and thus picks an optimal route with the shortest time. A cow's return route might be different from her original route to the party since roads are one-way.

Of all the cows, what is the longest amount of time a cow must spend walking to the party and back?

Input
Line 1: Three space-separated integers, respectively: NM, and X 
Lines 2.. M+1: Line i+1 describes road i with three space-separated integers: AiBi, and Ti. The described road runs from farm Ai to farm Bi, requiring Ti time units to traverse.
Output
Line 1: One integer: the maximum of time any one cow must walk.
Sample Input
4 8 21 2 41 3 21 4 72 1 12 3 53 1 23 4 44 2 3
Sample Output
10
Hint

Cow 4 proceeds directly to the party (3 units) and returns via farms 1 and 3 (7 units), for a total of 10 time units.

题意:一群牛分别从1~n号农场赶往x号农场参加聚会,农场与农场之间的路是单向的,在n个农场之间有m条路,给出 a ,b , t表示从a号农场到b号农场需要t时间。 每头牛都会选择最短的路,问来回路上(i→x+x→i)花费时间最长的牛花费的时间是多少?


思路:题意其实很简单,一开始的想法也很简单,每头牛从x返回i的最短距离我们很好求,我们以x为源点求一个dijkstra就可以了,其实从i到x的最短距离也好求,我们只需要以i为源点求一个dijkstra找到对应的到x的最短距离,

但是我们枚举每个点的的话,时间复杂度为n^3 明显会超时;

处理方法:我们可以将原图的矩阵转置,也就是说原本从i到x的路径变成了从x到i的路径,从x到i的路径变成了从i到x的路径,那么我们在以x为源点求一遍dijkstra,此时所求的最短距离就为i到x的最短距离.

其实矩阵也不需要真的转置,我们只需要将 横纵坐标对调一下就好了。


#include<iostream>

#include<cstdio>

#include<cstring>
#include<cmath>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
int map[1100][1100];
int book[1100],dg[1100],db[1100];
int n,m,x;
void dijkstra()
{   int i,j,min,v;
    for(i=1;i<=n;i++)
{   book[i]=0;
   db[i]=map[x][i];//回去的各个点的最短路径是以x为源点 
   dg[i]=map[i][x];//因为反向转置一下,以x为源点求去的时候的最短路径,正好是个反的 
}
for(i=1;i<=n;i++)
{    min=inf;
   for(j=1;j<=n;j++)
   {   if(!book[j]&&dg[j]<min)
    {   min=dg[j];
        v=j;
}
}
book[v]=1;
for(j=1;j<=n;j++)
{  if(dg[j]>dg[v]+map[j][v])//求回去的路径应该是相反的; 
  dg[j]=dg[v]+map[j][v];
}

memset(book,0,sizeof(book));
for(i=1;i<=n;i++)
{    min=inf;
   for(j=1;j<=n;j++)
   {   if(!book[j]&&db[j]<min)
    {   min=db[j];
        v=j;
}
}
book[v]=1;
for(j=1;j<=n;j++)
{  if(db[j]>db[v]+map[v][j])
  db[j]=db[v]+map[v][j];
}

return ;
}
int main()
{      int i,j,k;
       int a,b,c;
     while(cin>>n>>m>>x)
     {    
        for(i=1;i<=n;i++)
         for(j=1;j<=n;j++)
         {  if(i==j)
            map[i][j]=0;
            else
            map[i][j]=inf;
}
   for(i=1;i<=m;i++)
          {  scanf("%d %d %d",&a,&b,&c);
             map[a][b]=c;
 }
 dijkstra();
 int ans=-1;
 for(i=1;i<=n;i++)
 ans=max(ans,dg[i]+db[i]);
 printf("%d\n",ans);
}
return 0;
}

1 0
原创粉丝点击