This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "bits/stdc++.h"
using namespace std;
const int mxN = 100005;
const int mxBt = 10;
int dp[1 << mxBt][1 << mxBt][mxBt];
int ans[mxN];
int a[mxN];
int k[mxN];
void put(int d) {
for (int i = d - 1; 0 <= i; i --) {
int value = __builtin_popcount(a[i] & a[d]);
if (value == k[d] && ans[i] == ans[d] - 1) {
put(i);
break;
}
}
cout << d + 1 << ' ';
}
void solve() {
int N;
cin >> N;
for (int i = 0; i < N; i ++) {
cin >> a[i];
}
for (int i = 0; i < N; i ++) {
cin >> k[i];
}
int mask = (1 << mxBt) - 1;
int r = 0, d = -1;
for (int i = 0; i < N; i ++) {
ans[i] = 1;
for (int bottomMask = 0; bottomMask < (1 << mxBt); bottomMask ++) {
int need = k[i] - __builtin_popcount(bottomMask & a[i]);
if (0 <= need && need <= mxBt) {
ans[i] = max(ans[i], dp[bottomMask][a[i] >> mxBt][need] + 1);
}
}
for (int topMask = 0; topMask < (1 << mxBt); topMask ++) {
int value = __builtin_popcount((topMask << mxBt) & a[i]);
dp[a[i] & mask][topMask][value] = max(dp[a[i] & mask][topMask][value], ans[i]);
}
if (ans[i] > r) {
r = ans[i];
d = i;
}
}
cout << r << endl;
put(d);
}
int main() {
solve();
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |