1 /* 2 *题目大意: 3 *在一个有向图中,求从s到t两个点之间的最短路和比最短路长1的次短路的条数之和; 4 * 5 *算法思想: 6 *用A*求第K短路,目测会超时,直接在dijkstra算法上求次短路; 7 *将dist数组开成二维的,即dist[v][2],第二维分别用于记录最短路和次短路; 8 *再用一个cnt二维数组分别记录最短路和次短路的条数; 9 *每次更新路径的条数时,不能直接加1,,应该加上cnt[u][k],k为次短路径或者最短路径的标记; 10 *图有重边,不能用邻接矩阵存储; 11 *不知道为什么,题目上说的是N and M, separated by a single space, with 2≤N≤1000 and 1 ≤ M ≤ 10000; 12 *而我的代码硬是把N开成1W了才过,求解释,RE了无数次,擦; 13 **/ 14 #include<iostream> 15 #include<cstdio> 16 #include<cstring> 17 #include<string> 18 #include<algorithm> 19 using namespace std; 20 21 const int N=11111; 22 const int M=111111; 23 const int INF=0xffffff; 24 25 struct node 26 { 27 int to; 28 int w; 29 int next; 30 }; 31 32 node edge[N]; 33 int head[N]; 34 35 int dist[N][2],cnt[N][2]; 36 bool vis[N][2]; 37 int n,m,s,t,edges; 38 39 void addedge(int u,int v,int w) 40 { 41 edge[edges].w=w; 42 edge[edges].to=v; 43 edge[edges].next=head[u]; 44 head[u]=edges++; 45 } 46 47 int dijkstra() 48 { 49 int k; 50 for(int i=0; i<=n; i++) 51 { 52 dist[i][0]=dist[i][1]=INF; 53 vis[i][0]=vis[i][1]=0; 54 cnt[i][0]=cnt[i][1]=0; 55 } 56 cnt[s][0]=1,dist[s][0]=0; 57 58 for(int i=1; i<=n*2; i++) 59 { 60 int u=-1; 61 int min_dist=INF; 62 for(int j=1; j<=n; j++) 63 for(int flag=0; flag<2; flag++) 64 if(!vis[j][flag]&&dist[j][flag]<min_dist) 65 { 66 min_dist=dist[j][flag]; 67 u=j; 68 k=flag; 69 } 70 if(u==-1) 71 break; 72 vis[u][k]=true; 73 for(int e=head[u]; e!=-1; e=edge[e].next) 74 { 75 int j=edge[e].to; 76 int tmp=dist[u][k]+edge[e].w; 77 78 if(tmp<dist[j][0])//tmp小于最短路径长: 79 { 80 dist[j][1]=dist[j][0];//次短路径长 81 cnt[j][1]=cnt[j][0];//次短路径计数 82 dist[j][0]=tmp;//最短路径长 83 cnt[j][0]=cnt[u][k];//最短路径计数 84 } 85 86 else if(tmp==dist[j][0])//tmp等于最短路径长: 87 { 88 cnt[j][0]+=cnt[u][k];//最短路径计数 89 } 90 91 else if(tmp<dist[j][1])//tmp大于最短路径长且小于次短路径长: 92 { 93 dist[j][1]=tmp;//次短路径长 94 cnt[j][1]=cnt[u][k];//次短路径计数 95 } 96 97 else if(tmp==dist[j][1])//tmp等于次短路径长: 98 { 99 cnt[j][1]+=cnt[u][k];//次短路径计数 100 } 101 } 102 } 103 104 int res=cnt[t][0]; 105 if(dist[t][0]+1==dist[t][1])//判断最短路和次短路是否相差1 106 res+=cnt[t][1]; 107 return res; 108 } 109 110 int main() 111 { 112 //freopen("C:\\Users\\Administrator\\Desktop\\kd.txt","r",stdin); 113 int tcase; 114 scanf("%d",&tcase); 115 while(tcase--) 116 { 117 edges=0; 118 scanf("%d%d",&n,&m); 119 memset(head,-1,sizeof(head)); 120 int u,v,w; 121 for(int i=0; i<m; i++) 122 { 123 scanf("%d%d%d",&u,&v,&w); 124 addedge(u,v,w); 125 } 126 scanf("%d%d",&s,&t); 127 printf("%d\n",dijkstra()); 128 } 129 return 0; 130 }