Submission #394025

#	Time	Username	Problem	Language	Result	Execution time	Memory
394025		SoGood	Longest beautiful sequence (IZhO17_subsequence)	C++17	23 / 100	298 ms	8836 KiB

subsequence

//#pragma GCC optimize ("Ofast")
//#pragma GCC optimize ("unroll-loops")
//#pragma GCC target("avx,avx2,fma")

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
#define pb push_back
#define F first
#define S second
#define ins insert
#define mp make_pair
#define fo(i, n1, n, x) for(int i = n1; i <= n; i += x)
#define foo(i, n, n1, x) for(int i = n; i >= n1; i -= x)
#define bit __builtin_popcount
#define md (l + ((r - l) / 2))
#define all(x) x.begin(),x.end()
#define eb emplace_back
#define ub upper_bound
#define lb lower_bound
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)

using namespace std;
//using namespace __gnu_pbds;
//using namespace __gnu_cxx;


using ll = long long;

//#define int ll

using pii = pair<int,int>;
using pll = pair<ll,ll>;
using vi = vector<int>;



const int N = 2e5 + 11, mod = 1e9 + 7, mod2 = 998244353;
const int MAX = 1e5 + 11;
const int INF1 = 2e9 + 11;
const ll INF2 = 6e18 + 11;
const double INF3 = 1e8 + 11;
const int base = 500;
const int P = 31;
const int dx[] = {1, -1, 0, 0, 1, 1, -1, -1};
const int dy[] = {0, 0, 1, -1, 1, -1, 1, -1};
const double EPS = 1e-4;
const double PI = acos(-1.0);


//template<typename T> using ordered_set = tree <T, null_type, less <T>, rb_tree_tag, tree_order_statistics_node_update>;
template<typename T1, typename T2> inline void chmin(T1 &a, T2 b) { if (a > b) a = b; }
template<typename T1, typename T2> inline void chmax(T1 &a, T2 b) { if (a < b) a = b; }


mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int rnd(int l, int r) { return uniform_int_distribution<int> (l, r)(rng); }

int n, dp[(1 << 20) + 3], pr[(1<<20)+3], last[(1<<20)+3];
pii a[N];
main() {
  file("paint");
  ios;
  cin >> n;
  fo(i, 1, n, 1) cin >> a[i].F;
  fo(i, 1, n, 1) cin >> a[i].S;
  set <int> st;
  int mx = 0, sst = n;
  fo(i, 1, n, 1) {
    bool flag = 0;
    chmax(dp[a[i].F], 1);
    if (st.size() <= 5000) {
      for (auto submask : st) {
        if (bit(submask & a[i].F) == a[i].S) {
          if (dp[a[i].F] < dp[submask] + 1) {
            pr[i] = last[submask];
            dp[a[i].F] = dp[submask] + 1;
          }
        }
      }
    } else {
      for (int submask = 0; submask < (1 << 8) + 3; ++submask) {
        if (!last[submask]) continue;
        if (bit(submask & a[i].F) == a[i].S) {
          if (dp[a[i].F] < dp[submask] + 1) {
            pr[i] = last[submask];
            dp[a[i].F] = dp[submask] + 1;
          }
        }
      }
    }
    st.ins(a[i].F);
    last[a[i].F] = i;
    if (mx < dp[a[i].F]) mx = dp[a[i].F], sst = i;
  }
  cout << mx << "\n";
  vi ans;
  while (sst) ans.eb(sst), sst = pr[sst];
  while (!ans.empty()) cout << ans.back() << " ", ans.pop_back();
  return 0;
}
/**
4
1 2 3 4
10 0 1 0

5
5 3 5 3 5
10 1 20 1 20

1010 10
0100 4
1111 15
0100 4

4
10 4 15 4
10 0 1 1
*/

Compilation message (stderr)

subsequence.cpp:63:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   63 | main() {
      |      ^
subsequence.cpp: In function 'int main()':
subsequence.cpp:72:10: warning: unused variable 'flag' [-Wunused-variable]
   72 |     bool flag = 0;
      |          ^~~~
subsequence.cpp:23:48: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   23 | #define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
      |                                         ~~~~~~~^~~~~~~~~~~~~~~~~~~~
subsequence.cpp:64:3: note: in expansion of macro 'file'
   64 |   file("paint");
      |   ^~~~
subsequence.cpp:23:77: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   23 | #define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
      |                                                                      ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
subsequence.cpp:64:3: note: in expansion of macro 'file'
   64 |   file("paint");
      |   ^~~~

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