Submission #115969

#	Time	Username	Problem	Language	Result	Execution time	Memory
115969		oolimry	Stove (JOI18_stove)	C++14	100 / 100	24 ms	2684 KiB

stove

#include <bits/stdc++.h>

using namespace std;

int main()
{
    //freopen("i.txt","r",stdin);
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    int n, k;
    cin >> n >> k;

    vector<int> arr;
    for(int i = 0;i < n;i++){
        int x;
        cin >> x;
        arr.push_back(x);
    }
    sort(arr.begin(),arr.end());
    vector<int> diff;

    for(int i = 1;i < arr.size();i++){
        diff.push_back(arr[i] - arr[i-1]);
    }

    sort(diff.begin(),diff.end());
    reverse(diff.begin(),diff.end());

    int ans = arr.back() - arr[0];

    for(int i = 0;i < k-1;i++){
        if(diff[i] <= 1) break;
        ans -= (diff[i]);
        ans++;
    }
    ans++;
    cout << ans;
    return 0;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:22:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 1;i < arr.size();i++){
                   ~~^~~~~~~~~~~~

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