/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 328.0 KiB
#2 Accepted 1ms 764.0 KiB
#3 Wrong Answer 1ms 764.0 KiB
#4 Wrong Answer 3ms 532.0 KiB
#5 Wrong Answer 36ms 4.734 MiB
#6 Wrong Answer 74ms 4.812 MiB
#7 Wrong Answer 52ms 4.688 MiB
#8 Wrong Answer 29ms 536.0 KiB
#9 Wrong Answer 64ms 1.016 MiB
#10 Wrong Answer 98ms 4.684 MiB
#11 Wrong Answer 75ms 5.172 MiB
#12 Wrong Answer 76ms 5.195 MiB
#13 Wrong Answer 39ms 5.348 MiB
#14 Wrong Answer 112ms 5.156 MiB
#15 Wrong Answer 40ms 1.25 MiB
#16 Wrong Answer 32ms 5.023 MiB
#17 Wrong Answer 25ms 4.746 MiB
#18 Wrong Answer 24ms 4.812 MiB
#19 Accepted 170ms 30.676 MiB
#20 Accepted 147ms 30.66 MiB
#21 Accepted 125ms 29.418 MiB
#22 Accepted 99ms 37.289 MiB
#23 Wrong Answer 175ms 29.441 MiB
#24 Wrong Answer 188ms 30.223 MiB
#25 Wrong Answer 105ms 30.18 MiB
#26 Accepted 7ms 2.023 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;
    cin >> n;
    vector<ll> val(n + 1);
    for (int i = 1; i <= n; i++) {
        ll x; cin >> x;
        val[i] = x;
    }
    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);
    }
    
    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();
        g[u].clear();
        g[v].erase(u);
        // cout << u << " ";
        pq.pop();
        if(g[v].size() == 1 && val[v] == 0) {
            pq.push(v);
        }
    }
    ll ans = 0;
    auto dfs = [&](int u, int par, auto&& self) -> void {
        ll c = 0;
        for(auto &v: g[u]) {
            if(v == par) continue;
            ++c;
            // cout << u << " " << v << endl;
            self(v, u, self);
        }
        // cout << u << " => " << c << endl;
        ans += max(0LL, (c * 2LL) - 1);
    };
    for(int i = 1; i <= n; i++) {
        if(g[i].size() == 1) {
            ans = 0;
            dfs(i, -1, dfs);
            break;
        }
        // cout << i << " = ";
        // coutall(g[i]);
    }
    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 16:47:42
Judged At
2024-10-03 13:25:55
Judged By
Score
34
Total Time
188ms
Peak Memory
37.289 MiB