oj.uz

Submission #1115383

# Submission time Handle Problem Language Result Execution time Memory
1115383 2024-11-20T12:07:55 Z staszic_ojuz Stove (JOI18_stove) C++17
0 / 100
 33 ms 122704 KB 
#include <iostream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <stack>
#include <unordered_map>

using namespace std;

typedef long long ll;

struct triple {
    ll a;
    ll b;
    ll c;
};

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

    ll n, k; cin >> n >> k;
    vector<bool> os(1e9, false);
    ll pren = n;
    ll last = 0;
    for(int i = 0; i < n; i++) {
        ll t; cin >> t;
        os[t-1] = true;
        last = max(last, t);
    }

    vector<pair<ll,ll>> prze;
    if(os[0]) {
        prze.push_back({0, 1});
    }
    for(int i = 1; i < last + 1; i++) {
        if(os[i]) {
            if(os[i - 1]) {
                prze[prze.size() - 1].second+=1;
            } else {
                prze.push_back({i, 1});
            }
        }
    }
    n = prze.size();
    // ll st = n/k;
    // ll nad = n%k;
    
    vector<pair<ll, pair<ll,ll>>> diff(n - 1);
    for(int i = 0; i < n - 1; i++) {
        diff[i].first = prze[i + 1].first - prze[i].first - 1;
        diff[i].second = {i, i + 1};
        //cout << diff[i].first << " ";
    }

    sort(diff.begin(), diff.end(), [](pair<ll, pair<ll,ll>> a, pair<ll, pair<ll,ll>> b) {return a.first > b.first;});

    ll dod = prze.size() - k;
    ll wyn = pren;
    for(int i = 0; i < dod; i++) {
        wyn += diff[diff.size() - 1].first;
        diff.pop_back();
    }
    cout << wyn;


    return 0;
}

Subtask #1
0 / 20.0

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

Subtask #2
0 / 30.0

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

Subtask #3
0 / 50.0

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