Submission #982940

#	Time	Username	Problem	Language	Result	Execution time	Memory
982940		_callmelucian	Stove (JOI18_stove)	C++14	100 / 100	16 ms	2272 KiB

stove

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

typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> pl;
typedef pair<int,int> pii;
typedef tuple<int,int,int> tt;

#define all(a) a.begin(), a.end()
#define filter(a) a.erase(unique(all(a)), a.end())

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

    int n, m; cin >> n >> m;
    vector<int> a(n), vec;

    for (int i = 0; i < n; i++) cin >> a[i];
    sort(all(a));

    for (int i = 1; i < n; i++)
        vec.push_back(a[i] - a[i - 1] - 1);
    sort(all(vec)); reverse(all(vec));

    int ans = a.back() - a[0] + 1;
    for (int i = 0; i < vec.size() && i < m - 1; i++) ans -= vec[i];
    cout << ans;

    return 0;
}

Compilation message (stderr)

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

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