# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
953199 | happy_node | Financial Report (JOI21_financial) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include "ninja.h"
using namespace std;
const int MX=3e5+5;
int dp[MX], R[MX];
struct segtree {
pair<int,int> t[4*MX];
void update(int v, int l, int r, int pos, pair<int,int> val) {
if(l>pos||r<pos) return;
if(l==r) {
t[v]=val;
return;
}
int mid=(l+r)>>1;
update(v*2+0,l,mid+0,pos,val);
update(v*2+1,mid+1,r,pos,val);
t[v]=max(t[v*2+0],t[v*2+1]);
}
pair<int,int> query(int v, int l, int r, int ql, int qr) {
if(qr<l||r<ql) return make_pair(-1,-1);
if(ql<=l&&qr>=r) return t[v];
int mid=(l+r)>>1;
return max(query(v*2+0,l,mid+0,ql,qr),
query(v*2+1,mid+1,r,ql,qr));
}
void reset() {
for(int i=0;i<4*MX;i++) {
t[i]=make_pair(-1,-1);
}
}
} st;
struct segtreeMin {
int t[4*MX];
void update(int v, int l, int r, int pos, int val) {
if(l>pos||r<pos) return;
if(l==r) {
t[v]=val;
return;
}
int mid=(l+r)>>1;
update(v*2+0,l,mid+0,pos,val);
update(v*2+1,mid+1,r,pos,val);
t[v]=min(t[v*2+0],t[v*2+1]);
}
int query(int v, int l, int r, int ql, int qr) {
if(qr<l||r<ql) return 1e9;
if(ql<=l&&qr>=r) return t[v];
int mid=(l+r)>>1;
return min(query(v*2+0,l,mid+0,ql,qr),
query(v*2+1,mid+1,r,ql,qr));
}
void reset() {
for(int i=0;i<4*MX;i++) {
t[i]=1e9;
}
}
} tt;
int _N;
int getPos(int L, int R, int x) {
int l=L,r=R,res=L;
while(l<=r) {
int mid=(l+r)>>1;
if(tt.query(1,0,_N-1,mid,R)<=x) {
res=mid;
l=mid+1;
} else {
r=mid-1;
}
}
return res;
}
bool valid[1000][1000];
int rev[MX];
int solveChallange(int N, int D, std::vector<int> H) {
_N=N;
st.reset();
tt.reset();
vector<pair<int,int>> ord;
vector<int> P(N);
for(int i=0;i<N;i++) {
ord.push_back({H[i],-i});
}
sort(ord.begin(),ord.end());
int cnt=1;
for(auto [h,id]:ord) {
id=-id;
P[id]=cnt++;
rev[P[id]]=id;
}
for(int l=0;l<N;l++) tt.update(1,0,N-1,l,P[l]);
for(int l=N-1;l>=0;l--) {
// cari posisi terkanan sehingga min nya itu udah <= x
int p=getPos(l,min(N-1,l+D),P[l]);
R[l]=max(R[min(N-1,l+D)],R[p]);
}
for(int l=0;l<N;l++) {
int lst=l; // last yang <= H[l]
valid[l][l]=1;
for(int r=l+1;r<N;r++) {
if(r-lst<=D) valid[l][r]=valid[l][lst];
if(H[r]<=H[l]) lst=r;
}
for(int r=N-1;r>l;r--) {
if(valid[l][r]) {
R[l]=r;
break;
}
}
for(int r=l;r<=R[l];r++) {
assert(valid[l][r]);
}
}
for(int i=0;i<N;i++) {
while(true) {
auto [V,p]=st.query(1,1,N,1,P[i]-1);
if(p==-1) break;
if(R[p]<i) {
st.update(1,1,N,P[p],make_pair(-1,-1));
continue;
}
dp[i]=V+1;
break;
}
dp[i]=max(dp[i],1);
st.update(1,1,N,P[i],make_pair(dp[i],i));
}
return *max_element(dp,dp+N);
}
int main() {
int N, D;
std::vector<int> H;
scanf("%d %d", &N, &D);
H.resize(N);
for (auto &elem : H) {
scanf("%d", &elem);
}
int answer = solveChallange(N, D, H);
printf("%d\n", answer);
}