洛谷-1144 最短路计数

题目描述
给出一个N个顶点M条边的无向无权图，顶点编号为1−N。问从顶点1开始，到其他每个点的最短路有几条。
输入输出格式
输入格式：
第一行包含2个正整数N,M，为图的顶点数与边数。
接下来M行，每行2个正整数x,y，表示有一条顶点x连向顶点y的边，请注意可能有自环与重边。
输出格式：
共NNN行，每行一个非负整数，第iii行输出从顶点1到顶点i有多少条不同的最短路，由于答案有可能会很大，你只需要输出ansmod100003 后的结果即可。如果无法到达顶点i则输出0。

输入输出样例
输入样例#1：
5 7
1 2
1 3
2 4
3 4
2 3
4 5
4 5

输出样例#1：
1
1
1
2
4

说明
1到5的最短路有4条，分别为2条1−2−4−5和1−3−4−5由于4−5的边有2条）。
对于100%的数据，N<=1000000,M<=2000000

解释：首先跑出最短路来，那么dp[i]为从1到i最短路的条数，那么dp[i]+=dp[j],其中j为dist[j]+1=dist[i]的点，这样转移就好了

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<cstdlib>
#define INF 1000000009
#define N 1000003
#define mod 100003
using namespace std;
int head[N]={0};
int next1[4*N]={0};
int V[4*N]={0};
bool vis[N]={0};
int tot=0;
int dp[N]={0};
int dist[N]={0};
int n=0,m=0;
void add(int x,int y){
    tot++;
    next1[tot]=head[x];
    V[tot]=y;
    head[x]=tot;
}
void ok(int x){
    fill(dist+1,dist+1+n,INF);
    dist[x]=0;
    priority_queue<pair<int,int> >q;
    q.push(make_pair(0,x));
    while(q.size()){
        pair<int,int> temp=q.top();q.pop();
        if(vis[temp.second]) continue;
        vis[temp.second]=1;
        for(int i=head[temp.second];i;i=next1[i]){
            int to=V[i];
            if(dist[to]>dist[temp.second]+1){
                dist[to]=dist[temp.second]+1;
                q.push(make_pair(-dist[to],to));
            }
        }
    }
}
void DP(int x){
    if(vis[x]) return;
    if(x==1){
        dp[x]=1;
        vis[x]=1;
        return;
    }
    for(int i=head[x];i;i=next1[i]){
        int to=V[i];
        if(to&&dist[x]==dist[to]+1){
            DP(to);
            dp[x]+=dp[to];
            dp[x]%=mod;
        }
    }
    vis[x]=1;
    return;
}
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        int a,b;cin>>a>>b;
        add(a,b);add(b,a);
    }
    ok(1);memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++) DP(i);
    for(int i=1;i<=n;i++) cout<<dp[i]<<endl;
    return 0;
}