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