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>
#define ll long long
#define ld long double
#define ff first
#define ss second
#define pb push_back
#define vr(v) v.begin(),v.end()
#define rv(v) v.rbegin(),v.rend()
#define Code ios_base::sync_with_stdio(false);
#define By cin.tie(NULL);
#define Davit cout.tie(NULL);
const int B = 8;
using namespace std;
//#include "algo/debug.h"
struct State {
int len, end;
};
int main() {
int n;
cin >> n;
vector<int> a(n);
vector<int> k(n);
for (int i = 0; i < n; i++)cin >> a[i];
for (int i = 0; i < n; i++)cin >> k[i];
vector<State> dp1(1 << B,{-1,-1});
int ans = 1;
int last = 0;
vector<int> prev(n);
for (int i = 0; i < n; i++)prev[i] = i;
for (int i = 0; i < n; i++) {
if(dp1[i].len == -1){
dp1[a[i]] = {1, i};
}
for (int j = 0; j < (1 << B); j++) {
if (__builtin_popcount(a[i] & j) == k[i]) {
if (dp1[j].len + 1 > dp1[a[i]].len) {
dp1[a[i]] = {dp1[j].len + 1, i};
prev[i] = dp1[j].end;
}
}
}
if (dp1[a[i]].len > ans) {
ans = dp1[a[i]].len;
last = i;
}
}
cout << ans << endl;
vector<int> res;
while (prev[last] != last) {
res.pb(last);
last = prev[last];
}
res.pb(last);
reverse(vr(res));
for (int i = 0; i < (int) res.size(); i++) {
cout << res[i] + 1 << " ";
}
}
# | 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... |