Submission #36087

#	Time	Username	Problem	Language	Result	Execution time	Memory
36087		UncleGrandpa925	Longest beautiful sequence (IZhO17_subsequence)	C++14	23 / 100	79 ms	2336 KiB

subsequence

/*input
5
5 3 5 3 5
10 1 20 1 20
4
1 2 3 4
10 0 1 0
4
1 2 3 4
10 0 1 0
*/
#include <bits/stdc++.h>
using namespace std;
#define sp ' '
#define endl '\n'
#define fi first
#define se second
#define mp make_pair
#define int long long
#define N 5005
#define bit(x,y) ((x>>y)&1LL)
#define na(x) (#x) << ":" << x
ostream& operator << (ostream &os, vector<int>&x) {
	for (int i = 0; i < x.size(); i++) os << x[i] << sp;
	return os;
}
ostream& operator << (ostream &os, pair<int, int> x) {
	cout << x.fi << sp << x.se << sp;
	return os;
}
ostream& operator << (ostream &os, vector<pair<int, int> >&x) {
	for (int i = 0; i < x.size(); i++) os << x[i] << endl;
	return os;
}

int n;
int a[N], b[N];
int dp[N];
int par[N];
vector<int> order;
pair<int, int> bestMask[(1 << 8)];

void solveSub2() {
	pair<int, int> ans = mp(1, 1);
	for (int i = 1; i <= n; i++) {
		dp[i] = 1;
		for (int j = 1; j <= i - 1; j++) {
			if (dp[j] + 1 > dp[i] && __builtin_popcount(a[j]&a[i]) == b[i]) {
				dp[i] = dp[j] + 1;
				par[i] = j;
				ans = max(ans, mp(dp[i], i));
			}
		}
	}
	cout << ans.fi << endl;
	while (ans.fi >= 1) {
		order.push_back(ans.se);
		ans.se = par[ans.se];
		ans.fi--;
	}
	reverse(order.begin(), order.end());
	cout << order << endl;
}

void solveSub3() {
	pair<int, int> ans = mp(1, 1);
	for (int i = 1; i <= n; i++) {
		dp[i] = 1;
		for (int mask = 0; mask < (1 << 8); mask++) {
			if (bestMask[mask].fi + 1 > dp[i] && __builtin_popcount(mask & a[i]) == b[i]) {
				dp[i] = bestMask[mask].fi + 1;
				par[i] = bestMask[mask].se;
				ans = max(ans, mp(dp[i], i));
			}
		}
		bestMask[a[i]] = max(bestMask[a[i]], mp(dp[i], i));
	}
	cout << ans.fi << endl;
	while (ans.fi >= 1) {
		order.push_back(ans.se);
		ans.se = par[ans.se];
		ans.fi--;
	}
	reverse(order.begin(), order.end());
	cout << order << endl;
}

signed main() {
	ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	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) solveSub2();
	else solveSub3();
}

Compilation message (stderr)

subsequence.cpp: In function 'std::ostream& operator<<(std::ostream&, std::vector<long long int>&)':
subsequence.cpp:24:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < x.size(); i++) os << x[i] << sp;
                    ^
subsequence.cpp: In function 'std::ostream& operator<<(std::ostream&, std::vector<std::pair<long long int, long long int> >&)':
subsequence.cpp:32:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < x.size(); i++) os << x[i] << endl;
                    ^

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