Number of Shortest paths 题解(at_abc211 D 最短路径dijkstra堆优化 + dp思想)

题目

Number of Shortest paths 地址：at_abc211_D

解析

• TAG: 最短路径，dp

• DESCRIPITION:
    给定一个双向边图，边长都是1，求1到n的最短路径的数量。

• INPUT:
  

• SOLUTION:
  （1）既然要求单源最短路径，很容易想到dijkstra，经过测试用了堆优化能过，直接套模板。
  这是我平时用的算法板子：最短路径一般算法总结（板子）
  （2）在此基础上，为了求最短路径数量，我们加入了dp思想: $dp[i]$表示到 $i$ 点的最短路径数量。然后在dijkstra算法的每一次更新路径长度数组dist时更新每一个点的dp值即可
  分为俩种情况：
    搜索到相同路径长度的前节点时：$dp[v]=dp[v]+dp[v_p]$ ($v_p表示前一个点$)
    搜索到更短路径长度的前节点时：$dp[v]=dp[ft];$

参考源码

#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
using namespace std;
const int MAXN = 2e5;
const int p = 1e9+7;

typedef struct edge{
	int v,w;  // 连接点编号，边权重 
	edge(int v0,int w0):v(v0),w(w0){}
	bool operator >(const edge &b)const{
		return w>b.w;
	}
}edge;

vector<edge> G[MAXN+10];  // 用edge的vector保存整个图，G[i]表示以i点为起点的边的数组 

int dist[MAXN+5];
int in[MAXN+5];
int dp[MAXN+5];
int n;



void dijkstra(int p_s){
	dp[p_s] = 1;
	for(int i=1;i<=n;i++){  //  dist,in初始化 
		dist[i] = 0x3f3f3f3f;
		in[i] = 0;
	}  
	dist[p_s] = 0;
	
	priority_queue<edge,vector<edge>,greater<edge> > Q;

	Q.push(edge(p_s,0));
	while(!Q.empty()){
		// 首先获得离集合最近的点信息 
		int ft = Q.top().v;  
		int dis = Q.top().w;
		Q.pop();
		//if(in[ft]) continue;
		// 最近点入集合并更新堆和新的当前最短路径  
		in[ft] = 1;
		for(int i=0;i<G[ft].size();i++){
			int v = G[ft][i].v;
			if(!in[v]&&dist[v]>dis+G[ft][i].w){
				dist[v] = dis+G[ft][i].w;
				Q.push(edge(v,dist[v]));
				dp[v] = dp[ft];
			}
			else if(!in[v]&&dist[v]==dis+G[ft][i].w){
				dp[v] = (dp[v] + dp[ft])%p;
			}
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	int N,M;
	cin>>N>>M;
	n=N;
	
	for(int i=0;i<M;i++){
		int v1,v2;
		cin>>v1>>v2;
		G[v1].push_back(edge(v2,1));
		G[v2].push_back(edge(v1,1));	
	}
	dijkstra(1);
	cout<<dp[N];
	return 0;
}