Submission #949980

#	Time	Username	Problem	Language	Result	Execution time	Memory
949980		Pragmatism	Longest beautiful sequence (IZhO17_subsequence)	C++17	23 / 100	28 ms	2812 KiB

subsequence

//Bismillahir-Rahmanir-Rahim

#include <bits/stdc++.h>
#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("O1,O2,O3,Ofast,unroll-loops")
#pragma GCC target("sse,sse2,sse3,sse4,sse4.1,sse4.2,popcnt,avx,avx2")

#define pb push_back
#define pii pair <int, int>
#define pll pair <long long, long long>
#define pld pair <long double, long double>
#define ll long long
#define ld long double
#define x first
#define y second
#define all(v) v.begin(),v.end()
#define sz(s) (int)s.size()
#define skip continue
#define bpop(x) (ll)__builtin_popcountll(x)

using namespace std;

const int N = 1e5 + 7;
const int M = 1e5 + 7;
const int maxA = 1e5 + 7;
const int inf = 1e9 + 7;
const ll INF = 2e18 + 7;
const int MOD = 998244353;
const ld eps = 1e-9;

pii dir[] = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};

//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
//clock_t start = clock();

#define int long long

pii dp[N];
int n, a[N], b[N], pr[N];
void subtask12() {
    int mx = 0, pos = 0;
    for (int i = 1;i <= n;i++) {
        dp[i] = {1, i};
        for (int j = 1;j < i;j++) {
            if (bpop(a[i] & a[j]) == b[i]) {
                //cout << i << ' ' << j << ':' << dp[j].x + 1 << '\n';
                if (dp[j].x + 1 > dp[i].x)dp[i] = {dp[j].x + 1, i}, pr[i] = j;
            }
        }
        if (dp[i].x > mx)mx = dp[i].x, pos = i;
    }
    cout << mx << '\n';
    vector <int> ans;
    while (pos != 0)ans.pb(pos), pos = pr[pos];
    reverse(all(ans));
    for (auto i : ans)cout << i << ' ';
}
void solve() {
    cin >> n;
    for (int i = 1;i <= n;i++)cin >> a[i];
    for (int i = 1;i <= n;i++)cin >> b[i];
    if (n <= 5000)subtask12(), exit(0);
    int mx = 0, pos = 0;
    for (int i = 1;i <= n;i++) {
        //cout << i << ":\n";
        for (int x = 0;x < (1ll << 8);x++) {
            if (bpop(x & a[i]) == b[i]) {
                //if (dp[x] > dp[a[i]])cout << x << ':' << dp[x].x << ' ' << dp[x].y << '\n';
                dp[a[i]] = max(dp[a[i]], {dp[x].x + 1, dp[x].y});
            }
        }
        pr[i] = dp[a[i]].y, dp[a[i]].y = i;
        if (dp[a[i]].x > mx)mx = dp[a[i]].x, pos = i;
    }
    cout << mx << '\n';
    vector <int> ans;
    while (pos != 0)ans.pb(pos), pos = pr[pos];
    reverse(all(ans));
    for (auto i : ans)cout << i << ' ';
}
signed main() {
    //srand(time(NULL));
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    //freopen("subsequence.in", "r", stdin);
    //freopen("subsequence.out", "w", stdout);
    int test = 1;
    //cin >> test;
    for (int t = 1;t <= test;t++) {
        //cout << "Case " << t << ": ";
        solve();
    }
    return 0;
}

Compilation message (stderr)

subsequence.cpp:4: warning: ignoring '#pragma comment ' [-Wunknown-pragmas]
    4 | #pragma comment(linker, "/stack:200000000")
      |

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