洛谷P1144 最短路计数 图论最短路 记忆化搜索

洛谷P1144 最短路计数
图论最短路 记忆化搜索

题意 求 起点 到各个点的最短路 有几条

注意 要 %
最短路计数
首先求一遍单源最短路 可以用 SPFA 或者 堆优化 + dijkstra
然后就可以求 每个点 到 1 的最短距离
然后 记忆化搜索下去 就可以求得 到这一个点的最短路有几条

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iomanip>
#include <iostream>
#include <queue>
using namespace std ; 

const int maxn = 1000011,maxe = 2000011,inf = 1e9 ; 
struct node{
    int to,pre ; 
};
node e[2*maxe] ; 
int n,m,cnt,u,v,x,y,ans ; 
int dist[maxn],head[maxn],dp[maxn] ;
bool vis[maxn],f[maxn] ; 
queue <int> Q ;

inline void adde(int x,int y) 
{
    e[++cnt].to = y ; 
    e[cnt].pre = head[ x ] ; 
    head[ x ] = cnt ; 
}

inline void SPFA(int s,int t) 
{
    for(int i=1;i<=n;i++) dist[ i ] = inf,vis[ i ] = 0 ; 
    Q.push(s) ; 
    vis[ s ] = 1 ;
    dist[ s ] = 0 ; 
    while(!Q.empty()) 
    {
        u = Q.front() ; 
        vis[ u ] = 0 ;
        Q.pop() ;
        for(int i=head[ u ];i;i = e[ i ].pre) 
        {
            v = e[ i ].to ; 
            if( dist[ u ] + 1 < dist[ v ] ) 
            {
                dist[ v ] = dist[ u ]+1 ;
                if(!vis[ v ]) 
                {
                    vis[ v ] = 1; 
                    Q.push(v) ;  
                }
            }
        }
    } 
}

int dfs(int u)    //   记忆化搜索     
{
    int v ; 
    if(dp[ u ]) return dp[ u ] ;       // 
    for(int i=head[ u ];i;i=e[i].pre) 
    {
        v = e[ i ].to ;
        if(dist[v]+1==dist[ u ]) dp[ u ] = (dp[ u ]+dfs(v)) % 100003 ; 
    }
    return dp[ u ] ; 
}

int main() 
{
    scanf("%d%d",&n,&m) ; 
    for(int i=1;i<=m;i++) 
    {
        scanf("%d%d",&x,&y) ; 
        adde(x,y) ;  
        adde(y,x) ; 
    }
    SPFA(1,n) ;  
    dp[ 1 ] = 1 ;
    
    for(int i=1;i<=n;i++) 
        printf("%d
",dfs( i ) ) ; 
      return 0 ; 
}