Submission #597011

#	Time	Username	Problem	Language	Result	Execution time	Memory
597011		OttoTheDino	Mousetrap (CEOI17_mousetrap)	C++17	0 / 100	312 ms	55008 KiB

mousetrap

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

#define rep(i,s,e)                  for (int i = s; i <= e; ++i)
#define rrep(i,s,e)                 for (int i = s; i >= e; --i)
#define pb                          push_back
#define all(a)                      a.begin(), a.end()
#define len(a)                      (int)a.size()
typedef vector<int> vi;

const int mx = 1e6;
vi adj[mx+1], road;
int n, t, m;

bool dfs (int u, int p) {
    if (u==t) return 1;
    for (int v : adj[u]) {
        if (v==p) continue;
        if (dfs (v, u)) {
            road.pb(u);
            return 1;
        }
    }
    return 0;
}

int dfs2 (int u, int p) {
    if (len(adj[u])<=2) return len(adj[u])-1;

    vi nxt;
    for (int v : adj[u]) {
        if (v==p) continue;
        nxt.pb(dfs2 (v, u));
    }

    sort(all(nxt),greater<int>());

    return nxt[1]+len(adj[u])-1;
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> t >> m;

    rep (i,1,n-1) {
        int u, v; cin >> u >> v;
        adj[u].pb(v);
        adj[v].pb(u);
    }

    dfs (m,-1);

    int l = len(road), s = 0;
    vi dp[l];

    reverse(all(road));
    road.pb(t);

    rrep (i,l-1,0) {
        s += len(adj[road[i]])-2;
        for (int v : adj[road[i]]) {
            if ((i>0 && v==road[i-1]) || v==road[i+1]) continue;
            dp[i].pb(dfs2(v,road[i])+s+1);
        }
        sort(all(dp[i]), greater<int>());
    }

    int lo = 0, hi = mx;
    if (len(dp[0])) lo = dp[0][0]; 

    while (lo<hi) {
        int mid = (lo+hi)/2, x = 0, suc = 1;
        rep (i,1,l-1) {
            x++;
            for (int j : dp[i]) if (j>mid) x--;
            suc &= (x>=0);
        }
        if (suc) hi = mid;
        else lo = mid+1;
    }

    cout << lo << "\n";

    return 0;
}

• Subtask 1: 0 / 20
• Subtask 2: 0 / 25
• Subtask 3: 0 / 20
• Subtask 4: 0 / 35