PAT--Advanced Level--1003 Emergency

做题思路：

给出一幅图，图中每个结点都有一个值。从一个点到另一个点，求最短路径的数量，作为第一个输出。在求得每条最短路径同时，求每条最短路径中，路过结点的值的和最大是多少，作为第二个输出。

Dijkstra满足拓扑序，所以即使是有环图，也可以进行最短路数量的计数。

我这里用的是堆优化的Dijkstra，普通的Dijkstra也可以过。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>

#define x first
#define y second

using namespace std;

const int N = 505, M = N << 2;

typedef pair<int, int> PII;

int num[N];
bool st[N];
int g[N][N];
int dist[N];
int n, m, c1, c2;
int h[N], e[M], ne[M], w[M], idx;
int cnt1[N], cnt2[N];

void Dijkstra()
{
	priority_queue<PII, vector<PII>, greater<PII> > heap;
	memset(dist, 0x3f, sizeof dist);
	dist[c1] = 0;
	cnt1[c1] = 1;
	cnt2[c1] = num[c1];
	heap.push({0, c1});
	
	
	while(heap.size())
	{
		PII t = heap.top();
		heap.pop();
		
		int ver = t.y;
		
		if(st[ver]) continue;
		st[ver] = true;
		
		for(int i = 0; i < n; i ++ )
		{
			if(st[i]) continue;
			if(dist[i] > dist[ver] + g[ver][i])
			{
				dist[i] = dist[ver] + g[ver][i];
				cnt1[i] = cnt1[ver];
				cnt2[i] = cnt2[ver] + num[i];
				heap.push({dist[i], i});
			}
			else if(dist[i] == dist[ver] + g[ver][i])
			{
				cnt1[i] = cnt1[ver] + cnt1[i];
				cnt2[i] = max(cnt2[i], cnt2[ver] + num[i]);
			}
		}
	}
	
}

int main()
{
	memset(g, 0x3f, sizeof g);
	scanf("%d%d%d%d", &n, &m, &c1, &c2);
	for(int i = 0; i < n; i ++ )
		scanf("%d", &num[i]);
		
	memset(h, -1, sizeof h);
	while(m -- )
	{
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		g[a][b] = g[b][a] = min(g[a][b], c);
	}
	
	Dijkstra();
	
	printf("%d %d", cnt1[c2], cnt2[c2]);
	
	return 0;
}