[백준/BOJ] 백준 14554번 : The Other Way

https://www.acmicpc.net/problem/14554

14554번: The Other Way

vector <int> short_cnt(100001, 0)에 [위치] = 해당 위치로 최단경로로 올 수 있는 개수를 저장하여, 해당 위치가 지금까지 구한 최단경로와 같은 비용의 길일 때 지금 경로에서 오는 경로의 개수를 추가하여 문제를 해결했다.

 

코드

#include <iostream>
#include <algorithm>
#include <vector>
#include <utility>
#include <queue>
#include <limits>
using namespace std;

int n, m, s, e;
vector<pair<int, int>> adj[100001];
vector<int> short_cnt(100001, 0); //[위치] = 해당 위치로 최단경로로 올 수 있는 개수

int Solve(int start, int dest)
{
	vector<long long> short_dest(100001, numeric_limits<long long>::max());
	priority_queue<pair<long long, int>> pq;

	pq.push(make_pair(-0, start));
	short_dest[start] = 0;
	short_cnt[start] = 1;

	while (!pq.empty())
	{
		int here = pq.top().second;
		long long here_cost = -pq.top().first;
		pq.pop();

		if (short_dest[here] < here_cost)
			continue;

		if (here == dest)
			continue;

		for (int i = 0; i < adj[here].size(); i++)
		{
			int there = adj[here][i].second;
			long long there_cost = here_cost + (long long)adj[here][i].first;

			//지금까지 구한 최단경로보다 더 작은 비용의 길을 발견 했을때
			if (short_dest[there] > there_cost)
			{
				short_dest[there] = there_cost;
				short_cnt[there] = short_cnt[here];

				pq.push(make_pair(-there_cost, there));
			}

			//지금까지 구한 최단경로와 같은 비용의 길일때
			else if (short_dest[there] == there_cost)
			{
				short_cnt[there] = (short_cnt[there] + short_cnt[here]) % 1000000009;
			}
		}

	}

	return short_cnt[dest];
}

int main()
{
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);

	cin >> n >> m >> s >> e;

	for (int i = 0; i < m; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;

		adj[a].push_back(make_pair(c, b));
		adj[b].push_back(make_pair(c, a));
	}

	cout << Solve(s, e);

	return 0;
}