제출 #541078

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
541078		vinnipuh01	Longest beautiful sequence (IZhO17_subsequence)	C++17	23 / 100	47 ms	18800 KiB

subsequence

#include <iostream>
#include <bits/stdc++.h>
#include <cmath>
#include <algorithm>
#include <vector>
#include <deque>
#include <set>
#include <stack>
#include <string>
#include <map>
#include <queue>
#define int long long
 
using namespace std;
 
const long long oo = 1000000000000000000;
 
long long sum, ans = 0, mx = 0, mn = 1000000000, num, pos;
 
 
/*
    ViHHiPuh
 
   (( `'-""``""-'` ))
     )-__-_.._-__-(
   / --- (o _ o) --- \
   \ .-* ( .0. ) *-. /
   _'-. ,_ '=' _, .-'_
  / `;#'#'# - #'#'#;` \
 \_)) -----'#'----- ((_/
      # --------- #
  '# ------- ------ #'
  /..-'# ------- #'-.\
  _\...-\'# -- #'/-.../_
  ((____)- '#' -(____))
 
 
    cout << fixed << setprecision(6) << x;
 
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
 
    freopen ( "sum.in", "r", stdin )
*/
 
int n, a[ 1000001 ], b[ 1000001 ], mp[ 1000001 ], p[ 1000001 ], mpp[ 2000001 ], bl[ 1000001 ];
vector <int> v;
 
main () {
	cin >> n;
	for ( int i = 0; i <= 2000000; i ++ ) {
		mpp[ i ] = __builtin_popcount( i );
	}
	for ( int i = 1; i <= n; i ++ )
		cin >> a[ i ];
	for ( int i = 1; i <= n; i ++ )
		cin >> b[ i ];
	if ( n <= 5000 ) {
		mp[ 1 ] = 1;
		p[ 1 ] = -1;
		for ( int i = 2; i <= n; i ++ ) {
			p[ i ] = -1;
			mp[ i ] = 1;
			for ( int j = i - 1; j >= 1; j -- ) {
				if ( mpp[ ( a[ i ] & a[ j ] ) ] == b[ i ] && mp[ i ] < mp[ j ] + 1 ) {
					mp[ i ] = mp[ j ] + 1;
					p[ i ] = j;
				}
			}
		}
		mx = 0;
		for ( int i = 1; i <= n; i ++ ) {
			if ( mx < mp[ i ] ) {
				mx = mp[ i ];
				pos = i;
			}
		}
		for ( int i = pos; i != -1; i = p[ i ] ) {
			v.push_back( i );
		}
		cout << mx << "\n";
		reverse( v.begin(), v.end() );
		for ( auto i : v )
			cout << i << " ";
		return 0;
	}
	bl[ a[ 1 ] ] = 1;
	p[ 1 ] = -1;
	mp[ a[ 1 ] ] = 1;
	for ( int i = 2; i <= n; i ++ ) {
		if ( !mp[ a[ i ] ] )
			mp[ a[ i ] ] = 1, bl[ a[ i ] ] = i;
		p[ i ] = -1;
		for ( int mask = a[ i ]; mask; mask = ( mask - 1 ) & a[ i ] ) {
			if ( bl[ mask ] != i && ( mask & a[ i ] ) == b[ i ] && mp[ mask ] + 1 > mp[ a[ i ] ] ) {
				mp[ a[ i ] ] = mp[ mask ] + 1;
				p[ i ] = bl[ mask ];
				bl[ a[ i ] ] = i;
			}
		}
	}
	mx = 0;
	for ( int i = 1; i <= n; i ++ ) {
		if ( mx < mp[ a[ i ] ] ) {
			mx = mp[ a[ i ] ];
			pos = bl[ a[ i ] ];
		}
	}
	for ( int i = pos; i != -1; i = p[ i ] )
		v.push_back( i );
	reverse( v.begin(), v.end() );
	for ( auto i : v )
		cout << i << " ";
}

컴파일 시 표준 에러 (stderr) 메시지

subsequence.cpp:49:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   49 | main () {
      | ^~~~

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