oj.uz

Submission #767029

# Submission time Handle Problem Language Result Execution time Memory
767029 2023-06-26T10:24:39 Z LucaIlie Homecoming (BOI18_homecoming) C++17
0 / 100
 1000 ms 35936 KB 
#include <bits/stdc++.h>
#include "homecoming.h"

using namespace std;

const long long INF = 1e16;
const int MAX_N = 2e6;

struct SegTree {
    vector<long long> segTree, lazy;

    void init( int n ) {
        segTree.resize( 4 * n );
        lazy.resize( 4 * n );
        for ( int v = 0; v < 4 * n; v++ ) {
            segTree[v] = -INF;
            lazy[v] = 0;
        }
    }

    void propag( int v, int l, int r ) {
        segTree[v] += lazy[v];
        if ( l != r ) {
            lazy[v * 2 + 1] += lazy[v];
            lazy[v * 2 + 2] += lazy[v];
        }
        lazy[v] = 0;
    }

    void update( int v, int l, int r, int lu, int ru, long long x ) {
        propag( v, l, r );

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

        if ( lu <= l && r <= ru ) {
            lazy[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] );
    }

    long long query( int v, int l, int r, int lq, int rq ) {
        propag( v, l, r );

        if ( l > rq || r < lq )
            return -INF;

        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 ) );
    }
};

SegTree dp;

long long solve( int n, int k, int a[], int b[] ) {
    long long ans = 0;

    for ( int take = 0; take <= 1; take++ ) {
        dp.init( n + k + 1 );
        if ( !take ) {
            int s = 0;
            for ( int j = 0; j <= k; j++ ) {
                dp.update( 0, 0, n + k, n + j, n + j, INF - s );
                s += b[j];
            }
        } else {
            for ( int j = 0; j <= k; j++ )
                dp.update( 0, 0, n + k, n + j, n + j, INF );
        }

        for ( int i = n - 1; i >= 0; i-- ) {
            if ( i >= k * take ) {
                long long x = 0;
                x = dp.query( 0, 0, n + k, i + 1, n + k );
                dp.update( 0, 0, n + k, i, i, INF + x );
            }

            dp.update( 0, 0, n + k, i + 1, n + k, -b[i] );
            dp.update( 0, 0, n + k, i + k, n + k, a[i] );
        }

        ans = max( ans, dp.query( 0, 0, n + k, 0, n + k ) );
    }

    return ans;
}

Subtask #1
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 212 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 18.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 212 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 31.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1033 ms 35936 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 38.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 212 KB Output isn't correct
2 Halted 0 ms 0 KB -