Submission #753297

#	Time	Username	Problem	Language	Result	Execution time	Memory
753297		Olympia	K blocks (IZhO14_blocks)	C++17	53 / 100	1088 ms	596 KiB

blocks

#include <vector>
#include <iostream>
#include <cassert>
#include <cmath>
#include <map>
#include <set>
using namespace std;
struct SparseTableMax {
	vector<vector<int64_t>> dp;
	int64_t query (int l, int r) {
		int sz = log2(r - l + 1);
		return max(dp[l][sz], dp[r - (1 << sz) + 1][sz]);
	}
	SparseTableMax (vector<int> a) {
		int n = a.size();
		dp.resize(n);
		for (int i = 0; i < a.size(); i++) {
			dp[i].resize((int)log2(n) + 1);
			dp[i][0] = a[0];
		}
		for (int j = 1; j < dp[0].size(); j++) {
			for (int i = 0; i < a.size(); i++) {
				dp[i][j] = max(dp[i][j - 1], dp[min(i + (1 << (j - 1)), n - 1)][j - 1]);
			}
		}
	}
};
int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int N, K;
    cin >> N >> K;
    int64_t arr[N];
    for (int i = 0; i < N; i++) {
        cin >> arr[i];
    }
    int64_t INF = 1e17;
    vector<int64_t> prev(N + 1), cur(N + 1);
    prev.assign(N + 1, INF), cur.assign(N + 1, INF);
    for (int j = 1; j <= N; j++) {
        prev[j] = ((j == 1) ? arr[0] : max(prev[j - 1], arr[j - 1]));
    }

    for (int i = 2; i <= K; i++) {
        vector<int64_t> val;
        val.assign(N + 1, INF);
        for (int j = 1; j <= N; j++) {
            if (j == 1) {
            	val[0] = prev[0] + arr[0];
            	cur[1] = prev[0] + arr[0];
            } else {
                cur[j] = INF;
                int64_t myMax = 0;
                int ind = -1;
                for (int l = j - 1; l >= 0; l--) {
                    myMax = max(myMax, arr[l]);
                    if (myMax == arr[j - 1]) {
                        cur[j] = min(cur[j], prev[l] + arr[j - 1]);
                        val[l] = prev[l] + arr[j - 1];
                        ind = l;
                    } else {
                    	break;
                    }
                }
                for (int l = ind - 1; l >= 0; l--) {
                	cur[j] = min(cur[j], val[l]);
                }
            }
        }
        swap(cur, prev);
    }
    cout << prev[N];
}

Compilation message (stderr)

blocks.cpp: In constructor 'SparseTableMax::SparseTableMax(std::vector<int>)':
blocks.cpp:17:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   17 |   for (int i = 0; i < a.size(); i++) {
      |                   ~~^~~~~~~~~~
blocks.cpp:21:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |   for (int j = 1; j < dp[0].size(); j++) {
      |                   ~~^~~~~~~~~~~~~~
blocks.cpp:22:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   22 |    for (int i = 0; i < a.size(); i++) {
      |                    ~~^~~~~~~~~~

• Subtask 1: 14 / 14
• Subtask 2: 18 / 18
• Subtask 3: 21 / 21
• Subtask 4: 0 / 47