Submission #1302299

#TimeUsernameProblemLanguageResultExecution timeMemory
1302299amrharbRace (IOI11_race)C++20
0 / 100
0 ms332 KiB
#include <bits/stdc++.h>
using namespace std;

int best_path(int n, int k, int H[][2], int L[]) {
    vector<vector<int>> adj(n);

    for (int i = 0; i < n - 1; i++) {
        int x = H[i][0];
        int y = H[i][1];
        adj[x].push_back(y);
        adj[y].push_back(x);
    }

    vector<bool> visited(n);
    vector<int> sz(n);

    function<void(int,int)> dfs_sz = [&](int u, int p) {
        sz[u] = 1;
        for (auto v : adj[u]) {
            if (v == p || visited[v]) continue;
            dfs_sz(v, u);
            sz[u] += sz[v];
        }
    };

    vector<int> dist;

    function<void(int,int,int)> dfs = [&](int u, int p, int cur) {
        dist.push_back(cur);
        for (auto v : adj[u]) {
            if (v == p || visited[v]) continue;
            dfs(v, u, cur + 1);
        }
    };

    function<int(int,int,int)> find_centroid = [&](int u, int p, int size) {
        for (auto v : adj[u]) {
            if (v == p || visited[v]) continue;
            if (sz[v] > size / 2)
                return find_centroid(v, u, size);
        }
        return u;
    };

    vector<int> cnt(n + 1);
    int ans = 0;

    function<void(int)> decompose = [&](int u) {
        dfs_sz(u, -1);
        int centroid = find_centroid(u, -1, sz[u]);
        visited[centroid] = true;

        cnt[0] = 1;

        for (auto v : adj[centroid]) {
            if (visited[v]) continue;
            dist.clear();
            dfs(v, centroid, 1);

            for (auto d : dist) {
                if (d <= k)
                    ans += cnt[k - d];
            }
            for (auto d : dist) {
                cnt[d]++;
            }
        }

        for (int i = 0; i <= sz[u]; i++)
            cnt[i] = 0;

        for (auto v : adj[centroid]) {
            if (!visited[v])
                decompose(v);
        }
    };

    decompose(0);
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...