POJ 3463 最短路 次短路

本题是求最短路和比最短路距离长1的次短路的个数，于是就用到了dijkstra

主要的改变就是数组都开到了二维，第二维用来表示是最短路还是次短路

比如d[][]数组和vis[][]数组

而cnt数组使用来存取最短路和次短路的次数

那么最外层的循环就要到2*n-1次了，其中n-1次是用来求最短路的，还有n次是次短路的

然后松弛的条件就要改变了，有四种情况

1.比最短路短2.等于最短路3.长与最短路但短于次短路4.等于次短路

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#define MAXN 1005
#define MAXM 500005
#define INF 1000000000
using namespace std;
struct node
{
int v, next, w;
}edge[MAXM];
int d[MAXN][2], e, n, m;
int cnt[MAXN][2];
int head[MAXN];
bool vis[MAXN][2];
void init()
{
e = 0;
memset(head, -1, sizeof(head));
}
void insert(int x, int y, int w)
{
edge[e].v = y;
edge[e].w = w;
edge[e].next = head[x];
head[x] = e++;
}
int dijkstra(int s, int t)
{
int flag, u;
memset(vis, 0, sizeof(vis));
memset(cnt, 0, sizeof(cnt));
for(int i = 1; i <= n; i++)
d[i][0] = d[i][1] = INF;
cnt[s][0] = 1;
d[s][0] = 0;
for(int i = 1; i < 2 * n; i++)
{
int mini = INF;
for(int j = 1; j <= n; j++)
{
if(!vis[j][0] && d[j][0] < mini)
{
u = j;
flag = 0;
mini = d[j][0];
}
else if(!vis[j][1] && d[j][1] < mini)
{
u = j;
flag = 1;
mini = d[j][1];
}
}
if(mini == INF) break;
vis[u][flag] = 1;
for(int j = head[u] ; j != -1; j = edge[j].next)
{
int w = edge[j].w;
int v = edge[j].v;
if(d[v][0] > mini + w)
{
d[v][1] = d[v][0];
cnt[v][1] = cnt[v][0];
d[v][0] = mini + w;
cnt[v][0] = cnt[u][flag];
}
else if(d[v][0] == mini + w) cnt[v][0] += cnt[u][flag];
else if(d[v][1] > mini + w)
{
d[v][1] = mini + w;
cnt[v][1] = cnt[u][flag];
}
else if(d[v][1] == mini + w) cnt[v][1] += cnt[u][flag];
}
}
int ans = 0;
if(d[t][1] == d[t][0] + 1) ans = cnt[t][1] + cnt[t][0];
else ans = cnt[t][0];
return ans;
}
int main()
{
int s, t, T, x, y, w;
scanf("%d", &T);
while(T--)
{
init();
scanf("%d%d", &n, &m);
for(int i = 1; i <= m; i++)
{
scanf("%d%d%d", &x, &y, &w);
insert(x, y, w);
}
scanf("%d%d", &s, &t);
printf("%d\n", dijkstra(s, t));
}
return 0;
}