HDOJ 6026 最短路

链接：

http://acm.hdu.edu.cn/showproblem.php?pid=6026

题意：

给你一个无向图，要求你删边删成一棵树，要求从0到每个节点的路径都是原图中的最短路

题解：

最短路，先建一个只包含最短路的有向无环图，每一个点选择任意一条入边即可生成一个树形图，那么树的种类就等于每个点的入度乘积

代码：

int n;
int G[60][60];
int d[60];
ll a[60];

void dijkstra(int s) {
    priority_queue<PII, vector<PII>, greater<PII> > que;
    memset(d, 0x3f, sizeof(d));
    d[s] = 0;
    que.push(mp(0, s));
    while (!que.empty()) {
        PII p = que.top(); que.pop();
        int v = p.second;
        if (d[v] < p.first) continue;
        rep(i, 0, n) if (G[v][i]) {
            if (d[i] > d[v] + G[v][i]) {
                d[i] = d[v] + G[v][i];
                que.push(mp(d[i], i));
                a[i] = 1;
            }
            else if (d[i] == d[v] + G[v][i]) a[i]++;
        }
    }
}

int main() {
    while (cin >> n) {
        memset(a, 0, sizeof(a));
        rep(i, 0, n) rep(j, 0, n) scanf("%1d", &G[i][j]);
        dijkstra(0);
        ll ans = 1;
        rep(i, 1, n) ans = (ans*a[i]) % MOD;
        cout << ans << endl;
    }
    return 0;
}