AtCoderD - Number of Shortest paths BFS

Posted 2021-10-30 karshey

D
参考

想要不T，就要减少重复加进队列的元素。
也就是说，如果一个元素在同一层的遍历里加进去过队列，我们就不必再把它加进去（否则T三个点），只需要更新它的ans值即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 
#define fir(i,a,n) for(int i=a;i<=n;i++)
//======================
const int N=2e5+10;
const int MOD=1e9+7;
int n,m;
vector<int>g[N];
int ans[N];
int dis[N];
int st[N];
queue<int>q;
void bfs()
{
	mem(dis,-1);
	q.push(1);
	dis[1]=1;
	ans[1]=1;
//	st[1]=1;
	while(q.size())
	{
		int t=q.front();
		q.pop();
	
		
			for(int i=0;i<g[t].size();i++)
			{
				int tt=g[t][i];
				if(dis[tt]==-1)//没走过
				{
					q.push(tt);
					dis[tt]=dis[t]+1;
					ans[tt]=(ans[tt]+ans[t])%MOD;
				} 
				else if(dis[tt]==dis[t]+1) //走过但是也是最短
				{
					ans[tt]=(ans[tt]+ans[t])%MOD;
					//答案加上，但是不重新放回队列 
				} 
			}
		}
	
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b;scanf("%d%d",&a,&b);
	//	if(a>b) swap(a,b);//要a小 
		g[a].pb(b);
		g[b].pb(a);
	}
	
	for(int i=1;i<=n;i++) sort(g[i].begin(),g[i].end());
	
	bfs();
	cout<<ans[n];
	return 0; 
}