oj.uz

Submission #1109964

# Submission time Handle Problem Language Result Execution time Memory
1109964 2024-11-08T09:13:46 Z ljtunas Magic Tree (CEOI19_magictree) C++17
70 / 100
 145 ms 101192 KB 
// author : anhtun_hi , nqg
#include <bits/stdc++.h>
#define ll long long
#define ii pair<ll, ll>
#define fi first
#define se second
#define all(x) x.begin(), x.end()
#define reset(h, v)    memset(h, v, sizeof h)
#define Forv(i, a)     for(auto& i : a)
#define For(i, a, b)   for(int i = a; i <= b; ++i)
#define Ford(i, a, b)  for(int i = a; i >= b; --i)
#define c_bit(i)     __builtin_popcountll(i)
#define Bit(mask, i)    ((mask >> i) & 1LL)
#define onbit(mask, i)  ((mask) bitor (1LL << i))
#define offbit(mask, i) ((mask) &~ (1LL << i))
using namespace std;
namespace std {
    template <typename T, int D>
    struct _vector : public vector <_vector <T, D - 1>> {
        static_assert(D >= 1, "Dimension must be positive!");
        template <typename... Args>
        _vector(int n = 0, Args... args) : vector <_vector <T, D - 1>> (n, _vector <T, D - 1> (args...)) {}
    };
    template <typename T> struct _vector <T, 1> : public vector <T> {
        _vector(int n = 0, const T& val = T()) : vector <T> (n, val) {}
    };
    template <class A, class B> bool minimize(A &a, B b){return a > b ? a = b, true : false;}
    template <class A, class B> bool maximize(A &a, B b){return a < b ? a = b, true : false;}
}
const int dx[] = {0, 0, +1, -1}, dy[] = {-1, +1, 0, 0}, LOG = 20, base = 311, inf = 1e9 + 5;
const int MAXN = 1e6 + 5;
const  ll  mod = 1e9 + 7;
const  ll   oo = 1e18;

#define int long long
int n, m, k; vector<int> g[MAXN];

ii c[MAXN];

map<int, ll> mp[MAXN]; bool ok[MAXN];
ll dp[MAXN][2];

void dfs(int u, int par){
    Forv(v, g[u]) if(v ^ par){
        dfs(v, u);
        if(mp[u].size() < mp[v].size()) swap(mp[u], mp[v]);
        Forv(tmp, mp[v]) mp[u][tmp.fi] += tmp.se;
    }
    if (ok[u]){
        auto [day, w] = c[u];
        mp[u][day] += w;
        auto it = mp[u].upper_bound(day);
        while (it != mp[u].end()){
            if (it->se >= w){ it->se -= w; break; }
            auto cur = it; ++it;
            mp[u].erase(cur);
        }
    }
}

void Solve() {
    cin >> n >> m >> k;
    For(i, 2, n){
        int p; cin >> p; g[p].push_back(i);
    }
    For(i, 1, m){
        int node, day, w;
        cin >> node >> day >> w;
        c[node] = {day, w};
        ok[node] = true;
    }
    dfs(1, 0);
    ll ans = 0;
    for(auto [_, w] : mp[1]) ans += w;
    cout << ans << '\n';
}

int32_t main() {
    cin.tie(0) -> sync_with_stdio(0);
    if(fopen("CEOI19_magictree.inp", "r")) {
        freopen("CEOI19_magictree.inp", "r", stdin);
        freopen("CEOI19_magictree.out", "w", stdout);
    }

    int t = 1;
//    cin >> t;

    for (int test = 1; test <= t; test++) {
        Solve();
    }
    return 0;
}

Compilation message

magictree.cpp: In function 'int32_t main()':
magictree.cpp:81:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   81 |         freopen("CEOI19_magictree.inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
magictree.cpp:82:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   82 |         freopen("CEOI19_magictree.out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 72016 KB Output is correct
2 Correct 24 ms 72016 KB Output is correct
3 Correct 27 ms 72008 KB Output is correct
4 Correct 26 ms 72008 KB Output is correct
5 Correct 27 ms 72056 KB Output is correct
6 Correct 26 ms 72016 KB Output is correct
7 Correct 28 ms 72016 KB Output is correct
8 Correct 32 ms 72016 KB Output is correct
9 Correct 36 ms 72040 KB Output is correct

Subtask #2
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 78 ms 83016 KB Output is correct
2 Correct 68 ms 84192 KB Output is correct
3 Correct 145 ms 100424 KB Output is correct
4 Correct 95 ms 83652 KB Output is correct
5 Correct 123 ms 87144 KB Output is correct

Subtask #3
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 33 ms 72008 KB Output is correct
2 Correct 37 ms 72264 KB Output is correct
3 Correct 40 ms 72396 KB Output is correct
4 Correct 99 ms 92488 KB Output is correct
5 Correct 95 ms 96356 KB Output is correct
6 Correct 110 ms 92568 KB Output is correct

Subtask #4
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 76 ms 79188 KB Output is correct
2 Correct 76 ms 78408 KB Output is correct
3 Correct 70 ms 83528 KB Output is correct
4 Correct 62 ms 80124 KB Output is correct
5 Correct 71 ms 92256 KB Output is correct

Subtask #5
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 72016 KB Output is correct
2 Correct 24 ms 72016 KB Output is correct
3 Correct 27 ms 72008 KB Output is correct
4 Correct 26 ms 72008 KB Output is correct
5 Correct 27 ms 72056 KB Output is correct
6 Correct 26 ms 72016 KB Output is correct
7 Correct 28 ms 72016 KB Output is correct
8 Correct 32 ms 72016 KB Output is correct
9 Correct 36 ms 72040 KB Output is correct
10 Correct 86 ms 83272 KB Output is correct
11 Correct 85 ms 81992 KB Output is correct
12 Correct 74 ms 83544 KB Output is correct
13 Correct 69 ms 80068 KB Output is correct
14 Correct 81 ms 94340 KB Output is correct

Subtask #6
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 38 ms 72776 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 72016 KB Output is correct
2 Correct 24 ms 72016 KB Output is correct
3 Correct 27 ms 72008 KB Output is correct
4 Correct 26 ms 72008 KB Output is correct
5 Correct 27 ms 72056 KB Output is correct
6 Correct 26 ms 72016 KB Output is correct
7 Correct 28 ms 72016 KB Output is correct
8 Correct 32 ms 72016 KB Output is correct
9 Correct 36 ms 72040 KB Output is correct
10 Correct 33 ms 72008 KB Output is correct
11 Correct 37 ms 72264 KB Output is correct
12 Correct 40 ms 72396 KB Output is correct
13 Correct 99 ms 92488 KB Output is correct
14 Correct 95 ms 96356 KB Output is correct
15 Correct 110 ms 92568 KB Output is correct
16 Correct 86 ms 83272 KB Output is correct
17 Correct 85 ms 81992 KB Output is correct
18 Correct 74 ms 83544 KB Output is correct
19 Correct 69 ms 80068 KB Output is correct
20 Correct 81 ms 94340 KB Output is correct
21 Correct 59 ms 75848 KB Output is correct
22 Correct 89 ms 85012 KB Output is correct
23 Correct 98 ms 89168 KB Output is correct
24 Correct 136 ms 101192 KB Output is correct
25 Correct 85 ms 83652 KB Output is correct
26 Correct 123 ms 87624 KB Output is correct
27 Correct 113 ms 85832 KB Output is correct

Subtask #8
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 72016 KB Output is correct
2 Correct 24 ms 72016 KB Output is correct
3 Correct 27 ms 72008 KB Output is correct
4 Correct 26 ms 72008 KB Output is correct
5 Correct 27 ms 72056 KB Output is correct
6 Correct 26 ms 72016 KB Output is correct
7 Correct 28 ms 72016 KB Output is correct
8 Correct 32 ms 72016 KB Output is correct
9 Correct 36 ms 72040 KB Output is correct
10 Correct 78 ms 83016 KB Output is correct
11 Correct 68 ms 84192 KB Output is correct
12 Correct 145 ms 100424 KB Output is correct
13 Correct 95 ms 83652 KB Output is correct
14 Correct 123 ms 87144 KB Output is correct
15 Correct 33 ms 72008 KB Output is correct
16 Correct 37 ms 72264 KB Output is correct
17 Correct 40 ms 72396 KB Output is correct
18 Correct 99 ms 92488 KB Output is correct
19 Correct 95 ms 96356 KB Output is correct
20 Correct 110 ms 92568 KB Output is correct
21 Correct 76 ms 79188 KB Output is correct
22 Correct 76 ms 78408 KB Output is correct
23 Correct 70 ms 83528 KB Output is correct
24 Correct 62 ms 80124 KB Output is correct
25 Correct 71 ms 92256 KB Output is correct
26 Correct 86 ms 83272 KB Output is correct
27 Correct 85 ms 81992 KB Output is correct
28 Correct 74 ms 83544 KB Output is correct
29 Correct 69 ms 80068 KB Output is correct
30 Correct 81 ms 94340 KB Output is correct
31 Incorrect 38 ms 72776 KB Output isn't correct
32 Halted 0 ms 0 KB -