백준 25516번 거리가 k이하인 트리 노드에서 사과 수확하기 C++

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

25516번: 거리가 k이하인 트리 노드에서 사과 수확하기

 

dfs

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, k;
int ans = 0;

vector<int> v;
vector<vector<int>> tree;

void dfs(int curr, int dist) {
    if (dist > k) return;
    ans += v[curr];
    for (auto next : tree[curr]) {
        dfs(next, dist + 1);
    }
}

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

   
    cin >> n >> k;
    v = vector<int>(n);
    tree = vector<vector<int>>(n, vector<int>());
    
    for (int i = 0; i < n - 1; i++) {
        int s, e;
        cin >> s >> e;
        tree[s].push_back(e);
    }
    for (int i = 0; i < n; i++) {
        int apple;
        cin >> apple;
        v[i] = apple;
    }
    dfs(0, 0);
    cout << ans;
}