Submission #1322484

#TimeUsernameProblemLanguageResultExecution timeMemory
1322484chithanhnguyenLongest beautiful sequence (IZhO17_subsequence)C++20
40 / 100
74 ms1600 KiB
/*
Author: Nguyen Chi Thanh - High School for the Gifted - VNU.HCM (i2528)
*/
#include <bits/stdc++.h>
using namespace std;

// #define int long long
#define ull unsigned long long
#define ld long double
#define pii pair<int, int>
#define fi first
#define se second
#define __builtin_popcount __builtin_popcountll
#define all(x) (x).begin(), (x).end()
#define BIT(x, i) (((x) >> (i)) & 1)
#define MASK(x) ((1ll << (x)))

#define debug(a, l, r) for (int _i = (l); _i <= (r); ++_i) cout << (a)[_i] << ' '; cout << '\n';

const int MAXN = 1e5 + 5;
int n, a[MAXN], k[MAXN];
int mxval; // For subtask3 check

void init() {
    cin >> n;
    for (int i = 1; i <= n; ++i) cin >> a[i];
    for (int i = 1; i <= n; ++i) cin >> k[i];

    mxval = *max_element(a + 1, a + n + 1);
}

namespace subtask12 {
    bool check() {return n <= 5000;}

    void solve() {
        vector<int> dp(n + 5, 0), trace(n + 5, 0);

        int res = 0, end_pos = 0;

        for (int i = 1; i <= n; ++i) {
            dp[i] = 1;
            for (int j = 1; j < i; ++j) {
                if (__builtin_popcount(a[j] & a[i]) != k[i]) continue;
                if (dp[j] + 1 > dp[i]) {
                    dp[i] = dp[j] + 1;
                    trace[i] = j;
                }
            }

            if (dp[i] > res) {
                res = dp[i];
                end_pos = i;
            }
        }

        cout << res << '\n';
        vector<int> path;

        int cur = end_pos;
        while (cur) {
            path.push_back(cur);
            cur = trace[cur];
        }

        reverse(all(path));
        for (auto u : path) cout << u << ' ';
    }
}

namespace subtask3 {
    bool check() {return mxval < MASK(8);}

    void solve() {
        int M = MASK(8);
        vector<int> dp_len(M, 0), dp_last(M, 0);
        vector<int> trace(n + 5, 0);

        int res = 0, end_pos = 0;

        for (int i = 1; i <= n; ++i) {
            int cur = a[i];

            int bestlen = 0, bestmask = 0;
            for (int premask = 0; premask < M; ++premask) {
                if (__builtin_popcount(premask & cur) != k[i]) continue;
                if (dp_len[premask] > bestlen) {
                    bestlen = dp_len[premask];
                    bestmask = premask;
                }
            }

            int newlen = bestlen + 1;
            trace[i] = dp_last[bestmask];

            if (newlen > dp_len[cur]) {
                dp_len[cur] = newlen;
                dp_last[cur] = i;
            }

            if (newlen > res) {
                res = newlen;
                end_pos = i;
            }
        }

        cout << res << '\n';
        vector<int> path;
        for (int cur = end_pos; cur; cur = trace[cur])
            path.push_back(cur);
        reverse(all(path));
        for (int u : path) cout << u << ' ';
    }
}

signed main() {
    #ifdef NCTHANH
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    #endif
    ios_base::sync_with_stdio(0);
    cin.tie(nullptr); cout.tie(nullptr);

    init();
    if (subtask12::check()) return subtask12::solve(), 0;
    if (subtask3::check()) return subtask3::solve(), 0;

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...