Submission #486676

#	Time	Username	Problem	Language	Result	Execution time	Memory
486676		my04	Longest beautiful sequence (IZhO17_subsequence)	C++17	23 / 100	6061 ms	2380 KiB

subsequence

#include <bits/stdc++.h>

#define pb push_back
#define mp make_pair
#define sz(x) (int)(x).size()
#define S second
#define F first
#define all(x) (x).begin(), (x).end()

using namespace std;
using ll = long long;

void setIO(string name = "") {
	ios_base::sync_with_stdio(NULL);
	cin.tie(NULL);
	cout.tie(NULL);

	if (sz(name)) {
		freopen((name + ".in").c_str(), "r", stdin);
		freopen((name + ".out").c_str(), "w", stdout);
	}
}

const int inf = 1e9;
const ll INF = 1e18;            
const int mod = 1e9 + 7;
const int MAXN = 1e6 + 5;

int n;
int a[MAXN];
int k[MAXN];

int dp[MAXN];
int par[MAXN];

void solve() {
	cin >> n;

	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}

	for (int i = 1; i <= n; i++) {
		cin >> k[i];
	}

	for (int i = 1; i <= n; i++) {
		par[i] = -1;
		dp[i] = 1;		
	}

	for (int i = 1; i <= n; i++) {
		for (int j = i + 1; j <= n; j++) {
			int x = (a[i] & a[j]);

			if (__builtin_popcount(x) == k[j]) {
				if (dp[j] < dp[i] + 1) {
					dp[j] = dp[i] + 1;
					par[j] = i;
				}
			}
		}
	}

	int x = 1;
	for (int i = 2; i <= n; i++) {
		if (dp[x] < dp[i]) {
			x = i;
		}
	}

	cout << dp[x] << '\n';
	
	vector<int> path;
	while (x != -1) {
		path.pb(x);
		x = par[x];
	}

	reverse(all(path));

	for (int i : path) {
		cout << i << ' ';
	}
}                   	

main() {
	setIO();
	
	int tt = 1;
	// cin >> tt;
	while (tt--) {
		solve();
	}
	
	return 0;
}

Compilation message (stderr)

subsequence.cpp:87:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   87 | main() {
      | ^~~~
subsequence.cpp: In function 'void setIO(std::string)':
subsequence.cpp:19:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   19 |   freopen((name + ".in").c_str(), "r", stdin);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
subsequence.cpp:20:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   20 |   freopen((name + ".out").c_str(), "w", stdout);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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