Submission #851031

#	Time	Username	Problem	Language	Result	Execution time	Memory
851031		LucaIlie	Financial Report (JOI21_financial)	C++17	100 / 100	711 ms	31608 KiB

Main

#include <bits/stdc++.h>

using namespace std;

const int MAX_N = 3e5;
int v[MAX_N], p[MAX_N];

struct SegTree {
    int segTree[4 * MAX_N];
    int lazy[4 * MAX_N];

    void propag( int v, bool leaf ) {
        if ( segTree[v] == 0 )
            return;
        segTree[v] = max( segTree[v], lazy[v] );
        if ( !leaf ) {
            lazy[v * 2 + 1] = max( lazy[v * 2 + 1], lazy[v] );
            lazy[v * 2 + 2] = max( lazy[v * 2 + 2], lazy[v] );
        }
        lazy[v] = 0;
    }

    void updatePos( int v, int l, int r, int pos, int x ) {
        propag( v, (l == r) );

        if ( l > pos || r < pos )
            return;

        if ( l == r ) {
            segTree[v] = max( segTree[v], x );
            return;
        }

        int mid = (l + r) / 2;
        updatePos( v * 2 + 1, l, mid, pos, x );
        updatePos( v * 2 + 2, mid + 1, r, pos, x );
        segTree[v] = max( segTree[v * 2 + 1], segTree[v * 2 + 2] );
    }

    void update( int v, int l, int r, int lu, int ru, int x ) {
        if ( lu > ru )
            return;

        propag( v, (l == r) );

        if ( l > ru || r < lu )
            return;

        if ( lu <= l && r <= ru ) {
            lazy[v] = max( lazy[v], x );
            segTree[v] = max( segTree[v], x );
            propag( v, (l == r) );
            return;
        }

        int mid = (l + r) / 2;
        update( v * 2 + 1, l, mid, lu, ru, x );
        update( v * 2 + 2, mid + 1, r, lu, ru, x );
        segTree[v] = max( segTree[v * 2 + 1], segTree[v * 2 + 2] );
    }

    int query( int v, int l, int r, int lq, int rq ) {
        if ( lq > rq )
            return 0;
        propag( v, (l == r));

        if ( l > rq || r < lq )
            return 0;
        if ( lq <= l && r <= rq )
            return segTree[v];

        int mid = (l + r) / 2;
        return max( query( v * 2 + 1, l, mid, lq, rq ), query( v * 2 + 2, mid + 1, r, lq, rq ) );
    }
};

struct DSU {
    int parent[MAX_N + 1], maxx[MAX_N + 1];

    void init() {
        for ( int i = 0; i <= MAX_N; i++ )
            parent[i] = maxx[i] = i;
    }

    int findParent( int u ) {
        if ( parent[u] == u )
            return u;

        parent[u] = findParent( parent[u] );
        return parent[u];
    }

    void join( int u, int v ) {
        u = findParent( u );
        v = findParent( v );

        maxx[u] = max( maxx[u], maxx[v] );
        parent[v] = u;
    }

    int getMax( int u ) {
        return maxx[findParent( u )];
    }
} dsu;

SegTree sir;
set<int> pos;

int main() {
    int n, d;

    cin >> n >> d;
    for ( int i = 1; i <= n; i++ )
        cin >> v[i], p[i] = i;

    sort( p + 1, p + 1 + n, []( int i, int j ) {
        if ( v[i] == v[j] )
            return i > j;
        return v[i] < v[j];
    } );

    dsu.init();
    pos.insert( 0 );
    pos.insert( n + 1 );
    for ( int ii = 1; ii <= n; ii++ ) {
        int i = p[ii];
        int len = sir.query( 0, 1, n, max( 1, i - d ), i - 1 ) + 1;
        sir.updatePos( 0, 1, n, i, len );

        auto ub = pos.upper_bound( i );
        auto lb = pos.lower_bound( i );
        lb--;

        if ( *ub <= n && *ub - i <= d )
            dsu.join( i, *ub );
        if ( *lb >= 1 && i - *lb <= d )
            dsu.join( i, *lb );

        sir.update( 0, 1, n, i + 1, dsu.getMax( i ), len );
        pos.insert( i );
    }

    cout << sir.query( 0, 1, n, n, n );

    return 0;
}

• Subtask 1: 14 / 14
• Subtask 2: 14 / 14
• Subtask 3: 20 / 20
• Subtask 4: 12 / 12
• Subtask 5: 5 / 5
• Subtask 6: 35 / 35