이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
template<typename T>
void out(T x) { cout << x << endl; exit(0); }
#define watch(x) cout << (#x) << " is " << (x) << endl
using ll = long long;
const int maxn = 1e6 + 5;
int n;
int a[maxn];
int k[maxn];
int dp[maxn], p[maxn];
bool compat(int i, int j) {
return __builtin_popcount(a[i]&a[j]) == k[j];
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
cin>>n;
bool task3 = true;
for (int i=0; i<n; i++) {
cin>>a[i];
task3 &= (a[i]<256);
}
for (int i=0; i<n; i++) {
cin>>k[i];
}
if (task3) {
//watch(task3);
vector<pair<int,int>> w(256, {-1,-1}); //dp[i],i
for (int i=0; i<n; i++) {
dp[i]=1;
p[i]=-1;
for (int x=0; x<256; x++) {
if (k[i]==__builtin_popcount(x&a[i])) {
if (w[x].first+1>dp[i]) {
dp[i]=w[x].first+1;
p[i]=w[x].second;
}
}
}
w[a[i]]=max(w[a[i]],{dp[i],i});
}
} else {
for (int i=0; i<n; i++) {
dp[i]=1;
p[i]=-1;
for (int j=i-1; j>=0; j--) {
if (compat(j,i) && 1+dp[j]>dp[i]) {
dp[i]=1+dp[j];
p[i]=j;
}
}
}
}
int at=n-1;
for (int i=0; i<n; i++) {
if (dp[i]>dp[at]) at=i;
}
vector<int> ans;
while (~at) {
ans.push_back(at);
at=p[at];
}
sort(ans.begin(),ans.end());
cout<<ans.size()<<"\n";
for (int i: ans) cout<<i+1<<" ";
cout<<endl;
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |