c++ 1240번 노드사이의 거리 (백준)

#include <bits/stdc++.h>
#define Q q.front()
using namespace std;

int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	int n, m, a, b, c; //a, b, c는 입력을 위한 변수
	cin >> n >> m;
	
	vector<pair<int, int> > vec[n+1]; //간선의 정보를 담을 벡터
	queue<pair<int, int> > q; //BFS에서 사용할 큐
	
	for(int i=1; i<n; i++) { //n-1개의 간선을 입력받음
		cin >> a >> b >> c;
		vec[a].push_back({b, c});
		vec[b].push_back({a, c});
	}
	
	for(int i=0; i<m; i++) { //m개의 시작-종점을 입력받음
		int min = 1e9; //최소거리
		bool v[n+1]; //방문처리를 위한 배열
		memset(v, false, sizeof(bool)*(n+1)); //방문배열을 전부 false로 초기화
		cin >> a >> b;
		q.push({a, 0}); //큐에 시작노드와 가중치를 0 으로 넣음
        v[a] = true;
		
		while(!q.empty()) { //BFS
			int x = get<0>(Q);
			int y = get<1>(Q);
			q.pop();
			
			
			
			for(pair<int, int> hi : vec[x]) { //x와 연결된 노드들
				int tmp = hi.first; //다음 노드
				y += hi.second; //가중치에 다음 노드로 갈 때의 가중치를 더합니다.
				if(tmp == b) { //다음 노드가 도착노드라면
					min = (min<y) ? min : y; //최솟값
					continue;
				}
				if(!v[tmp]) { //방문할 필요가 없지 않는 노드라면
                	v[tmp] = true; //tmp번 노드 방문 처리 (더이상 x번 노드에 올 이유가 없어졌으므로)
					q.push({tmp, y}); //큐에 다음 노드의 번호와 가중치 + 다음노드로 갈 때의 가중치 를 넣습니다.
				}
				y -= hi.second; //다시 다음 노드로 갈 때의 가중치를 빼줍니다.
			}
		} cout << min <<'\n';
	}
}

풀이 : BFS

 

각 노드에 대한 가변배열을 만든다음, 연결되는 노드들을 가중치와 함께 가변배열에 집어 넣습니다.

 

ex) 1 4 23

vec[1]에 {4, 23} 이런식으로 넣고

vec[4]에 {1, 23} 이런식으로 넣습니다.

 

그런 다음 m번 만큼 특정 두 노드의 최소거리를 BFS을 이용해 탐색합니다.

때문입니다.

그러므로 방문확인을, 큐에 넣자마자 해서는 안됩니다.

 

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

1240번: 노드사이의 거리