Submission #287983

#	Time	Username	Problem	Language	Result	Execution time	Memory
287983		PeppaPig	Magic Tree (CEOI19_magictree)	C++14	100 / 100	203 ms	36472 KiB

magictree

#include <bits/stdc++.h>

#define long long long
#define pii pair<int, int>
#define x first
#define y second

using namespace std;

const int N = 1e5+5;

int n, m, k;
long day[N], cost[N];
map<int, long> mp[N];
vector<int> g[N];

int sack(int u) {
    int sz = 1; pii ret(0, -1);
    for(int v : g[u]) {
        int z = sack(v);
        sz += z, ret = max(ret, pii(z, v));
    }
    if(ret.y == -1) {
        if(day[u]) mp[u][day[u]] += cost[u];
        return sz; 
    } else {
        swap(mp[u], mp[ret.y]);
        for(int v : g[u]) if(v != ret.y) 
            for(auto it : mp[v]) mp[u][it.x] += it.y;
        if(day[u]) {
            mp[u][day[u]] += cost[u];
            auto it = mp[u].upper_bound(day[u]);
            while(it != mp[u].end() && it->y <= cost[u]) {
                cost[u] -= it->y;
                it = mp[u].erase(it);
            }
            if(it != mp[u].end()) it->y -= cost[u], cost[u] = 0;
        }
        return sz;
    }
}

int main() {
    scanf("%d %d %d", &n, &m, &k);
    for(int i = 2, p; i <= n; i++) {
        scanf("%d", &p);
        g[p].emplace_back(i);
    }
    for(int i = 1, a; i <= m; i++) {
        scanf("%d", &a);
        scanf("%lld %lld", day + a, cost + a);
    }
    sack(1);
    long ans = 0;
    for(auto it : mp[1]) ans += it.y;
    printf("%lld\n", ans);

    return 0;
}

Compilation message (stderr)

magictree.cpp: In function 'int main()':
magictree.cpp:44:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   44 |     scanf("%d %d %d", &n, &m, &k);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
magictree.cpp:46:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   46 |         scanf("%d", &p);
      |         ~~~~~^~~~~~~~~~
magictree.cpp:50:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   50 |         scanf("%d", &a);
      |         ~~~~~^~~~~~~~~~
magictree.cpp:51:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   51 |         scanf("%lld %lld", day + a, cost + a);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• 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