PAT甲级 1003 Emergency (25 分) 最短路更新

题意：

给定一个包含n个点m条边的无向图，每个结点有一个权值（a[]数组保存），寻找C1到C2的最短路

问：不同的最短路的条数 和 所有最短路中权值和最大是多少；

思路：

利用dijstra算法。

首先在求最短路的时候需要设定一个d[]数组，d[i]表示从C1出发到达i点的最短距离，然后每次取出集合外最优值进行更新；

这时候我们还需要两个数组cnt[], sum[];

cnt[i] 表示从C1出发到达i点最短路条数，由t点更新i点时，①若d[i]被更新则，cnt[i] = cnt[t];②若d[i]未被更新，则cnt[i] += cnt[t];

sum[i] 表示从C1出发到达i点最短路径权值的最大值，由t点更新i点时，①若d[i]被更新则，sum[i] = sum[t];②若d[i]未被更新，则sum[i] = max(sum[i],  sum[t] + a[i]);

 

#include<bits/stdc++.h>
#define FI first
#define SE second

using namespace std;
const int maxn = 500 + 7;
typedef long long ll;
typedef pair<int,int> P;
const int INF = 0x7f7f7f7f;

int n, m, C1, C2;
int a[maxn];
int dis[maxn][maxn];
int d[maxn], cnt[maxn], sum[maxn];
bool vis[maxn];

void init() {
    for(int i = 0; i <= n; ++i) {
        for(int j = 0; j <= n; ++j) {
            dis[i][j] = INF;
        }
    }
}
void dij() {
    for(int i = 0; i < n; ++i) {
        d[i] = dis[C1][i];
    }
    d[C1] = 0;
    cnt[C1] = 1;
    sum[C1] = a[C1];
    while(1) {
        int t = -1;
        for(int i = 0; i < n; ++i) {
            if(!vis[i] && d[i] != INF && (t == -1 || d[i] < d[t])) t = i;
        }
        if(t == -1) break;
        vis[t] = 1;
        for(int i = 0; i < n; ++i) {
            if(d[i] > (d[t] + dis[t][i])) {
                d[i] = (d[t] + dis[t][i]);
                cnt[i] = cnt[t];
                sum[i] = (sum[t] + a[i]);
            }
            else if(d[i] == (d[t] + dis[t][i])) {
                cnt[i] += cnt[t];
                sum[i] = max(sum[i], (sum[t] + a[i]));
            }
        }
    }
    cout << cnt[C2] << " " << sum[C2] << endl;
}
int main() {
    scanf("%d%d%d%d", &n, &m, &C1, &C2);
    init();
    for(int i = 0; i < n; ++i) {
        scanf("%d", &a[i]);
    }
    int c1, c2, t;
    for(int i = 0; i < m; ++i) {
        scanf("%d%d%d", &c1, &c2, &t);
        dis[c1][c2] = dis[c2][c1] = t;
    }
    dij();

    return 0;
}

/*
5 6 0 2
1 2 1 5 3
0 1 1
0 2 2
0 3 1
1 2 1
2 4 1
3 4 1
*/