Submission #917475

#	Time	Username	Problem	Language	Result	Execution time	Memory
917475		hariaakas646	Fireworks (APIO16_fireworks)	C++17	100 / 100	158 ms	91984 KiB

fireworks

#include <bits/stdc++.h>
 
using namespace std;
 
#define scd(t) scanf("%d", &t)
#define sclld(t) scanf("%lld", &t)
#define forr(i, l, r) for(int i=l; i<r; i++)
#define frange(i, l) forr(i, 0, l)
#define pb push_back
#define mp make_pair
#define f first
#define s second
#define all(x) x.begin(), x.end()
 
typedef vector<int> vi;
typedef pair<int, int> pii;
typedef vector<pii> vii;
typedef long long lli;
typedef vector<vi> vvi;
typedef vector<lli> vll;
typedef vector<bool> vb;
typedef set<int> seti;
typedef multiset<int> mseti;
 
void usaco()
{
    freopen("/media/hariaakash646/785EF1075EF0BF46/CompetitiveProgramming/input.in", "r", stdin);
}
 
int n, m;
 
vector<vector<pair<int, lli>>> graph;
// vll dp;
 
struct Data {

    lli a = 0;
    lli b = 0;

    priority_queue<lli> pq;

};

Data dfs(int x, lli c) {
    if(x > n) {
        Data out;
        out.pq.push(c);
        out.pq.push(c);
        out.a = 1;
        out.b = -c;
        return out;
    }
    Data out;
    for(auto e : graph[x]) {
        auto v = dfs(e.f, e.s);
        out.a += v.a;
        out.b += v.b;
        if(out.pq.size() < v.pq.size()) swap(out.pq, v.pq);
        while(v.pq.size()) {
            lli x = v.pq.top();
            v.pq.pop();
            out.pq.push(x);
        }
    }
    while(out.a > 1) {
        out.a--;
        out.b += out.pq.top();
        out.pq.pop();
    }

    lli v1 = out.pq.top();
    out.pq.pop();
    lli v0 = out.pq.top();
    out.pq.pop();
    out.pq.push(v1 + c);
    out.pq.push(v0 + c);
    out.b -= c;
    return out;
}
 
int main() {
    // usaco();
    
    scd(n);
    scd(m);
 
    graph = vector<vector<pair<int, lli>>>(n+1);
 
    forr(i, 2, n+m+1) {
        int pa;
        lli c;
        scd(pa);
        sclld(c);
        graph[pa].pb(mp(i, c));
    }
    Data out = dfs(1, 0);
 
    while(out.a > 0) {
        out.a--;
        out.b += out.pq.top();
        out.pq.pop();
    }

    printf("%lld", out.b);

}

Compilation message (stderr)

fireworks.cpp: In function 'void usaco()':
fireworks.cpp:27:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |     freopen("/media/hariaakash646/785EF1075EF0BF46/CompetitiveProgramming/input.in", "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
fireworks.cpp: In function 'int main()':
fireworks.cpp:5:21: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    5 | #define scd(t) scanf("%d", &t)
      |                ~~~~~^~~~~~~~~~
fireworks.cpp:84:5: note: in expansion of macro 'scd'
   84 |     scd(n);
      |     ^~~
fireworks.cpp:5:21: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    5 | #define scd(t) scanf("%d", &t)
      |                ~~~~~^~~~~~~~~~
fireworks.cpp:85:5: note: in expansion of macro 'scd'
   85 |     scd(m);
      |     ^~~
fireworks.cpp:5:21: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    5 | #define scd(t) scanf("%d", &t)
      |                ~~~~~^~~~~~~~~~
fireworks.cpp:92:9: note: in expansion of macro 'scd'
   92 |         scd(pa);
      |         ^~~
fireworks.cpp:6:23: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    6 | #define sclld(t) scanf("%lld", &t)
      |                  ~~~~~^~~~~~~~~~~~
fireworks.cpp:93:9: note: in expansion of macro 'sclld'
   93 |         sclld(c);
      |         ^~~~~

• Subtask 1: 7 / 7
• Subtask 2: 19 / 19
• Subtask 3: 29 / 29
• Subtask 4: 45 / 45