티스토리 뷰

PS

[백준] 1240번 노드사이의 거리 C++

Eastplanet 2022. 6. 24. 13:53

문제 출처: https://www.acmicpc.net/problem/1240

 

1240번: 노드사이의 거리

N(2≤N≤1,000)개의 노드로 이루어진 트리가 주어지고 M(M≤1,000)개의 두 노드 쌍을 입력받을 때 두 노드 사이의 거리를 출력하라.

www.acmicpc.net

N개의 노드로 이루어진 트리가 주어진다.

트리로 이루어져 있기 때문에 사이클이 없고 노드와 노드 사이의 길이 하나만 존재함을 알 수 있다.

따라서 거리를 알고 싶은 노드 A에서 BFS를 시작하여 B를 만나는 지점에서 그 거리를 갱신하고 BFS를 종료하여 해결함.

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include <string>

using namespace std;

typedef pair<int, int>P;

vector<P> adj[1001];
int visited[1001];

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

    int N;
    int M;
    cin >> N >> M;
    for (int i = 0; i < N-1; i++) {
        int a, b, d;
        cin >> a >> b >> d;
        adj[a].push_back(P(b, d));
        adj[b].push_back(P(a, d));
    }

    for (int i = 0; i < M; i++) {
        fill(&visited[0], &visited[1001], 0);
        queue<P>q;
        int start, end;
        cin >> start >> end;
        q.push(P(start,0));
        visited[start] = 1;

        int min = 123456789;

        while (!q.empty()) {
            P curr = q.front(); q.pop();
            if (curr.first == end) {
                min = curr.second;
                break;
            }

            for (auto next : adj[curr.first]) {
                if (visited[next.first] == 0) {
                    visited[next.first] = 1;
                    q.push(P(next.first, curr.second + next.second));
                }
            }
        }
        cout << min << '\n';
    }

}

'PS' 카테고리의 다른 글

[백준] 1535번 안녕 C++  (0) 2022.06.24
[백준] 2529번 부등호 C++  (0) 2022.06.24
[백준] 2294번 동전2 C++  (0) 2022.06.23
[백준] 14606번 피자(small)  (0) 2022.06.22
[백준] 1926번 그림 C++  (0) 2022.06.21
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2026/01   »
1 2 3
4 5 6 7 8 9 10
11 12 13 14 15 16 17
18 19 20 21 22 23 24
25 26 27 28 29 30 31
글 보관함