Submission #123090

#	Time	Username	Problem	Language	Result	Execution time	Memory
123090		Hideo	Stove (JOI18_stove)	C++14	100 / 100	28 ms	2284 KiB

stove

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

#define ll long long
#define pb push_back
#define mk make_pair
#define fr first
#define sc second
#define vi vector < int >
#define vl vector < ll >
#define pi pair < int, ll >
#define pii pair < int, pi >
#define vii vector < pi >

const int N = 3e5 + 7;
const int INF = 1e9 + 7;

int a[N];
int n, k;
ll ans;

vi dif;

main(){
    cin >> n >> k;
    ans = n;
    for (int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
        if (i > 1)
            dif.pb(a[i] - a[i - 1] - 1);
    }
    sort(dif.begin(), dif.end());
    for (int i = 0; i < n - k; i++)
        ans += dif[i];
    cout << ans;
}

Compilation message (stderr)

stove.cpp:24:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main(){
      ^
stove.cpp: In function 'int main()':
stove.cpp:28:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &a[i]);
         ~~~~~^~~~~~~~~~~~~

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