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

1240번: 노드사이의 거리

dfs를 이용한 문제이다. 트리의 경우 노드 사이의 연결 방향이 양방향이므로 연결된 노드와 거리를 입력받을 때 양방향으로 저장해주었다. 그리고 M만큼 반복문을 돌며 dfs를 이용해 연결된 노드들을 따라 거리를 모두 더해 저장해준 뒤 출력해주었다. 쉽게 풀 수 있었던 문제였다.

#include <iostream>
#include <vector>
#include <cstring>

using namespace std;

int N, M;
vector<pair<int, int>> n[1001];
vector<pair<int, int>> m;
bool check[1001];
vector<int> result;

void dfs(int now, int end, int sum) {
    if (now == end) {
        result.push_back(sum);
        return;
    }

    for (int i = 0; i < n[now].size(); i++) {
        int next = n[now][i].first;
        int ns = sum + n[now][i].second;

        if (check[next]) continue;

        check[next] = true;
        dfs(next, end, ns);
    }
}

void solution() {
    for (int i = 0; i < M; i++) {
        memset(check, false, sizeof(check));

        int start = m[i].first;
        int end = m[i].second;

        check[start] = true;
        dfs(start, end, 0);
    }

    for (int i = 0; i < M; i++) {
        cout << result[i] << endl;
    }
}

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

    cin >> N >> M;

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

    for (int i = 0; i < M; i++) {
        cin >> a >> b;
        m.push_back({ a,b });
    }

    solution();

    return 0;
}