题目链接
题目中有这样的一个要求:要求次短路刚好和最短路差一个单位距离的时候才可以走这条次短路。
问:最短路的条数和次短路的条数和数是多少?
我的做法是我们可以去维护从起点到达目前点的最短距离,并且记录其条数。但是一开始用spfa()去写的时候,发现由于需要不断的更新,所以会使得最后得到的总数和偏大,这样就存在了问题,所以我们需要再去考虑一个不会有重复的方法。
那么我想到的就是去用优先队列去维护路径,我们知道有dist[ ][0]和dist[ ][1]分别管理最短路和次短路,那么我们不断的取出最短的点,也就是意味着最小的点不会再被覆盖,那么就可以维护了每个点不会重复的多加了。
接下去,考虑有最短路与次短路,有时候我们会知道某个点的次短路比一个点的最短路都要短,所以,我们要还是要输出最短的路,也就是意味着,假如这个次短路比那个最短路都要短的时候,我们也是先去判断这个次短路。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
#define MP3(a, b, c) MP(MP(a, b), c)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 1e3 + 7, maxE = 1e4 + 7;
int N, M, head[maxN], cnt, s, t, dist[maxN][2], num[maxN][2];
struct Eddge
{
int nex, to, val;
Eddge(int a=-1, int b=0, int c=0):nex(a), to(b), val(c) {}
}edge[maxE];
inline void addEddge(int u, int v, int val)
{
edge[cnt] = Eddge(head[u], v, val);
head[u] = cnt++;
}
struct node
{
int id, op;
node(int a=0, int b=0):id(a), op(b) {}
friend bool operator < (node e1, node e2) { return dist[e1.id][e1.op] > dist[e2.id][e2.op]; }
};
priority_queue<node> Q;
bool vis[maxN][2];
inline void Dijkstra()
{
while(!Q.empty()) Q.pop();
Q.push(node(s, 0));
while(!Q.empty())
{
node now = Q.top(); Q.pop();
int u = now.id, op = now.op;
if(vis[u][op]) continue;
vis[u][op] = true;
for(int i=head[u], v, w; ~i; i=edge[i].nex)
{
v = edge[i].to; w = edge[i].val;
if(dist[v][0] > dist[u][op] + w)
{
if(num[v][0])
{
dist[v][1] = dist[v][0];
num[v][1] = num[v][0];
Q.push(node(v, 1));
}
dist[v][0] = dist[u][op] + w;
num[v][0] = num[u][op];
Q.push(node(v, 0));
}
else if(dist[v][0] == dist[u][op] + w)
{
num[v][0] += num[u][op];
}
else if(dist[v][1] > dist[u][op] + w)
{
dist[v][1] = dist[u][op] + w;
num[v][1] = num[u][op];
Q.push(node(v, 1));
}
else if(dist[v][1] == dist[u][op] + w)
{
num[v][1] += num[u][op];
}
}
}
}
inline void init()
{
cnt = 0;
memset(head, -1, sizeof(head));
memset(num, 0, sizeof(num));
memset(dist, INF, sizeof(dist));
memset(vis, false, sizeof(vis));
}
int main()
{
int T; scanf("%d", &T);
while(T--)
{
scanf("%d%d", &N, &M);
init();
for(int i=1, u, v, w; i<=M; i++)
{
scanf("%d%d%d", &u, &v, &w);
addEddge(u, v, w);
}
scanf("%d%d", &s, &t);
dist[s][0] = 0;
num[s][0] = 1;
Dijkstra();
printf("%d\n", num[t][0] + (dist[t][0] == dist[t][1] - 1 ? num[t][1] : 0) );
}
return 0;
}