首页 > 其他 > 详细

P1144 最短路计数 (bfs)

时间:2020-01-31 15:33:42      阅读:94      评论:0      收藏:0      [点我收藏+]

题目链接

技术分享图片

 

 方法1:

  bfs

Code:

#include <bits/stdc++.h>
# define LL long long
using namespace std;

const int maxn=1000000+10;
int N,M;
vector<int> adj[maxn];
int res[maxn];
int dis[maxn];
const int mod=100003;

void bfs(){
    memset(dis,-1,sizeof(dis));
    queue<int> q;
    q.push(1);
    res[1]=1;
    dis[1]=0;
    while(!q.empty()){
        int cur=q.front();
        q.pop();
        for(int i=0;i<adj[cur].size();++i){
            int v=adj[cur][i];
            if(dis[v]==-1){
                dis[v]=1+dis[cur];
                q.push(v);
            }
            if(dis[v]==1+dis[cur]){
                res[v]=(res[v]+res[cur])%mod;
            }
        }
    }
}

int main(){
    scanf("%d %d", &N, &M);
    for(int i=1;i<=M;++i){
        int a,b;
        scanf("%d %d", &a, &b);
        adj[a].push_back(b);
        adj[b].push_back(a);
    }

    bfs();
    for(int i=1;i<=N;++i){
        printf("%d\n", res[i]);
    }
    return 0;
}

方法2:

  SPFA

Code:

#include <bits/stdc++.h>
# define LL long long
using namespace std;

const int maxn=1000000+10;
int N,M;
vector<int> adj[maxn];
int res[maxn];
int dis[maxn];
const int mod=100003;

void spfa(){
    memset(dis,127,sizeof(dis));
    dis[1]=0;
    res[1]=1;
    queue<int> q;
    int inqueue[maxn];
    q.push(1);
    inqueue[1]=1;
    while(!q.empty()){
        int cur=q.front();
        q.pop();
        inqueue[cur]=0;
        for(int i=0;i<adj[cur].size();++i){
            int v=adj[cur][i];
            if(dis[cur]+1<dis[v]){
                res[v]=res[cur];
                dis[v]=1+dis[cur];
                if(inqueue[v]==0){
                    inqueue[v]=1;
                    q.push(v);
                }
            }else if(dis[cur]+1==dis[v]){
                res[v]=(res[v]+res[cur])%mod;
            }
        }
    }
}

int main(){
    scanf("%d %d", &N, &M);
    for(int i=1;i<=M;++i){
        int a,b;
        scanf("%d %d", &a, &b);
        adj[a].push_back(b);
        adj[b].push_back(a);
    }

    spfa();
    for(int i=1;i<=N;++i){
        printf("%d\n", res[i]);
    }
    return 0;
}

 

P1144 最短路计数 (bfs)

原文:https://www.cnblogs.com/FEIIEF/p/12245429.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!