Submission #394168

#	Time	Username	Problem	Language	Result	Execution time	Memory
394168		SoGood	Longest beautiful sequence (IZhO17_subsequence)	C++17	100 / 100	5770 ms	98092 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[N], pr[N], sst, mx, best[(1 << 10) + 5][(1 << 10) + 5][23];
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;
  fo(i, 1, n, 1) {
    dp[i] = 1;
    int left = (a[i].F >> 10);
    int right = (a[i].F & ((1 << 10) - 1));
    for (int mask = 0; mask < (1 << 10); ++mask) {
      int rem = (a[i].S - bit(a[i].F & mask));
      if (!(0 <= rem and rem <= 10)) continue;
      if (dp[best[left][mask][rem]] + 1 > dp[i]) {
        dp[i] = dp[best[left][mask][rem]] + 1;
        pr[i] = best[left][mask][rem];
      }
    }
    for (int mask = 0; mask < (1 << 10); ++mask) {
      int cnt = bit(left & mask);
      if (dp[best[mask][right][cnt]] < dp[i]) {
        best[mask][right][cnt] = i;
      }
    }
    if (mx < dp[i]) mx = dp[i], sst = i;
//    cerr << "CH " << i << " " << dp[i] << "\n";
  }
  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: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: 17 / 17
• Subtask 4: 60 / 60