Submission #646272

#	Time	Username	Problem	Language	Result	Execution time	Memory
646272		Pety	Magic Tree (CEOI19_magictree)	C++14	100 / 100	136 ms	36028 KiB

magictree

#include <bits/stdc++.h>
#define ll long long

using namespace std;

const ll INF = 1e9;
const ll MOD = 1e9 + 9;

ll n, m, w[100002], d[100002], k;
vector<ll>G[100002];
map<ll, ll>aib[100002];



void dfs (ll nod) {
  for (auto it : G[nod]) {
    dfs(it);
    if (aib[nod].size() < aib[it].size())
      swap(aib[nod], aib[it]);
    for (auto it2 : aib[it])
      aib[nod][it2.first] += it2.second;
  }
  if (w[nod]) {
    aib[nod][d[nod]] += w[nod]; 
    ll x = 0;
    while (1) {
      auto it = aib[nod].upper_bound(d[nod]);
      if (it == aib[nod].end())
        break;
      if (it->second <= w[nod] - x) {
        x += it->second;
        aib[nod].erase(it);
      
      }
      else {
        it->second -= w[nod];
        it->second += x;
        break;
      }
    }
  }
}


int main () 
{
  ios_base::sync_with_stdio(false);
  cin >> n >> m >> k;
  for (ll i = 2; i <= n; i++) {
    ll x;
    cin >> x;
    G[x].push_back(i);
  }
  for (ll i = 1; i <= m; i++) {
    ll v;
    cin >> v;
    cin >> d[v] >> w[v];
  }
  dfs(1);
  ll ans = 0;
  for (auto it : aib[1])
    ans += it.second;
  cout << ans;
  return 0;
}

• Subtask 1: 6 / 6
• Subtask 2: 3 / 3
• Subtask 3: 11 / 11
• Subtask 4: 12 / 12
• Subtask 5: 16 / 16
• Subtask 6: 13 / 13
• Subtask 7: 22 / 22
• Subtask 8: 17 / 17