PAT1003 Emergency (25 分)

1.用fill函数将一个区间的元素都赋同一个值，在<algorithm>里：

    fill(d, d+MAX, INF);//一维数组d
    fill(G[0], G[0]+MAX*MAX, INF);//二维数组G

memset函数在<string.h>里，只能用于置0或者-1。

2.细小的错误耗费了大量的时间，写的时候就该认真。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MAX 505
#define INF 0x3fffffff

int G[MAX][MAX];
int visited[MAX] = {false};
int d[MAX];
int wCollected[MAX];
int weight[MAX];
int N, M, C1, C2;
int num[MAX];


void Dijkstra(int s){
//初始化起点的参数：距起点距离、收集到的资源数、路径条数。需要初始化的变量都该初始化，别漏
    d[s] = 0;
    wCollected[s] = weight[s];
    num[s] = 1;//1不该漏
//    visited[s] = true;//2不该在这里置为已访问，否则结果就是0 0
    
    for (int i=0; i<N; i++) {//3 对于有n个顶点的图，Dijkstra算法里主体要执行n次。for循环不能漏。
        //找d最小的u
        int u = -1, minD = INF;
        for (int i=0; i<N; i++) {
            if (visited[i] == false && d[i] < minD) {
                u = i;
                minD = d[i];
            }
        }
        
        if (u == -1) {
            return;
        }
        
        //置u为已访问
        visited[u] = true;//4 u错写成s。。复制的时候要看一看有没有需要改动的啊
        for (int v=0; v<N; v++) {
            if (visited[v] == false) {
                if (d[u] + G[u][v] < d[v]) {
                    num[v] = num[u];
                    d[v] = d[u] + G[u][v];
                    wCollected[v] = wCollected[u] + weight[v];
                }else if(d[u] + G[u][v] == d[v]){
                    if (wCollected[v] < wCollected[u] + weight[v]) {
                        wCollected[v] = wCollected[u] + weight[v];
                    }
                    num[v] += num[u];
                }
            }
        }
    }
}

int main(){
    fill(d, d+MAX, INF);
    fill(G[0], G[0]+MAX*MAX, INF);
    scanf("%d %d %d %d", &N, &M, &C1, &C2);
    for (int i=0; i<N; i++) {
        scanf("%d", &weight[i]);
    }
    for (int i=0; i<M; i++) {
        int c1, c2, L;
        scanf("%d %d %d", &c1, &c2, &L);
        G[c1][c2] = G[c2][c1] = L;
    }
    
    Dijkstra(C1);
    cout<<num[C2]<<" "<<wCollected[C2]<<endl;
    return 0;
}



//Sample Input:
//
//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
//Sample Output:
//
//2 4