Submission #1070214

#	Time	Username	Problem	Language	Result	Execution time	Memory
1070214		VMaksimoski008	Longest beautiful sequence (IZhO17_subsequence)	C++17	0 / 100	0 ms	348 KiB

subsequence

#include <bits/stdc++.h>
//#define int long long

using namespace std;

using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

const int mod = 1e9 + 7;
const int LOG = 20;
const int maxn = 1e5 + 5;
const int M = 8;

signed main() {
    int n, id = 1, res = 0;
    cin >> n;

    vector<int> a(n+1), k(n+1), prv(n+1);
    for(int i=1; i<=n; i++) cin >> a[i];
    for(int i=1; i<=n; i++) cin >> k[i];

    // if(n <= 5000) {
    //     vector<int> dp(n+1);
    //     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;
    //                 prv[i] = j;
    //                 if(dp[i] > dp[id]) id = i;
    //             }
    //         }
    //     }

    //     cout << dp[id] << '\n';
    //     vector<int> ans;
    //     while(id) {
    //         ans.push_back(id);
    //         id = prv[id];
    //     }
    //     reverse(ans.begin(), ans.end());
    //     for(int &x : ans) cout << x << " ";
    //     return 0;
    // }

    vector<pair<int, int> > dp(256, { -1, -1 });
    for(int i=1; i<=n; i++) {
        prv[i] = i;
        if(dp[a[i]] == pii{ -1, -1 }) dp[a[i]] = { 1, i };
        for(int j=0; j<256; j++) {
            if(__builtin_popcount(a[i] & j) == k[i] && dp[a[i]].first < dp[j].first + 1) {
                dp[a[i]].first = dp[j].first + 1;
                prv[i] = dp[j].second;
                dp[a[i]].second = i;
            }
        }

        if(dp[a[i]].first > res) {
            res = dp[a[i]].first;
            id = i;
        }
    }

    cout << res << '\n';
    vector<int> ans;
    while(id != prv[id]) {
        ans.push_back(id);
        id = prv[id];
    }
    if(id) ans.push_back(id);
    reverse(ans.begin(), ans.end());
    for(int &x : ans) cout << x << " ";
    return 0;
}

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