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;
typedef pair<int, int> pii;
const int N = 1e5+10;
const int K = 1024;
const int X = 10;
int p[N], a[N], x[N], n;
pii dp[K][2*X+1];
int main() {
cin>>n;
for (int i=1; i<=n; ++i) cin>>a[i];
for (int i=1; i<=n; ++i) cin>>x[i];
pii ans={1, 1};
for (int i=1; i<=n; ++i) {
int lf=a[i]&((1<<X)-1), rf=((a[i]>>X)&((1<<X)-1));
pair<int, int> mx={0, 0};
for (int lms=0; lms<K; ++lms) {
int v=x[i]-__builtin_popcount(lms&lf);
if (v < 0) continue;
mx=max(mx, dp[lms][v]);
} p[i]=mx.second;
for (int rms=0; rms<K; ++rms) {
int v=__builtin_popcount(rms&rf);
dp[lf][v]=max(dp[lf][v], {mx.first+1, i});
}
ans=max(ans, {mx.first+1, i});
}
vector<int> v;
int x=ans.second;
while (x != 0) v.push_back(x), x=p[x];
reverse(v.begin(), v.end());
assert((int)v.size() == ans.first);
cout<<ans.first<<"\n";
for (auto u : v) cout<<u<<" ";
cout<<"\n";
}
# | 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... |