oj.uz

Submission #830097

# Submission time Handle Problem Language Result Execution time Memory
830097 2023-08-18T18:48:31 Z t6twotwo Mousetrap (CEOI17_mousetrap) C++17
25 / 100
 776 ms 88200 KB 
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N, t, m;
    cin >> N >> t >> m;
    t--, m--;
    vector<vector<int>> adj(N);
    for (int i = 0; i < N - 1; i++) {
        int x, y;
        cin >> x >> y;
        x--, y--;
        adj[x].push_back(y);
        adj[y].push_back(x);
    }
    vector<int> p(N, -1);
    queue<int> q;
    p[m] = m;
    q.push(m);
    while (!q.empty()) {
        int x = q.front();
        q.pop();
        for (int y : adj[x]) {
            if (p[y] == -1) {
                p[y] = x;
                q.push(y);
            }
        }
    }
    vector<int> path{t};
    while (path.back() != m) {
        path.push_back(p[path.back()]);
    }
    vector<bool> in(N);
    for (int x : path) {
        in[x] = 1;
    }
    vector<int> f(N);
    auto dfs = [&](auto dfs, int x, int p) -> void {
        vector<int> s;
        for (int y : adj[x]) {
            if (y == p || in[y]) {
                continue;
            }
            dfs(dfs, y, x);
            s.push_back(f[y]);
        }
        sort(s.rbegin(), s.rend());
        if (s.empty()) {
            return;
        }
        if (s.size() == 1) {
            f[x] = 1;
            return;
        }
        f[x] = s[1] + s.size();
    };
    vector<vector<int>> ch(N);
    for (int x : path) {
        for (int y : adj[x]) {
            if (!in[y]) {
                ch[x].push_back(y);
            }
        }
        dfs(dfs, x, -1);
    }
    int M = path.size();
    vector<int> z(M + 1);
    for (int i = 0; i < M; i++) {
        z[i + 1] = z[i] + ch[path[i]].size();
    }
    vector<int> dp(M, 1E9);
    dp[1] = f[path[1]];
    for (int i = 2; i < M; i++) {
        vector<int> s;
        for (int j : ch[path[i]]) {
            s.push_back(f[j]);
        }
        sort(s.rbegin(), s.rend());
        if (s.empty()) {
            dp[i] = max(0, dp[i - 1] - 1);
        } else {
            dp[i] = max({0, dp[i - 1] - 1, s[0] + z[i + 1]});
            if (s.size() == 1) {
                dp[i] = min(dp[i], dp[i - 1] + 1);
            } else {
                dp[i] = min(dp[i], max(dp[i - 1] + 1, s[1] + z[i + 1]));
            }
        }
    }
    cout << dp[M - 1] << "\n";
    return 6/22;
}

Subtask #1
0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 212 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 280 ms 88200 KB Output is correct
2 Correct 242 ms 79580 KB Output is correct
3 Correct 776 ms 88040 KB Output is correct
4 Correct 307 ms 44108 KB Output is correct
5 Correct 726 ms 88080 KB Output is correct
6 Correct 768 ms 88132 KB Output is correct

Subtask #3
0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 212 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 35.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 212 KB Output isn't correct
2 Halted 0 ms 0 KB -