Submission #1453676


Source Code Expand

#include <bits/stdc++.h>

#define PB          push_back
#define PF          push_front
#define MP          make_pair
#define FI          first
#define SE          second
#define SIZE( A )   int( ( A ).size( ) )
#define ALL( A )    ( A ).begin( ), ( A ).end( )
#define ALLR( A )   ( A ).rbegin( ), ( A ).rend( )

using namespace std;

typedef long long           ll;
typedef unsigned long long  ull;
typedef long double         lf;
typedef pair< int, int >    pii;
typedef pair< ll, ll >      pll;
typedef vector< bool >      vb;
typedef vector< lf >        vd;
typedef vector< ll >        vll;
typedef vector< int >       vi;
typedef vector< pii >       vpii;

const int MAXN = int( 2e5 );
const int MOD  = int( 360 );
const int oo   = INT_MAX;

int n, k;
ll arr[ MAXN ];
ll acc_l[ MAXN ], acc_r[ MAXN ], acc[ MAXN ];

int main( ) {

  #ifdef LOCAL
    freopen( "input", "r", stdin );
    //freopen( "output", "w", stdout );
  #else
    //freopen( "input", "r", stdin );
    //freopen( "output", "w", stdout );
    ios_base::sync_with_stdio( 0 );
    cin.tie( 0 );
  #endif

  while( cin >> n >> k ) {
    acc[ 0 ] = 0;
    for( int i = 1; i <= n; ++i ) {
      cin >> arr[ i ];
      acc[ i ] = acc[ i-1 ]+arr[ i ];
    }
    acc_l[ 0 ] = 0;
    for( int i = 1; i <= n; ++i ) {
      acc_l[ i ] = acc_l[ i-1 ] + ( arr[ i ] > 0 ? arr[ i ] : 0 );
    }
    acc_r[ n+1 ] = 0;
    for( int i = n; i > 0; --i ) {
      acc_r[ i ] = acc_r[ i+1 ] + ( arr[ i ] > 0 ? arr[ i ] : 0 );
    }
    ll ans = 0;
    for( int i = 1; i+k-1 <= n; ++i ) {
      ll cur = acc_l[ i-1 ]+acc_r[ i+k ];
      ll val = acc[ i+k-1 ]-acc[ i-1 ];
      if( val > 0 ) {
        cur += val;
      }
      ans = max( ans, cur );
    }
    cout << ans << '\n';
  }

  return 0;
}

Submission Info

Submission Time
Task B - Contiguous Repainting
User vjudge1
Language C++14 (GCC 5.4.1)
Score 400
Code Size 1748 Byte
Status AC
Exec Time 14 ms
Memory 5632 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 400 / 400
Status
AC × 4
AC × 26
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt
Case Name Status Exec Time Memory
0_00.txt AC 2 ms 4352 KB
0_01.txt AC 2 ms 4352 KB
0_02.txt AC 2 ms 4352 KB
0_03.txt AC 2 ms 4352 KB
1_00.txt AC 2 ms 4352 KB
1_01.txt AC 2 ms 4352 KB
1_02.txt AC 2 ms 4352 KB
1_03.txt AC 9 ms 5632 KB
1_04.txt AC 13 ms 5632 KB
1_05.txt AC 14 ms 5632 KB
1_06.txt AC 13 ms 5632 KB
1_07.txt AC 13 ms 5632 KB
1_08.txt AC 13 ms 5632 KB
1_09.txt AC 13 ms 5632 KB
1_10.txt AC 13 ms 5632 KB
1_11.txt AC 13 ms 5632 KB
1_12.txt AC 12 ms 5632 KB
1_13.txt AC 13 ms 5632 KB
1_14.txt AC 13 ms 5632 KB
1_15.txt AC 13 ms 5632 KB
1_16.txt AC 13 ms 5632 KB
1_17.txt AC 12 ms 5632 KB
1_18.txt AC 13 ms 5632 KB
1_19.txt AC 12 ms 5632 KB
1_20.txt AC 13 ms 5632 KB
1_21.txt AC 13 ms 5632 KB