#include "unique.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> PickUnique(int N) {
vector<int> pre(N + 1, 0ll), suf(N + 1, 0ll), ans(N, 0ll);
pre[0] = 1;
suf[N - 1] = 1;
for(int i = 0; i < N; i++) {
if(i != 0) {
pre[i] = UniqueCount(0, i);
}
if(i != N - 1) {
suf[i] = UniqueCount(i, N - 1);
}
}
for(int i = 0; i < N; i++) {
if(i == 0) {
if(suf[i] > suf[i + 1]) ans[i] = 1;
} else if(i == N - 1) {
if(pre[i - 1] < pre[i]) ans[i] = 1;
} else{
if(pre[i-1] < pre[i] && suf[i] > suf[i + 1]) ans[i] = 1;
}
}
return ans;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |