码迷,mamicode.com
首页 > 其他好文 > 详细

poj 3463 最短路+次短路

时间:2015-05-14 00:51:13      阅读:147      评论:0      收藏:0      [点我收藏+]

标签:

独立写查错不能,就是维护一个次短路的dist

题意:给定一个有向图,问从起点到终点,最短路+比最短路距离长1的路的个数。

Sample Input
2
5 8
1 2 3
1 3 2
1 4 5
2 3 1
2 5 3
3 4 2
3 5 4
4 5 3
1 5
5 6
2 3 1
3 2 1
3 1 10
4 5 2
5 2 7
5 2 7
4 1

Sample Output
3
2

2015-05-14

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<algorithm>
  4 #include<cstring>
  5 #include<cmath>
  6 #include<queue>
  7 #include<map>
  8 using namespace std;
  9 #define MOD 1000000007
 10 const int INF=0x3f3f3f3f;
 11 const double eps=1e-5;
 12 typedef long long ll;
 13 #define cl(a) memset(a,0,sizeof(a))
 14 #define ts printf("*****\n");
 15 const int MAXN=1005;
 16 int n,m;
 17 #define typec int
 18 int tot,src,des;
 19 int head[MAXN],cnt[MAXN][2],dist[MAXN][2],vis[MAXN][2];//dis[i][0]为最短路,dis[i][1]为次短路
 20 struct Edge
 21 {
 22     int to,next,w;
 23 }edge[MAXN*10];
 24 void addedge(int u,int v,int w)
 25 {
 26     edge[tot].to=v;
 27     edge[tot].w=w;
 28     edge[tot].next=head[u];
 29     head[u]=tot++;
 30 }
 31 void init()
 32 {
 33     memset(head,-1,sizeof(head));
 34     tot=0;
 35 }
 36 int dij()
 37 {
 38     int flag;
 39     memset(dist,0x3f,sizeof(dist));
 40     cl(vis);
 41     cl(cnt);
 42     dist[src][0]=0;
 43     cnt[src][0]=1;
 44     for(int j=1;j<2*n;j++)
 45     {
 46         int u;
 47         int Min=INF;
 48         for(int i=1;i<=n;i++)
 49             if(!vis[i][0]&&dist[i][0]<Min)   //找到比最短路还短的路
 50             {
 51                 u=i;
 52                 flag=0;
 53                 Min=dist[i][0];
 54             }
 55             else if(!vis[i][1]&&dist[i][1]<Min)  //说明比最短路长,比次短路短
 56             {
 57                 u=i;
 58                 flag=1;
 59                 Min=dist[i][1];
 60             }
 61         if(u==-1)break;
 62         vis[u][flag]=true;
 63         for(int i=head[u];i!=-1;i=edge[i].next)
 64         {
 65             int v=edge[i].to;
 66             int w=edge[i].w+Min;
 67             if(dist[v][0]>w) //如果找到的新的值比最短路小,则更新最短路和次短路的值
 68             {
 69                 dist[v][1]=dist[v][0];//更新次短路
 70                 dist[v][0]=w;// 更新最短路
 71                 cnt[v][1]=cnt[v][0];
 72                 cnt[v][0]=cnt[u][flag];//更新最短路和次短路的个数
 73 
 74             }
 75             else if(dist[v][0]==w)   //如果值等于最短路
 76                 cnt[v][0]+=cnt[u][flag];//更新最短路的个数
 77             else if(dist[v][1]>w)    //如果找到的值小于次短路的值,更新次短路
 78             {
 79                 dist[v][1]=w;    //更新次短路的值
 80                 cnt[v][1]=cnt[u][flag];  //更新次短路的个数
 81             }
 82             else if(dist[v][1]==w)   //如果找到的值等于次短路的值
 83                 cnt[v][1]+=cnt[u][flag];//更新次短路的个数
 84         }
 85     }
 86     if(dist[des][0]+1==dist[des][1])//如果次短路的值等于最短路值+1
 87         cnt[des][0]+=cnt[des][1];
 88     return cnt[des][0];
 89 }
 90 int main()
 91 {
 92     int tt,u,v,w;
 93     #ifndef ONLINE_JUDGE
 94     freopen("1.in","r",stdin);
 95     #endif
 96     scanf("%d",&tt);
 97     while(tt--)
 98     {
 99         init();
100         scanf("%d%d",&n,&m);
101         while(m--)
102         {
103             scanf("%d%d%d",&u,&v,&w);
104             addedge(u,v,w);
105         }
106         scanf("%d%d",&src,&des);
107         int ans=dij();
108         printf("%d\n",ans);
109     }
110     return 0;
111 }

 

 

2015-1-30:

技术分享
 1 #include<stdio.h>
 2 #include<string.h>
 3 #include<stdlib.h>
 4 #include<queue>
 5 #define VM 1005
 6 #define EM 10010
 7 using namespace std;
 8 const int inf=0x3f3f3f3f;
 9 int head[VM],cnt[VM][2],dist[VM][2],vis[VM][2];//dis[i][0]为最短路,dis[i][1]为次短路
10 int e,src,des,n,m;
11 struct E
12 {
13     int to,w,next;
14 } edge[EM];
15 void add(int cu,int cv,int cw)
16 {
17     edge[e].to=cv;
18     edge[e].w=cw;
19     edge[e].next=head[cu];
20     head[cu]=e ++;
21 }
22 int dij()
23 {
24     int i,j,u,min,flag;
25     memset(dist,0x3f,sizeof(dist));
26     memset(vis,0,sizeof(vis));
27     memset(cnt,0,sizeof(cnt));
28     dist[src][0]=0;
29     cnt[src][0]=1;
30     for(i=1;i<2*n;i++)
31     {
32         min=inf;
33         for(j=1;j<=n;j++)           //找新的最短路和次短路
34             if(!vis[j][0]&&dist[j][0]<min)
35             {
36                 u=j;
37                 flag=0;
38                 min=dist[j][0];
39             }
40             else if(!vis[j][1]&&dist[j][1]<min)
41             {
42                 u=j;
43                 flag=1;
44                 min=dist[j][1];
45             }
46         if(min==inf)
47             break;
48         vis[u][flag]=1;
49         for(j=head[u];j!=-1;j=edge[j].next)
50         {
51             int v=edge[j].to;
52             int w=edge[j].w+min;
53             if(dist[v][0]>w) //如果找到的新的值比最短路小,则更新最短路和次短路的值
54             {
55                 dist[v][1]=dist[v][0];//更新次短路
56                 dist[v][0]=w;// 更新最短路
57                 cnt[v][1]=cnt[v][0];
58                 cnt[v][0]=cnt[u][flag];//更新最短路和次短路的个数
59 
60             }
61             else if(dist[v][0]==w)   //如果值等于最短路
62                 cnt[v][0]+=cnt[u][flag];//更新最短路的个数
63             else if(dist[v][1]>w)    //如果找到的值小于次短路的值,更新次短路
64             {
65                 dist[v][1]=w;    //更新次短路的值
66                 cnt[v][1]=cnt[u][flag];  //更新次短路的个数
67             }
68             else if(dist[v][1]==w)   //如果找到的值等于次短路的值
69                 cnt[v][1]+=cnt[u][flag];//更新次短路的个数
70         }
71 
72     }
73     if(dist[des][0]+1==dist[des][1])//如果次短路的值等于最短路值+1
74         cnt[des][0]+=cnt[des][1];
75     return cnt[des][0];
76 }
77 int main()
78 {
79     int T,u,v,w;
80     scanf("%d",&T);
81     while(T--)
82     {
83         scanf("%d%d",&n,&m);
84         memset(head,0xff,sizeof(head));
85         e=0;
86         while(m--)
87         {
88             scanf("%d%d%d",&u,&v,&w);
89             add(u,v,w);
90         }
91         scanf("%d%d",&src,&des);
92         int ans=dij();
93         printf("%d\n",ans);
94     }
95     return 0;
96 }
View Code

 

poj 3463 最短路+次短路

标签:

原文地址:http://www.cnblogs.com/cnblogs321114287/p/4263009.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!