Submission #1124395

#	Time	Username	Problem	Language	Result	Execution time	Memory
1124395		votranngocvy	Longest beautiful sequence (IZhO17_subsequence)	C++20	40 / 100	109 ms	1964 KiB

subsequence

#include <bits/stdc++.h>
using namespace std;

#define pii pair<int,int>
#define fi first
#define se second
const int N = 1e5 + 5;
int a[N],k[N],n;

namespace sub2 {
    int dp[N],trace[N];
    void solve() {
        int ans = 0,cur = 0;
        for (int i = 1; i <= n; i++) {
            dp[i] = 1;
            for (int j = 1; j < i; j++)
                if (__builtin_popcount(a[i] & a[j]) == k[i] && dp[i] < dp[j] + 1) {
                    dp[i] = dp[j] + 1;
                    trace[i] = j;
                }
            if (dp[i] > ans) ans = dp[i],cur = i;
        }
        vector<int>vec;
        for (int i = cur; i > 0; i = trace[i]) vec.push_back(i);
        reverse(vec.begin(),vec.end());
        cout << ans << "\n";
        for (auto x: vec) cout << x << " ";
        cout << "\n";
    }
}

namespace sub3 {
    int dp[N],trace[N];
    pii best[260];
    bool check_condition() {
        for (int i = 1; i <= n; i++)
            if (a[i] >= (1 << 8)) return false;
        return true;
    }
    void solve() {
        int ans = 0,cur = 0;
        for (int i = 1; i <= n; i++) {
            dp[i] = 1;
            for (int mask = 0; mask < (1 << 8); mask++) {
                if (__builtin_popcount(mask & a[i]) == k[i] && best[mask].fi + 1 > dp[i]) {
                    dp[i] = best[mask].fi + 1;
                    trace[i] = best[mask].se;
                }
            }
            if (dp[i] > best[a[i]].fi) best[a[i]] = {dp[i],i};
            if (dp[i] > ans) ans = dp[i],cur = i;
        }
        vector<int>vec;
        for (int i = cur; i > 0; i = trace[i]) vec.push_back(i);
        reverse(vec.begin(),vec.end());
        cout << ans << "\n";
        for (auto x: vec) cout << x << " ";
        cout << "\n";
    }
}

signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i];
    for (int i = 1; i <= n; i++) cin >> k[i];
    if (n <= 5000) sub2::solve();
    else if (sub3::check_condition()) sub3::solve();
    //else sub4::solve();
}

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