Submission #1368623

#	Time	Username	Problem	Language	Result	Execution time	Memory
1368623		m0rtu_us0512	Snacks (NOI25_snacks)	C++20	100 / 100	273 ms	20984 KiB

Main

#include<bits/stdc++.h>
using namespace std;

int64_t N, Q;
map<int64_t, int64_t> A;
deque< tuple <int64_t, int64_t, int64_t> > query;
int64_t sum = 0;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    cin >> N >> Q;

    query.assign( Q , {0,0,0} );

    for( auto n=N ; n-- > 0 ; ){
        int64_t a ;
        cin >> a;
        A[a] ++ ;
    }
    for( auto &[l,r,x] : query ){
        cin >> l >> r >> x;
    }

    int64_t sum = 0 ;
    for( auto &[k,v] : A ){
        sum += k*v;
    }
    cout << sum << endl;

    while( query.size() ){
        auto [l,r,x] = query[0] ;
        query.pop_front();

        int64_t replaced = 0;

        // iterator
        auto it = A.lower_bound(l);

        for( ; it!=A.end() ; ){
            auto [k,v] = *it ;
            if( r < k )
                break;

            sum -= k*v;
            replaced += v;

            it = A.erase(it);
        }

        if( replaced ){
            A[x] += replaced;
            sum += x*replaced;
        }

        cout << sum << endl;
    }

    return 0;
}

• Subtask 0: 0 / 0
• Subtask 1: 5 / 5
• Subtask 2: 12 / 12
• Subtask 3: 21 / 21
• Subtask 4: 13 / 13
• Subtask 5: 16 / 16
• Subtask 6: 33 / 33