oj.uz

Submission #915101

# Submission time Handle Problem Language Result Execution time Memory
915101 2024-01-23T10:49:53 Z RaresFelix Fireworks (APIO16_fireworks) C++17
7 / 100
 1 ms 348 KB 
#include <bits/stdc++.h>
//#pragma GCC optimize("O3")
//#pragma GCC target("avx,avx2,fma")

#define sz(x) int((x).size())
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()

using namespace std;
using ll = long long;
using ld = long double;  // or double, if TL is tight
using str = string; 
using ii = pair<int, int>;
using pl = pair<ll, ll>;
using vi = vector<int>;
using vll = vector<ll>;

struct fun {
    ll ct = 0;
    vll V;
    fun &operator+=(const fun &rhs) {
        ct += rhs.ct;
        for(auto it : rhs.V)
            V.push_back(it);
        return *this;
    }

    fun clean(ll w) {
        sort(all(V));
        deque<ll> D;
        for(auto it : V) D.push_back(it);
        V.clear();
        while(D.size() > 2) {
            ct += D.back() - D.front();
            D.pop_back();
            D.pop_front();
        }
        for(auto it : D) V.push_back(it + w);
        if(V.size() == 1) V.push_back(V.back());
        return *this;
    }

    ll eval() {
        sort(all(V));
        deque<ll> D;
        for(auto it : V) D.push_back(it);
        V.clear();
        while(!D.empty()) {
            ct += D.back() - D.front();
            D.pop_back();
            if(!D.empty())
                D.pop_front();
        }
        return ct;
    }
    ll ev(ll i) {
        ll val = ct;
        for(auto it : V) val += abs(it - i);
        return val;
    }
};

struct tree {
    ll n;
    ll limExp;
    vi par;
    vector<vector<ii> > L, G;
    tree(ll N, ll LimExpl) : n(N), limExp(LimExpl), par(N), L(N), G(N) {}

    void add_edge(ll u, ll v, ll w) {
        L[u].push_back({v, w});
        L[v].push_back({u, w});
    }

    void init() {
        function<fun(ll, ll)> dfs = [&](ll u, ll p) {
            if(u >= limExp) { /// frunza cu bomba, nu face nimic 
                fun re;
                re.V.push_back(0);
                re.V.push_back(0);
                return re; 
            }
            fun re;
            for(auto [it, w] : L[u])
                if(it != p) {
                    auto fiu = dfs(it, u).clean(w);
                    re += fiu;
                }
            return re;
        };
        ll re = dfs(0, -1).eval();
        assert(re % 2 == 0);
        cout << re / 2 << "\n";
    }
};



int main() {
    cin.tie(0);
    ios_base::sync_with_stdio(0);
    ll n, m;
    cin >> n >> m;
    tree T(n + m, n);
    for(ll i = 1; i < n + m; ++i) {
        ll u, v, w;
        cin >> u >> w;
        --u;
        T.add_edge(u, i, w);
    }
    T.init();
    return 0;
}

Compilation message

fireworks.cpp: In function 'int main()':
fireworks.cpp:106:15: warning: unused variable 'v' [-Wunused-variable]
  106 |         ll u, v, w;
      |               ^

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 344 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 1 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 344 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct

Subtask #2
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 1 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 344 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 1 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 344 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Incorrect 1 ms 348 KB Output isn't correct
14 Halted 0 ms 0 KB -

Subtask #4
0 / 45.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 344 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 1 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 344 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Incorrect 1 ms 348 KB Output isn't correct
14 Halted 0 ms 0 KB -