#include <bits/stdc++.h>
//#include "grader.h"
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll int
#define ll1 long long
#define ull unsigned long long
#define dou long double
#define str string
#define vll vector<ll>
#define vi vector<int>
#define pll pair<ll, ll>
#define vpll vector<pll>
#define vbool vector<bool>
#define vstr vector<str>
#define vvll vector<vll>
#define pb push_back
#define pf push_front
#define endl "\n"
#define fr first
#define se second
// #define sortcmp(a) sort(a.begin(), a.end(), cmp)
#define sort(a) sort(a.begin(), a.end())
#define all(a) a.begin(), a.end()
#define reverse(a) reverse(a.begin(), a.end())
#define speed ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define INF 5000000000000000000
#define ordered_set tree<ll, null_type, less_equal<ll>, rb_tree_tag, tree_order_statistics_node_update>
using namespace std;
using namespace __gnu_pbds;
mt19937 rng(1488);
ll randll(ll l, ll r){
return uniform_int_distribution<ll>(l, r)(rng);
}
/*vector<map<ll, vll>> s;
vll p;
ll find(ll v) {
if (p[v] == v) return v;
return p[v] = find(p[v]);
}
void unite(ll a, ll b) {
a = find(a);
b = find(b);
if (a == b) return;
if (s[a].size() < s[b].size()) {
p[a] = b;
for (auto i : s[a]) {
for (int j : i.se) {
s[b][i.fr].pb(j);
}
}
s[a].clear();
}
else {
p[b] = a;
for (auto i : s[b]) {
for (int j : i.se) {
s[a][i.fr].pb(j);
}
}
s[b].clear();
}
}
vll tree1;
void bt(ll v, ll tl, ll tr, vll& a) {
if (tl == tr) {
tree1[v] = a[tl];
return;
}
ll tm = (tl+tr)/2;
bt(v*2, tl, tm, a);
bt(v*2, tm+1, tr, a);
tree1[v] = tree1[v*2]+tree1[v*2+1];
}
ll get(ll l, ll r, ll v, ll tl, ll tr) {
if (l <= tl && tr <= r) return tree1[v];
if (tl > r || tr < l) return 0;
ll tm = (tl+tr)/2;
return get(l, r, v*2, tl, tm)+get(l, r, v*2+1, tm+1, tr);
}
void update(ll idx, ll val, ll v, ll tl, ll tr) {
if (tl == tr) {
tree1[v] = val;
return;
}
ll tm = (tl+tr)/2;
if (idx <= tm) update(idx, val, v*2, tl, tm);
else update(idx, val, v*2+1, tm+1, tr);
tree1[v] = tree1[v*2]+tree1[v*2+1];
}*/
ll cnt(ll x) {
ll res = 0;
for (int i = 0; i < 21; i ++) {
if (x & (1 << i)) res ++;
}
return res;
}
int main() {
ll n;
cin >> n;
vll a(n), k(n);
for (int i =0 ; i < n; i ++) {
cin >> a[i];
}
for (int i =0 ; i < n; i ++) {
cin >> k[i];
}
vll dp(n, 1);
ll maxl = 0;
vll p(n, -1);
ll x = 0;
for (int i = 1; i < n; i ++) {
for (int j = i-1; j >= 0; j --) {
if (cnt(a[j] & a[i]) == k[i] && dp[j]+1 > dp[i]) {
dp[i] =dp[j]+1;
p[i] = j;
}
}
if (dp[i] > maxl) {
maxl = dp[i];
x = i;
}
}
cout << maxl << endl;
vll b;
while (x != -1) {
b.pb(x);
x = p[x];
}
reverse(b);
for (int i : b) {
cout << i+1 << ' ';
}
cout << endl;
}
# | 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... |