Submission #1212431

#	Time	Username	Problem	Language	Result	Execution time	Memory
1212431		yanb	Petrol stations (CEOI24_stations)	C++20	18 / 100	3594 ms	11744 KiB

Main

#include <bits/stdc++.h>
 
#define int long long
 
using namespace std;

using pii = pair<int, int>;

int n, k;
vector<vector<pii>> g;
vector<int> ans;

int dfs(int v, int p, int f) {
    int sz = 1;
    for (auto [u, vu] : g[v]) {
        if (u != p) {
            int csz;
            if (f >= vu) {
                csz = dfs(u, v, f - vu);
            } else {
                csz = dfs(u, v, k - vu);
                ans[v] += csz;
            }
            sz += csz;
        }
    }
    return sz;
}

signed main() {
    ios_base::sync_with_stdio(0); cin.tie(0);
    
    cin >> n >> k;
    g.resize(n);
    ans.resize(n);
    for (int i = 0; i < n - 1; i++) {
        int x, y, w;
        cin >> x >> y >> w;
        g[x].emplace_back(y, w);
        g[y].emplace_back(x, w);
    }

    for (int i = 0; i < n; i++) {
        dfs(i, i, k);
    }

    for (int i = 0; i < n; i++) {
        cout << ans[i] << "\n";
    }
}

• Subtask 0: 0 / 0
• Subtask 1: 18 / 18
• Subtask 2: 0 / 8
• Subtask 3: 0 / 10
• Subtask 4: 0 / 12
• Subtask 5: 0 / 17
• Subtask 6: 0 / 35