/ SeriousOJ /

Record Detail

Time Exceeded


  
# Status Time Cost Memory Cost
#1 Accepted 2ms 540.0 KiB
#2 Accepted 2ms 492.0 KiB
#3 Time Exceeded ≥1039ms ≥572.0 KiB
#4 Time Exceeded ≥1052ms ≥492.0 KiB
#5 Time Exceeded ≥1021ms ≥3.922 MiB
#6 Time Exceeded ≥1042ms ≥3.859 MiB
#7 Accepted 263ms 5.621 MiB
#8 Time Exceeded ≥1044ms ≥572.0 KiB
#9 Time Exceeded ≥1044ms ≥584.0 KiB
#10 Time Exceeded ≥1055ms ≥2.844 MiB
#11 Time Exceeded ≥1054ms ≥4.512 MiB
#12 Time Exceeded ≥1039ms ≥2.395 MiB
#13 Time Exceeded ≥1055ms ≥5.371 MiB
#14 Time Exceeded ≥1034ms ≥3.91 MiB
#15 Time Exceeded ≥1098ms ≥484.0 KiB
#16 Time Exceeded ≥1077ms ≥4.801 MiB
#17 Accepted 96ms 5.422 MiB
#18 Accepted 82ms 5.391 MiB
#19 Time Exceeded ≥1017ms ≥30.605 MiB
#20 Time Exceeded ≥1083ms ≥30.711 MiB
#21 Accepted 545ms 37.266 MiB
#22 Accepted 180ms 39.914 MiB
#23 Time Exceeded ≥1026ms ≥34.246 MiB
#24 Time Exceeded ≥1058ms ≥30.086 MiB
#25 Time Exceeded ≥1055ms ≥30.086 MiB
#26 Time Exceeded ≥1093ms ≥1.973 MiB

Code

#include<bits/stdc++.h>
#define endl        '\n'
#define F           first
#define S           second
#define pb          push_back
#define yes         cout<<"YES\n"
#define no          cout<<"NO\n"
#define all(x)      x.begin(),x.end()
#define allr(x)     x.rbegin(),x.rend()
#define error1(x)   cerr<<#x<<" = "<<(x)<<endl
#define error2(a,b) cerr<<"("<<#a<<", "<<#b<<") = ("<<(a)<<", "<<(b)<<")\n";
#define coutall(v)  for(auto &it: v) cout << it << " "; cout << endl;
using namespace std;
using ll = long long;
using ld = long double;

void solve() {
    ll n, cnt = 0;
    cin >> n;
    vector<ll> val(n + 1);
    for (int i = 1; i <= n; i++) {
        ll x; cin >> x;
        val[i] = x;
        if(x) ++cnt;
    }
    vector<set<ll>> g(n + 1);
    for(int i = 0; i + 1 < n; i++) {
        ll u, v; cin >> u >> v;
        g[u].insert(v);
        g[v].insert(u);
    }
    if(cnt == 0) {
        cout << 0 << endl;
        return;
    }

    priority_queue<ll> pq;
    for(int i = 1; i <= n; i++) {
        if(g[i].size() == 1 && val[i] == 0) pq.push(i);
    }
    while(pq.size()) {
        ll u = pq.top();
        ll v = *g[u].begin();
        // cout << u << " <= ";
        pq.pop();
        if(val[v] == 0) {
            pq.push(v);
        }
        g[u].clear();
        g[v].erase(u);
    }

    vector<ll> depth(n + 1, 0);
    auto dfs2 = [&](int u, int par, int d, auto&& self) -> void {
        depth[u] = d;
        for(auto &i: g[u]) {
            if(i == par) continue;
            self(i, u, d + 1, self);
        }
    };
    ll root = 1;
    priority_queue<pair<ll, ll>> pp;
    for(int i = 1; i <= n; i++) {
        if(g[i].size() == 1) {
            dfs2(i, -1, 0, dfs2);
            break;
        }
    }
    for(int i = 1; i <= n; i++) {
        if(g[i].size() == 1) pp.push({depth[i], i});
    }

    ll ans = 0;
    auto dfs = [&](int u, int par, int d, auto&& self) -> ll {
        ll mxD = d;
        for(auto &v: g[u]) {
            if(v == par) continue;
            // cout << u << " " << v << endl;
            ++ans;
            mxD = max(mxD, self(v, u, d + 1, self));
            ++ans;
        }
        return mxD;
        // cout << u << " => " << c << endl;
    };
    ll Ans = LLONG_MAX, T = 20;
    while(T-- && pp.size()) {
        ans = 0;
        ans -= dfs(pp.top().S, -1, 0, dfs);
        Ans = min(Ans, ans);
        // cout << pp.top().S << " = " << ans << endl;
        pp.pop();
    }
    if(Ans == LLONG_MAX) Ans = 0;
    cout << Ans << endl;
    return;
}

signed main() {
    ios::sync_with_stdio(false); cin.tie(0);
    int tc = 1;
    cin >> tc;
    for (int t = 1; t <= tc; t++) {
        // cout << "Case " << t << ": ";
        solve();
    }
    return 0;
}

Information

Submit By
Type
Submission
Problem
P1078 Apple on Tree
Contest
Bangladesh 2.0
Language
C++20 (G++ 13.2.0)
Submit At
2024-08-16 17:35:08
Judged At
2024-10-03 13:21:24
Judged By
Score
25
Total Time
≥1098ms
Peak Memory
≥39.914 MiB