Submission #66561025

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/all>
using namespace atcoder;

int main(){
    ll ans = 0;
    int N; cin >> N;
    vector<ll> x(N);
    for(int i = 0; i < N; i++) cin >> x[i];
    vector<vector<pair<ll, ll>>> G(N);
    for(int i = 0; i < N - 1; i++){
        ll u, v, w; cin >> u >> v >> w;
        u--; v--;
        G[u].push_back({v, w});
        G[v].push_back({u, w});
    }
    vector<bool> vis(N, false);
    auto f = [&](int pos, auto f)->int{
        vis[pos] = true;
        ll cnt = x[pos];
        for(auto [nex, cost] : G[pos]){
            if(!vis[nex]){
                ll d = f(nex, f); 
                cnt += d;
                ans += cost * abs(d);
            }
        }
        return cnt;
    };
    f(0, f);
    cout << ans << endl;
}

Submission Info

Submission Time
Task E - Pair Annihilation
User iphi7
Language C++ 23 (gcc 12.2)
Score 425
Code Size 847 Byte
Status AC
Exec Time 102 ms
Memory 21876 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 425 / 425
Status
AC × 3
AC × 27
Set Name Test Cases
Sample 00_sample_01.txt, 00_sample_02.txt, 00_sample_03.txt
All 00_sample_01.txt, 00_sample_02.txt, 00_sample_03.txt, 01_test_01.txt, 01_test_02.txt, 01_test_03.txt, 01_test_04.txt, 01_test_05.txt, 01_test_06.txt, 01_test_07.txt, 01_test_08.txt, 01_test_09.txt, 01_test_10.txt, 01_test_11.txt, 01_test_12.txt, 01_test_13.txt, 01_test_14.txt, 01_test_15.txt, 01_test_16.txt, 01_test_17.txt, 01_test_18.txt, 01_test_19.txt, 01_test_20.txt, 01_test_21.txt, 01_test_22.txt, 01_test_23.txt, 01_test_24.txt
Case Name Status Exec Time Memory
00_sample_01.txt AC 1 ms 3500 KiB
00_sample_02.txt AC 1 ms 3496 KiB
00_sample_03.txt AC 1 ms 3528 KiB
01_test_01.txt AC 25 ms 5448 KiB
01_test_02.txt AC 100 ms 11324 KiB
01_test_03.txt AC 11 ms 4228 KiB
01_test_04.txt AC 100 ms 11272 KiB
01_test_05.txt AC 93 ms 10808 KiB
01_test_06.txt AC 102 ms 11284 KiB
01_test_07.txt AC 58 ms 8376 KiB
01_test_08.txt AC 99 ms 11376 KiB
01_test_09.txt AC 89 ms 10800 KiB
01_test_10.txt AC 100 ms 11288 KiB
01_test_11.txt AC 100 ms 11268 KiB
01_test_12.txt AC 102 ms 11340 KiB
01_test_13.txt AC 38 ms 6524 KiB
01_test_14.txt AC 100 ms 11384 KiB
01_test_15.txt AC 43 ms 7056 KiB
01_test_16.txt AC 102 ms 11368 KiB
01_test_17.txt AC 25 ms 5476 KiB
01_test_18.txt AC 100 ms 11408 KiB
01_test_19.txt AC 62 ms 8948 KiB
01_test_20.txt AC 100 ms 11244 KiB
01_test_21.txt AC 76 ms 21840 KiB
01_test_22.txt AC 78 ms 21876 KiB
01_test_23.txt AC 61 ms 11064 KiB
01_test_24.txt AC 55 ms 11064 KiB