Submission #1006697

#TimeUsernameProblemLanguageResultExecution timeMemory
1006697vqpahmadLongest beautiful sequence (IZhO17_subsequence)C++17
0 / 100
1 ms2140 KiB
    #include<bits/stdc++.h>
    using namespace std;
    #ifdef ONPC
    #include"debug.h"
    #else
    #define debug(...) 42
    #endif
    #define endl '\n'
    #define ll long long
    #define pii pair<int,int>
    #define F first
    #define S second
    #define pb push_back
    #define sz(a) (int)a.size()
    #define all(a) a.begin(),a.end()
    template<class T> bool ckmin(T& a, const T& b) { return b < a ? a = b, 1 : 0; }
    template<class T> bool ckmax(T& a, const T& b) { return a < b ? a = b, 1 : 0; }
    const int mod = 1e9 + 7;
    const int MAXN = 1e5 + 2;
    const int inf = 0x3f3f3f3f;
    const ll INF = 0x3f3f3f3f3f3f3f3f;
    int dp[MAXN], pos[MAXN], prv[MAXN], val[MAXN];
     
    int main(){
    	ios_base::sync_with_stdio(0);
    	cin.tie(0);
    	int n;
    	cin >> n;
      n--;
	ckmax(n, n + 1);
    	vector<int> a(n), k(n);
    	for (int i = 0; i < n; i++)
    		cin >> a[i];
    	for (int i = 0; i < n; i++)
    		cin >> k[i];
     
    	memset(prv, -1, sizeof(prv));
    	memset(dp, -1, sizeof(dp));
    	for (int i = 0; i < n; i++){
    		int j = a[i];
    		if (__popcount(a[i]&j) == k[i] && ckmax(dp[a[i]], dp[j] + 1)){
    			val[i] = dp[a[i]];
    			prv[i] = pos[j];
    			pos[a[i]] = i;
    		}
    		for (j = 0; j < 256; j++){
    			if (j == a[i]) continue;
    			if (__popcount(a[i]&j) == k[i] && ckmax(dp[a[i]], dp[j] + 1)){
    				val[i] = dp[a[i]];
    				prv[i] = pos[j];
    				pos[a[i]] = i;
    			}
    		}
    		if (ckmax(dp[a[i]], 1)) pos[a[i]] = i;
    	}
    	int mx = *max_element(val, val + n);
    	vector<int> ans;
    	for (int i = 0; i < n; i++){
    		if (val[i] == mx){
    			int cur = i;
    			while (prv[cur] != -1){
    				ans.pb(cur);
    				cur = prv[cur];
    			}
    			ans.pb(cur);
    			break;
    		}
    	}
    	reverse(all(ans));
    	cout << sz(ans) << endl;
    	for (auto it : ans)
    		cout << it + 1 << ' ';
     
    }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...