Submission #1216409

#	Time	Username	Problem	Language	Result	Execution time	Memory
1216409		hengliao	Tree (IOI24_tree)	C++20	0 / 100	54 ms	24676 KiB

tree

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

#define F first
#define S second
#define pll pair<ll, ll>
#define vll vector<ll>
#define pb push_back

typedef long long ll;

namespace{

    const ll mxN=2e5+5;

    ll n, l, r;
    ll sum[mxN];
    multiset<pll> st[mxN];
    ll ans;
    ll p[mxN];
    vll adj[mxN];
    ll w[mxN];
    ll leafcnt;
    ll f, s;
}

void init(std::vector<int> P, std::vector<int> W) {
    n=W.size();
    for(ll i=1;i<n;i++){
        p[i]=P[i];
        adj[p[i]].pb(i);
    }
    for(ll i=0;i<n;i++){
        w[i]=W[i];
    }
    leafcnt=0;
    for(ll i=0;i<n;i++){
        if(adj[i].empty()){
            f+=w[i];
        }
    }
    for(ll i=0;i<n;i++){
        if(!adj[i].empty()){
            s+=w[i]*((ll) adj[i].size()-1);
        }
    }
}

long long query(int L, int R) {
    l=L;
    r=R;
    ll curans=f*l+s*r;
    return curans;
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 10
• Subtask 2: 0 / 13
• Subtask 3: 0 / 18
• Subtask 4: 0 / 7
• Subtask 5: 0 / 11
• Subtask 6: 0 / 22
• Subtask 7: 0 / 19