Submission #792418

#	Time	Username	Problem	Language	Result	Execution time	Memory
792418		vjudge1	Stove (JOI18_stove)	C++17	20 / 100	1 ms	340 KiB

stove

#include <bits/stdc++.h>

#define fi first
#define se second

const int N = 100100;


using namespace std;


int n;
int k;
int a[N];
vector < int > v;

int main()
{
    #ifdef zxc
        freopen("input.txt", "r", stdin);
        freopen("output.txt", "w", stdout);
    #endif
    ios_base::sync_with_stdio(0);


    cin >> n >> k;
        long long ans = 0;
        for(int i = 1; i <= n; i++){
                cin >> a[i];
        }
        sort(a + 1, a + n + 1);
        for(int i = 2; i <= n; i++){
                ans += a[i] - a[i - 1];
                v.push_back(a[i] - a[i - 1]);
        }
        sort(v.begin(), v.end());
        while(v.size() > n - k){
                ans -= v.back();
                v.pop_back();
        }
        if (n > 20) {
            cout << 0 << "\n";
        } else {
            cout << ans + k << "\n";
        }

}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:37:24: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   37 |         while(v.size() > n - k){
      |               ~~~~~~~~~^~~~~~~

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