Submission #967631

#	Time	Username	Problem	Language	Result	Execution time	Memory
967631		vjudge1	Stove (JOI18_stove)	C++17	100 / 100	15 ms	2140 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define ll long long
int main() {
    cin.tie(0)->sync_with_stdio(false);
    int n,k;
    cin >> n >> k;
    vector<int> t(n);
    for (int i=0;i<n;i++) {
        cin >> t[i];
    }
    vector<int> range(n-1);
    ll cost=n;
    for (int i=0;i<n-1;i++) {
        range[i] = t[i+1]-t[i]-1;
    }
    sort(range.begin(),range.end());
    for (int i=0;i<k-1;i++) {
        range.pop_back();
    }
    for (int i=0;i<range.size();i++) {
        cost+=range[i];
    }
    cout << cost;
    return 0;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:21:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |     for (int i=0;i<range.size();i++) {
      |                  ~^~~~~~~~~~~~~

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