Submission #1086535

#	Time	Username	Problem	Language	Result	Execution time	Memory
1086535		UNOwen07	Stove (JOI18_stove)	C++17	100 / 100	13 ms	3064 KiB

stove

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

long long n, k, arr[100005], sum;
priority_queue<long long> q;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    cin >> n >> k;
    long long tam = n-k;
    sum = n;
    cin >> arr[1];
    for (int i = 2; i <= n; i++){
        cin >> arr[i];
        long long dif = arr[i]-arr[i-1]-1;
        sum += dif;
        q.push(dif);
        if (q.size() > tam){
            sum -= q.top();
            q.pop();
        }
    }
    cout << sum;
    return 0;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:19:22: warning: comparison of integer expressions of different signedness: 'std::priority_queue<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   19 |         if (q.size() > tam){
      |             ~~~~~~~~~^~~~~

• Subtask 1: 20 / 20
• Subtask 2: 30 / 30
• Subtask 3: 50 / 50