oj.uz

Submission #1100929

# Submission time Handle Problem Language Result Execution time Memory
1100929 2024-10-15T04:18:02 Z akzytr Stove (JOI18_stove) C++17
20 / 100
 217 ms 262144 KB 
#include <bits/stdc++.h>
using namespace std;
template <typename T> using ve = vector<T>;
template <typename T, int sz> using ar = array<T, sz>;
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<ll> times;
	map<ll, ll> occ;
	for(ll i = 0; i < N; i++) {
		ll x;
		cin >> x;
		times.pb(x);
		occ[x] = 1;
		times.pb(x + 1);
	}

	sort(times.begin(), times.end());
	times.erase(unique(times.begin(), times.end()), times.end());

	ll dp[times.size() + 2][times.size() + 2][2];
	for(ll i = 0; i < times.size(); i++) {
		for(ll j = 0; j <= K; j++) {
			dp[i][j][0] = (ll)1e18;
			dp[i][j][1] = (ll)1e18;
		}
	}

	for(ll i = times.size() - 1; i >= 0; i--) {
		if(occ[times[i]]) {
			break;
		}
		for(ll k = 0; k <= K; k++) {
			dp[i][k][0] = 0;
		}
	}

	for(ll i = times.size() - 1; i >= 0; i--) {
		for(ll left = K; left >= 1; left--) {
			// ON
			if(occ[times[i]]) {
				for(ll j = i + 1; j < times.size(); j++) {
					dp[i][left][1] = min(dp[i][left][1], dp[j][left - 1][0] + times[j] - times[i]);
				}
			} else {
				for(ll j = i + 1; j < times.size(); j++) {
					if(occ[times[j]] == 1) {
						dp[i][left][0] = min(dp[i][left][0], dp[j][left][1]);
						break;
					}
				}
			}
		}
	}

	cout << min(dp[0][K][0], dp[0][K][1]) << endl;
}

/*
Subtask 1) N<=20 & Ti<=20

N * K * T greedy


DP[i][left][on] which means the stove is on on the i'th day

DP[i][left][off]
*/

Compilation message

stove.cpp: In function 'int main()':
stove.cpp:30:18: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int, std::allocator<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |  for(ll i = 0; i < times.size(); i++) {
      |                ~~^~~~~~~~~~~~~~
stove.cpp:50:25: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int, std::allocator<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   50 |     for(ll j = i + 1; j < times.size(); j++) {
      |                       ~~^~~~~~~~~~~~~~
stove.cpp:54:25: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int, std::allocator<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |     for(ll j = i + 1; j < times.size(); j++) {
      |                       ~~^~~~~~~~~~~~~~

Subtask #1
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 336 KB Output is correct
2 Correct 0 ms 336 KB Output is correct
3 Correct 0 ms 336 KB Output is correct
4 Correct 1 ms 336 KB Output is correct
5 Correct 0 ms 336 KB Output is correct
6 Correct 1 ms 336 KB Output is correct
7 Correct 1 ms 344 KB Output is correct
8 Correct 0 ms 336 KB Output is correct
9 Correct 1 ms 336 KB Output is correct

Subtask #2
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 336 KB Output is correct
2 Correct 0 ms 336 KB Output is correct
3 Correct 0 ms 336 KB Output is correct
4 Correct 1 ms 336 KB Output is correct
5 Correct 0 ms 336 KB Output is correct
6 Correct 1 ms 336 KB Output is correct
7 Correct 1 ms 344 KB Output is correct
8 Correct 0 ms 336 KB Output is correct
9 Correct 1 ms 336 KB Output is correct
10 Runtime error 217 ms 262144 KB Execution killed with signal 9
11 Halted 0 ms 0 KB -

Subtask #3
0 / 50.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 336 KB Output is correct
2 Correct 0 ms 336 KB Output is correct
3 Correct 0 ms 336 KB Output is correct
4 Correct 1 ms 336 KB Output is correct
5 Correct 0 ms 336 KB Output is correct
6 Correct 1 ms 336 KB Output is correct
7 Correct 1 ms 344 KB Output is correct
8 Correct 0 ms 336 KB Output is correct
9 Correct 1 ms 336 KB Output is correct
10 Runtime error 217 ms 262144 KB Execution killed with signal 9
11 Halted 0 ms 0 KB -