P1144 最短路计数

题目链接
题意:就是在无边权无向图里面求从1点出发到其他点的最短距离有多少种走法(注意要对给定数字取模),有可能有重边,注意自环。
代码如下:

#include 
using namespace std;
#define inf 0x3f3f3f3f
#define Maxn 1000005
#define Maxm 2000005
int n, m, cnt, head[Maxn], dis[Maxn], num[Maxn];
bool vis[Maxn];
inline int read() {
    int s = 0, w = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {
        if (ch == '-')
            w = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9')
        s = s * 10 + ch - '0', ch = getchar();
    return s * w;
}
struct edge {
    int to, dis, next;
} e[Maxm];
void add(int u, int v, int w) {
    cnt++;
    e[cnt].to = v;
    e[cnt].dis = w;
    e[cnt].next = head[u];
    head[u] = cnt;
}
void spfa() {
    memset(dis, inf, sizeof(dis));
    queue<int> q;
    dis[1] = 1, vis[1] = 1, num[1]++;
    q.push(1);
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        vis[u] = 0;
        for (int i = head[u]; i; i = e[i].next) {
            int v = e[i].to, w = e[i].dis;
            if (dis[v] > dis[u] + w) {
                dis[v] = dis[u] + w;
                num[v] = num[u];
                if (!vis[v]) {
                    q.push(v);
                    vis[v] = 1;
                }
            } else if (dis[v] == dis[u] + w) {
                num[v] += num[u];
                num[v] %= 100003;
            }
        }
    }
}
int main() {
    cin >> n >> m;
    for (int i = 0, u, v; i < m; ++i) {
        u = read(), v = read();
        add(u, v, 1);
        add(v, u, 1);
    }
    spfa();
    for (int i = 1; i <= n; ++i) {
        cout << num[i] << endl;
    }
    return 0;
}

题解:就是找最短路径顺便记一下数字,主要问题出在计数的方面(因为这里我debug了好久……)。简单来说就是A点如果能去到B点,那么B点的最短路径数量应该加上A点本身的最短路径数量。

你可能感兴趣的:(c++)