Submission #494445

#	Time	Username	Problem	Language	Result	Execution time	Memory
494445		balbit	Fireworks (APIO16_fireworks)	C++14	7 / 100	8 ms	14424 KiB

fireworks

#include <bits/stdc++.h>
#define int ll
using namespace std;
#define ll long long
#define y1 zck_is_king
#define pii pair<ll, ll>
#define ull unsigned ll
#define f first
#define s second
#define ALL(x) x.begin(),x.end()
#define SZ(x) (int)x.size()
#define SQ(x) (x)*(x)
#define MN(a,b) a = min(a,(__typeof__(a))(b))
#define MX(a,b) a = max(a,(__typeof__(a))(b))
#define pb push_back
#define REP(i,n) for (int i = 0; i<n; ++i)
#define RREP(i,n) for (int i = n-1; i>=0; --i)
#define REP1(i,n) for (int i = 1; i<=n; ++i)
#define SORT_UNIQUE(c) (sort(c.begin(),c.end()), c.resize(distance(c.begin(),unique(c.begin(),c.end()))))
#ifdef BALBIT
#define IOS()
#define bug(...) fprintf(stderr,"#%d (%s) = ",__LINE__,#__VA_ARGS__),_do(__VA_ARGS__);
template<typename T> void _do(T &&x){cerr<<x<<endl;}
template<typename T, typename ...S> void _do(T &&x, S &&...y){cerr<<x<<", ";_do(y...);}
#else
#define IOS() ios_base::sync_with_stdio(0);cin.tie(0);
#define endl '\n'
#define bug(...)
#endif

const int iinf = 1e9+10;
const ll inf = 1ll<<60;
const ll mod = 1e9+7 ;


void GG(){cout<<"0\n"; exit(0);}

ll mpow(ll a, ll n, ll mo = mod){ // a^n % mod
    ll re=1;
    while (n>0){
        if (n&1) re = re*a %mo;
        a = a*a %mo;
        n>>=1;
    }
    return re;
}

ll inv (ll b, ll mo = mod){
    if (b==1) return b;
    return (mo-mo/b) * inv(mo%b,mo) % mo;
}

const int maxn = 6e5+5;

vector<pii> g[maxn];
ll zero[maxn], p1[maxn], p2[maxn];

void dfs(int v) {
    if (SZ(g[v]) == 0) {
        p1[v] = p2[v] = 0;
        zero[v] = 0;
        return;
    }
    vector<int> chg;
    for (pii e : g[v]) {
        int u = e.f;
        dfs(u);
        chg.pb(p1[u] + e.s); chg.pb(p2[u]+e.s);
        zero[v] += zero[u] + e.s;
    }
    sort(ALL(chg));
    int n = SZ(g[v]);
    REP(i, n) {
        zero[v] -= chg[i];
    }
    p1[v] = chg[n-1]; p2[v] = chg[n];
    zero[v] += p1[v];
    bug(zero[v], p1[v], p2[v]);
}

signed main(){
    IOS();

    int n,m; cin>>n>>m;
    REP1(i, n+m-1) {
        int p,c; cin>>p>>c; --p; //--c;
        g[p].pb({i,c});
    }
    dfs(0);
    cout<<zero[0] - p1[0]<<endl;



}

• Subtask 1: 7 / 7
• Subtask 2: 0 / 19
• Subtask 3: 0 / 29
• Subtask 4: 0 / 45