Submission #141511

#	Time	Username	Problem	Language	Result	Execution time	Memory
141511		osaaateiasavtnl	Stove (JOI18_stove)	C++14	100 / 100	30 ms	3316 KiB

stove

#include<bits/stdc++.h>
using namespace std;
#define int long long 
#define ii pair <int, int>
#define app push_back
#define all(a) a.begin(), a.end()
signed main() {
    #ifdef HOME
    freopen("input.txt", "r", stdin);
    #else
    ios_base::sync_with_stdio(0); cin.tie(0);
    #endif
    int n, k; cin >> n >> k;
    vector <int> a(n);
    for (int i = 0; i < n; ++i) cin >> a[i];
    sort(all(a)); vector <int> b;
    for (int i = 0; i + 1 < a.size(); ++i) b.app(a[i + 1] - a[i] - 1);
    sort(all(b)); reverse(all(b));
    int ans = a.back() - a[0] + 1;
    for (int i = 0; i < k - 1; ++i) ans -= b[i];
    cout << ans << '\n';
}   

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:17:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i + 1 < a.size(); ++i) b.app(a[i + 1] - a[i] - 1);
                     ~~~~~~^~~~~~~~~~

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