제출 #1178353

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1178353		zyadhany	Longest beautiful sequence (IZhO17_subsequence)	C++20	100 / 100	4797 ms	127116 KiB

subsequence

#include <bits/stdc++.h>
#include <chrono>
#include <random>
#include <unordered_map>
#include <unordered_set>
 
#define ll long long
#define vi vector<ll>
#define vii vector<vi>
#define pl pair<ll, ll>
#define all(X) X.begin(),X.end()
#define vp vector<pl>
#define mi map<ll,ll>
 
#define ld long double
#define vc vector<char>
#define vcc vector<vc>
#define mc map<char,int>
#define sortx(X) sort(X.begin(),X.end());
#define allr(X) X.rbegin(),X.rend()
#define ln '\n'
#define YES {cout << "YES\n"; return;}
#define NO {cout << "NO\n"; return;}
#define MUN {cout << "-1\n"; return;}
using namespace std;

const int MODE = 998244353;


void solve(ll tc) {
    ll n; cin >> n;

    vi X(n), Y(n);

    for (int i = 0; i < n; i++) cin >> X[i];
    for (int i = 0; i < n; i++) cin >> Y[i];
    
    ll b = 10;
    vector<vii> mask(1<<b, vii(1<<b,vi(11, -1)));
    vi dp(n), par(n);
    for (int i = 0; i < n; i++)
    {
        dp[i] = 1, par[i] = i;
        ll la = (X[i] >> b);
        ll ra = (X[i] ^ (la << b));

        for (int j = 0; j < (1 << b); j++) {
            ll k = Y[i] - __builtin_popcount(la&j);
            if (k < 0 || k > 10) continue;
            ll ind = mask[j][ra][k];
            if (ind != -1 && dp[ind] >= dp[i]) {
                dp[i] = dp[ind]+1;
                par[i] = ind;
            } 
        }

        for (int j = 0; j < (1 << b); j++) {
            ll k = __builtin_popcount(ra&j);
            ll ind = mask[la][j][k];
            if (ind == -1 || dp[ind] < dp[i]) {
                mask[la][j][k] = i;
            } 
        }
    }
    

    ll mx = 0, at = -1;
    for (int i = 0; i < n; i++) if (dp[i] > mx) mx = dp[i], at = i;

    vi sol;
    while (par[at] != at) sol.push_back(at), at = par[at];
    sol.push_back(at);
    reverse(all(sol));

    cout << sol.size() << '\n';
    for (auto a : sol) cout << a + 1 << ' ';
    cout << '\n';
}
 
int32_t main()
{
    ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
    int size = 1;   

    // freopen("disrupt.in", "r", stdin   );
    // freopen("disrupt.out", "w", stdout);
    // cin >> size;
    for (int tc = 1; tc <= size; tc++){
        solve(tc);
       // if (tc != size) cout << '\n';
    }
}

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