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;
int n, m[1<<20], p[1<<20], in[1<<20];
vector <int> a, k;
int bitcount(int x){
int ans = 0;
while(x){
ans += (x%2);
x /= 2;
}
return ans;
}
int f(int x, int k2, int ind){
int ans = 0;
for(int i = 0; i <= (1<<8); i++){
// cout << x << ' ' << i << ' ' << (x&i) << ' ' << bitcount(x&i) << '\n';
if(bitcount(x&i) == k2){
// cout << y << ' ' << i << ' ' << m[i] << '\n';
if(ans < m[i]){
p[ind] = in[i];
}
ans = max(ans,m[i]);
}
}
return ans+1;
}
int main(){
cin >> n;
a.resize(n+1);
k.resize(n+1);
for(int i = 1; i <= n; i++){
cin >> a[i];
p[i] = -1;
}
for(int i = 1; i <= n; i++){
cin >> k[i];
}
int ans = 0, ind = 1;
for(int i = 1; i <= n; i++){
// cout << i << ' ' << k[i] << '\n';
m[a[i]] = max(f(a[i],k[i],i),m[a[i]]);
if(m[a[i]] > ans){
ind = i;
}
ans = max(ans,m[a[i]]);
in[a[i]] = i;
// cout << i << ' ' << m[a[i]] << '\n';
}
cout << ans << '\n';
vector <int> v;
while(ind > 0){
v.push_back(ind);
ind = p[ind];
}
reverse(v.begin(), v.end());
for(auto i : v){
cout << i << ' ';
}
}
# | 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... |