P1144 最短路计数

P1144 最短路计数

题目描述

给出一个N个顶点M条边的无向无权图，顶点编号为1-N。问从顶点11开始，到其他每个点的最短路有几条。

输入输出格式

输入格式：

第一行包含2个正整数N,M为图的顶点数与边数。

接下来M行，每行22个正整数x,y,表示有一条顶点x连向顶点y的边，请注意可能有自环与重边。

输出格式：

共N行，每行一个非负整数，第ii行输出从顶点11到顶点ii有多少条不同的最短路，由于答案有可能会很大，你只需要输出ans \bmod 100003ansmod100003后的结果即可。如果无法到达顶点ii则输出0。

 

　　思路：

　  跑最短路的同时维护一下答案就行，如果d[y]=d[x]+1,就把ans[y]+=ans[x],否则在更新d[y]时令ans[y]=ans[x];

　  代码：

　   

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
inline int read()
{
    char ch=getchar(); 
    int x=0,f=1;
    while((ch>'9'||ch<'0')&&ch!='-')
        ch=getchar();
    if(ch=='-')
    {
        f=-1;
        ch=getchar();
    }
    while('0'<=ch&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int mod=100003;
int n,m,x,y,tot=0;
const int N=1000005,M=4000005;
int head[N],to[M],nxt[M],d[N],ans[N];
bool p[N];
queue< int > q;
void add(int x,int y)
{
    to[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=m;i++)
    {
        x=read();y=read();
        add(x,y);
        add(y,x);
    }
    for(int i=1;i<=n;i++)
    {
        d[i]=1e9;p[i]=0;
    }
    d[1]=0;
    p[1]=1;
    ans[1]=1;
    q.push(1);
    while(q.size())
    {
        x=q.front();q.pop();
        p[x]=0;
        for(int i=head[x];i;i=nxt[i])
        {
            y=to[i];
            if(d[y]>d[x]+1)
            {
                d[y]=d[x]+1;
                ans[y]=ans[x];
                if(!p[y])
                {
                    q.push(y);
                    p[y]=1;
                }
            }
            else if(d[y]==d[x]+1)
            {
                ans[y]+=ans[x];
                ans[y]%=mod;
            }
        }
    }
    for(int i=1;i<=n;i++)
        printf("%d\n",ans[i]);
    return 0;   
}