最短路个数与次短路个数问题

模板题目

首先spfa不具备拓扑序，他会重复更新。
比如这个图，spfa先沿着路线1更新到终点，终点b的最短路径为1,a的最短路径数也为1，然后spfa会沿着路线2更新，那么更新后a的最短路路径数位2，但是此时终点的最短路径数会加上a的最短路径数，那么b的最短路径数位3，显然是不对的。

dijkstra是具备拓扑序更新这个特点，因为优先队列保证了每次一个点出队后，他不会在次出队，这样就形成拓扑序。

利用一个数组d[i][1] d[i][0] d(i,1)表示i号点的次短路个数，d(i,0)表示i号点的最短路次数。

每次更新需要讨论情况一共是四种情况

#include <bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define x first
#define y second
typedef pair<int,int> PII;
const int N=1e4+10;
int n,m,ecnt,k,ans;
int vis[N][2];//是否拜访过该点
int d[N][2],head[N],f[N][2];
int s,e;//起点和终点

struct node {
	int id,type,dist;//当前点编号、类型、最短路的值

	bool operator>(const node&a)const { //小根堆初始大于号
		return dist>a.dist;
	}
};

struct edge {
	int u,v,w,next;
} E[N<<1];

void add(int u,int v,int w) {
	E[++ecnt].u=u;
	E[ecnt].v=v;
	E[ecnt].w=w;
	E[ecnt].next=head[u];
	head[u]=ecnt;
}

int dijkstra(int s) {
	memset(d,INF,sizeof d);
	priority_queue<node,vector<node>,greater<node> >q;
	d[s][0]=0,f[s][0]=1;//起点的最短路为1
	q.push({s,0,0});
	while(!q.empty()) {
		auto t=q.top();
		q.pop();
		int u=t.id,type=t.type;
		if(vis[u][type])continue;
		vis[u][type]=1;

		for(int i=head[u]; i; i=E[i].next) {
			int v=E[i].v;
			if(d[v][0]>d[u][type]+E[i].w) { //v的最短路大于u路径时
				d[v][1]=d[v][0],f[v][1]=f[v][0];//最短路变为次短路
				q.push({v,1,d[v][1]});
				d[v][0]=d[u][type]+E[i].w,f[v][0]=f[u][type];//更新最短路
				q.push({v,0,d[v][0]});
			} else if(d[v][0]==d[u][type]+E[i].w) f[v][0]+=f[u][type];//等于u的路径时
			else if(d[v][1]>d[u][type]+E[i].w) { //次短路大于当前d[u][type]+E[i].w时
				d[v][1]=d[u][type]+E[i].w,f[v][1]=f[u][type];
				q.push({v,1,d[v][1]});
			} else if(d[v][1]==d[u][type]+E[i].w)f[v][1]+=f[u][type];
		}
	}

	int res=f[e][0];

	if(d[e][0]+1==d[e][1])res+=f[e][1];//次短路和最短路距离差1

	return res;
}



int main() {
	int T;
	cin>>T;
	while(T--) {
		memset(vis,0,sizeof vis);
		memset(head,0,sizeof head);
		memset(f,0,sizeof f);
		ecnt=0;
		cin>>n>>m;
		while(m--) {
			int u,v,w;
			cin>>u>>v>>w;
			add(u,v,w);
		}
		cin>>s>>e;
		cout<<dijkstra(s)<<endl;
	}




	return 0;
}