제출 #162585

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
162585		fenech20	Stove (JOI18_stove)	C++14	100 / 100	85 ms	2392 KiB

stove

/**
 * Martin Leshko ([e-mail])
 */

#include <bits/stdc++.h>

using namespace std;

#define forn(i, n) for (int i = 0; i < (int)n; i++)
#define sz(a) (int)a.size()

const int N = 300001;

int main() {
  int n, k;
  cin >> n >> k;
  vector<int> a(n);
  for (int i = 0; i < n; i++)
    cin >> a[i];
  vector<int> sums;
  for (int i = 1; i < n; i++)
    if (a[i] - a[i - 1] > 1)
      sums.push_back(a[i] - a[i - 1] - 1);
  sort(sums.rbegin(), sums.rend());
  int ans = a[sz(a) - 1] - a[0] + 1;
  for (int i = 0; i < min(sz(sums), k - 1); i++)
    ans -= sums[i];
  cout << ans << endl;

}

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