Submission #877151

#TimeUsernameProblemLanguageResultExecution timeMemory
877151__Davit__Longest beautiful sequence (IZhO17_subsequence)C++17
7 / 100
14 ms21084 KiB
#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];
    int ans = 1;
    int last = 0;
    vector<int> prev(n);
    for (int i = 0; i < n; i++)prev[i] = i;
    if (n <= 15) {
        vector<int> dp(n, 1);
        for (int i = 1; i < n; i++) {
            for (int j = 0; j < i; j++) {
                if (__builtin_popcount(a[i] & a[j]) == k[i]) {
                    if (dp[j] + 1 > dp[i]) {
                        dp[i] = dp[j] + 1;
                        prev[i] = j;
                    }
                }
            }
            if (dp[i] > ans) {
                ans = dp[i];
                last = i;
            }
        }
    } else {

        vector<vector<State>> dp1(n, vector<State>(1 << B, {-10, -10}));
        for (int i = 0; i < n; i++) {
            dp1[i][a[i]] = {1, i};
        }
        for (int i = 1; i < n; i++) {
            for (int j = 0; j < (1 << B); j++) {
                if (__builtin_popcount(a[i] & j) == k[i]) {
                    if (dp1[i - 1][j].len + 1 > dp1[i][a[i]].len) {
                        dp1[i][a[i]] = {dp1[i - 1][j].len + 1, i};
                        prev[i] = dp1[i - 1][j].end;
                    }
                }
            }
            if (dp1[i][a[i]].len > ans) {
                ans = dp1[i][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 timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...