제출 #347226

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
347226		BlancaHM	Stove (JOI18_stove)	C++14	100 / 100	61 ms	2156 KiB

stove

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int solucion(int K, vector<int> & momentos) {
	int N = (int) momentos.size();
	vector<int> huecos(N-1);
	for (int i = 0; i < N-1; i++)
		huecos[i] = momentos[i+1]-momentos[i]-1;
	sort(huecos.begin(), huecos.end(), greater<int>());
	int respuesta = momentos[N-1]+1 - momentos[0];
	for (int i = 0; i < K-1; i++) {
		respuesta -= huecos[i];
	}
	return respuesta;
}

int main() {
	int N, K;
	vector<int> momentos;
	cin >> N >> K;
	momentos = vector<int>(N);
	for (int i = 0; i < N; i++)
		cin >> momentos[i];
	cout << solucion(K, momentos) << '\n';
	return 0;
}

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