#include<bits/stdc++.h>
#define pb emplace_back
#define AI(i) begin(i), end(i)
using namespace std;
using ll = long long;
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] : ", args)
void kout() { cerr << endl; }
template<class T, class ...U>
void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
void debug(auto L, auto R) { while (L != R) cerr << *L << " \n"[next(L) == R], ++L; }
vector<int> ans{0,1,1,1,1,0,1,1,0,1,1};
int mxq;
bool isSubsequence(vector<int> a) {
for (int i = 0, j = 0;i < a.size();++i) {
while (j < ans.size() && ans[j] != a[i])
++j;
if (j == ans.size()) return false;
++j;
}
mxq = max(mxq, (int)a.size());
return true;
}
#else
#include "grader.h"
#define debug(...) 0
#define DE(...) 0
#endif
const int MAX_N = 300010, inf = 1e9;
const vector<int> zero{0}, one{1};
int n;
bool query(vector<int> a) {
if (a.size() > n) return false;
return isSubsequence(a);
}
vector<int> operator + (vector<int> a, vector<int> b) {
a.insert(end(a), AI(b));
return a;
}
vector<pair<int,int>> edge[MAX_N];
int dp[MAX_N];
vector<int> findSequence(int _n) {
n = _n;
int flip = 0;
auto makev = [&](int a, int v1, int b, int v2, int c, int v3) {
return vector<int>(a, v1^flip) + vector<int>(b, v2^flip) + vector<int>(c, v3^flip);
};
auto con = [&](int l, int r, int sum) {
DE(l, r, sum);
edge[r].pb(l-1, -sum);
};
auto concon = [&](int l, int r, int sum) {
edge[l-1].pb(r, sum);
};
DE(n);
vector<int> Z, O, ans;
while (Z.size() == O.size() && Z.size() + O.size() < n) {
if (query(Z + zero))
Z.pb(0);
if (query(O + one))
O.pb(1);
}
if (Z.size() > O.size())
swap(Z, O), flip ^= 1;
while (Z.size () + O.size () < n)
O.pb(0);
int m = Z.size();
//con(1, m+1, O.size());
auto bellmand = [&]() {
dp[m+1] = O.size();
fill(dp+1, dp+m+1, inf);
queue<pair<int,int>> q;
q.emplace(m+1, dp[m+1]);
q.emplace(0, 0);
while (q.size()) {
auto [x, len] = q.front(); q.pop();
if (len > dp[x]) continue;
for (auto [u, w] : edge[x])
if (dp[x] + w < dp[u]) {
dp[u] = dp[x] + w;
q.emplace(u, dp[u]);
}
}
};
for (int l0 = 0;l0 <= m;++l0)
for (int r0 = 0;r0 + l0 <= m;++r0) {
int mx1 = 0;
int l = 0, r = n - l0 - r0, m;
while (l < r) {
m = l + r >> 1;
if (query(makev(l0, 0, m, 1, r0, 0)))
r = m;
else
l = m+1;
}
mx1 = l;
con(l0+1, m+1 - r0, mx1);
if (l0 + r0 + mx1 < n)
concon(l0+1, m+1 - r0, mx1);
}
bellmand();
for (int i = m+1;i >= 1;--i)
dp[i] -= dp[i-1];
for (int i = 1;i <= m+1;++i) {
while (dp[i]--)
ans.pb( 1 ^ flip );
if (i <= m)
ans.pb( 0 ^ flip );
}
return ans;
}
#ifdef KEV
int32_t main() {
ios_base::sync_with_stdio(0), cin.tie(0);
vector<int> res = findSequence(ans.size());
if (res == ans) {
cerr << "AC!\n";
DE(mxq);
}
else {
cerr << "WA\n";
debug(AI(res));
cerr << "Should be\n";
debug(AI(ans));
}
}
#endif
Compilation message
hidden.cpp: In function 'bool query(std::vector<int>)':
hidden.cpp:34:15: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
34 | if (a.size() > n) return false;
| ~~~~~~~~~^~~
hidden.cpp: In lambda function:
hidden.cpp:28:17: warning: statement has no effect [-Wunused-value]
28 | #define DE(...) 0
| ^
hidden.cpp:51:3: note: in expansion of macro 'DE'
51 | DE(l, r, sum);
| ^~
hidden.cpp: In function 'std::vector<int> findSequence(int)':
hidden.cpp:28:17: warning: statement has no effect [-Wunused-value]
28 | #define DE(...) 0
| ^
hidden.cpp:57:2: note: in expansion of macro 'DE'
57 | DE(n);
| ^~
hidden.cpp:61:53: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
61 | while (Z.size() == O.size() && Z.size() + O.size() < n) {
| ~~~~~~~~~~~~~~~~~~~~^~~
hidden.cpp:74:31: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
74 | while (Z.size () + O.size () < n)
| ~~~~~~~~~~~~~~~~~~~~~~^~~
hidden.cpp:106:11: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
106 | m = l + r >> 1;
| ~~^~~
hidden.cpp:115:15: warning: 'm' may be used uninitialized in this function [-Wmaybe-uninitialized]
115 | con(l0+1, m+1 - r0, mx1);
| ~^~
grader.cpp: In function 'int main()':
grader.cpp:28:26: warning: format '%d' expects argument of type 'int', but argument 3 has type 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wformat=]
28 | fprintf (fifo_out, "%d\n", ans.size ());
| ~^ ~~~~~~~~~~~
| | |
| int std::vector<int>::size_type {aka long unsigned int}
| %ld
grader.cpp:29:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
29 | for (int i=0; i<ans.size () && i < N; i++)
| ~^~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
3061 ms |
7424 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1384 ms |
524292 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |