#include <vector>
#include <algorithm>
#include <iostream>
#include <set>
#include <cmath>
#include <map>
#include <random>
#include <cassert>
#include <ctime>
#include <stack>
#include <cstdlib>
#include <queue>
#include <cstdio>
#include <limits.h>
#pragma GCC target ("avx2")
#pragma GCC optimization ("O1")
#pragma GCC optimization ("unroll-loops")
using namespace std;
class SparseTable {
vector<vector<int>> dp;
vector<int> arr;
public:
int query (int l, int r) {
if (l == r) {
return arr[l];
}
int sz = log2(r - l);
return dp[l][sz] + query(l + (1 << sz), r);
}
SparseTable (vector<int> arr) {
this->arr = arr;
dp.resize(arr.size());
for (int i = 0; i < arr.size(); i++) {
dp[i].resize(32);
dp[i][0] = arr[i];
}
for (int j = 1; j < 32; j++) {
for (int i = 0; i < arr.size(); i++) {
dp[i][j] = dp[i][j - 1] + dp[min(i + (1 << (j - 1)), (int)arr.size() - 1)][j - 1];
}
}
}
};
bool submask (int mask, int cur) {
return ((cur | mask) == mask);
}
bool process (SparseTable& st, int A, int B, int N, int mask) {
bitset<2000> dp[N + 1];
for (int i = 0; i < N; i++) {
if (submask(mask, st.query(0, i))) {
dp[1].set(i);
}
}
for (int j = 2; j <= N; j++) {
for (int i = 0; i < N; i++) {
for (int prev = 0; prev < i; prev++) {
if (dp[j - 1][prev] && submask(mask,st.query(prev + 1, i))) {
//cout << "YES\n";
dp[j].set(i);
break;
}
}
}
}
for (int j = A; j <= B; j++) {
if (dp[j][N - 1]) {
return true;
}
}
return false;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int N, A, B;
cin >> N >> A >> B;
vector<int> arr(N);
for (int i = 0; i < N; i++) {
cin >> arr[i];
}
SparseTable st(arr);
for (int mask = 0; mask < 1000; mask++) {
if (process(st, A, B, N, mask)) {
cout << mask << '\n';
return 0;
}
}
}
Compilation message
sculpture.cpp:16: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
16 | #pragma GCC optimization ("O1")
|
sculpture.cpp:17: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
17 | #pragma GCC optimization ("unroll-loops")
|
sculpture.cpp: In constructor 'SparseTable::SparseTable(std::vector<int>)':
sculpture.cpp:35:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
35 | for (int i = 0; i < arr.size(); i++) {
| ~~^~~~~~~~~~~~
sculpture.cpp:40:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
40 | for (int i = 0; i < arr.size(); i++) {
| ~~^~~~~~~~~~~~
sculpture.cpp: In function 'bool process(SparseTable&, int, int, int, int)':
sculpture.cpp:52:5: error: 'bitset' was not declared in this scope
52 | bitset<2000> dp[N + 1];
| ^~~~~~
sculpture.cpp:15:1: note: 'std::bitset' is defined in header '<bitset>'; did you forget to '#include <bitset>'?
14 | #include <limits.h>
+++ |+#include <bitset>
15 | #pragma GCC target ("avx2")
sculpture.cpp:52:18: error: 'dp' was not declared in this scope
52 | bitset<2000> dp[N + 1];
| ^~
