제출 #841550

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
841550		Elvin_Fritl	Longest beautiful sequence (IZhO17_subsequence)	C++17	0 / 100	1 ms	560 KiB

subsequence

#include <bits/stdc++.h>
using namespace std;

const int N = 5050 , inf = 1e9 + 199;

int dp[N],ind[N];

int32_t main()
{

    int n;
    cin>>n;
    assert(n <= 5000);
    vector<int>a(n+1),k(n+1);
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        cin>>k[i];
    }
    dp[1]=1;
    for(int i=2;i<=n;i++){
        int maxi = 0 , indd = 1;
        if(__builtin_popcount(a[i-1]) < k[i-1]){
            continue;
        }
        for(int j=i-1;j>=1;j--){
            if(__builtin_popcount((a[i-1]&a[j-1])) == k[i-1]){
                if(dp[j]+1 > maxi) {
                    maxi = dp[j]+1;
                    indd = j;
                }
            }
        }
        if(maxi > 0){
            dp[i] = maxi;
            ind[i] = indd;
        }
        else{
            dp[i]=1;
            ind[i] = 0;
        }
    }
    int maxi = 0 , indd = 1;
    for(int i=1; i<=n; i++){
        if(dp[i] >= maxi){
            maxi = dp[i];
            indd = i;
        }
    }
    vector<int> res;
    while(indd != 0){
        res.push_back(indd);
        indd = ind[indd];
    }
    cout<<res.size()<<endl;
    sort(res.begin(), res.end());
    for(int i:res){
        cout<<i<<" ";
    }

}

/*



*/

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