Submission #1100960

#	Time	Username	Problem	Language	Result	Execution time	Memory
1100960		akzytr	Stove (JOI18_stove)	C++17	50 / 100	1057 ms	2128 KiB

stove

#include <bits/stdc++.h>
using namespace std;
template <typename T> using ve = vector<T>;
typedef long long ll;
#define pb push_back
#define fi first
#define se second
#define endl '\n'

int main() {

	int N, K;
	cin >> N >> K;

	ve<pair<int, int>> times(N);
	for(int i = 0; i < N; i++) {
		scanf("%d", &times[i].fi);
		times[i].se = 1;
	}

	int mi_left[K + 1] = {};
	int mi_left2[K + 1] = {};

	bool tk = false;

	for(int i = 0; i <= K; i++) {
		mi_left[i] = 0;
		mi_left2[i] = 1e9;
	}

	for(int i = times.size() - 1; i >= 0; i--) {

		if(i == times.size() - 1 || times[i].fi + 1 != times[i + 1].fi) {
			for(int left = K; left >= 0; left--) {
				mi_left2[left] = min(mi_left2[left], (tk == 0 ? 0 : (int)1e9) + times[i].fi + 1);
				if(left == 0) {
					break;
				}
				mi_left2[left] = min(mi_left2[left], mi_left[left] + times[i].fi + 1);
			}
		}

		tk = 1;
		for(int left = K; left >= 0; left--) {
			mi_left2[left] = min(mi_left2[left], (int)1e9 + times[i].fi);

			if(left == 0) {
				break;
			}
			mi_left[left] = min((int)1e9, mi_left2[left - 1] - times[i].fi);
		}
	}

	cout << mi_left[K] << endl;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:33:8: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int>, std::allocator<std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |   if(i == times.size() - 1 || times[i].fi + 1 != times[i + 1].fi) {
      |      ~~^~~~~~~~~~~~~~~~~~~
stove.cpp:17:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   17 |   scanf("%d", &times[i].fi);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~

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