Submission #1006686

#	Time	Username	Problem	Language	Result	Execution time	Memory
1006686		erering	Longest beautiful sequence (IZhO17_subsequence)	C++17	40 / 100	6007 ms	11892 KiB

subsequence

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define pb push_back
#define ll long long
#define int long long
const long long inf=1e18;
const int MOD=998244353;
const int N=(1<<20);
int bitcount[N];
signed main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int n; cin>>n;
    int a[n],k[n],sol[n],par[n];
    for(int i=0;i<n;i++)cin>>a[i];
    for(int i=0;i<(1<<20);i++){
        int x=i;
        while(x>0){
            if(x%2==1)bitcount[i]++;
            x/=2;
        }
    }
    int ans=0,idx=0;
    for(int i=0;i<n;i++){
        cin>>k[i];
        sol[i]=1;
        par[i]=i;
        for(int j=i-1;j>=0;j--){
            if(bitcount[(a[i]&a[j])]==k[i] && sol[j]+1>sol[i]){
                sol[i]=sol[j]+1;
                par[i]=j;
            }
        }
        if(ans<sol[i]){
            ans=sol[i]; idx=i;
        }
    }
    cout<<ans<<endl;
    vector<int> v;
    while(1){
        int x=idx;
        v.pb(x+1);
        idx=par[idx];
        if(idx==x)break;
    }
    for(int i=v.size()-1;i>=0;i--)cout<<v[i]<<" ";
}

• Subtask 1: 7 / 7
• Subtask 2: 16 / 16
• Subtask 3: 17 / 17
• Subtask 4: 0 / 60