#822449 (C++17) No.2139 K Consecutive Sushi

• 問題: No.2139 K Consecutive Sushi
• ユーザー: 👑 p-adic
• 提出日時: 2022-12-04 11:42:21
• 言語: C++17 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 27 ms / 2,000 ms
• コード長: 1,694 bytes
• コンパイル時間: 3,443 ms
• コンパイル使用メモリ: 209,100 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-19 18:54:04
• 合計ジャッジ時間: 4,650 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	26 ms; 5,376 KB
testcase_04	AC	26 ms; 5,376 KB
testcase_05	AC	26 ms; 5,376 KB
testcase_06	AC	26 ms; 5,376 KB
testcase_07	AC	27 ms; 5,376 KB
testcase_08	AC	27 ms; 5,376 KB
testcase_09	AC	27 ms; 5,376 KB
testcase_10	AC	26 ms; 5,376 KB
testcase_11	AC	26 ms; 5,376 KB
testcase_12	AC	27 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	1 ms; 5,376 KB
testcase_18	AC	2 ms; 5,376 KB
testcase_19	AC	2 ms; 5,376 KB
testcase_20	AC	1 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB
testcase_22	AC	2 ms; 5,376 KB
testcase_23	AC	4 ms; 5,376 KB
testcase_24	AC	10 ms; 5,376 KB
testcase_25	AC	17 ms; 5,376 KB
testcase_26	AC	4 ms; 5,376 KB
testcase_27	AC	9 ms; 5,376 KB
testcase_28	AC	9 ms; 5,376 KB
testcase_29	AC	5 ms; 5,376 KB
testcase_30	AC	10 ms; 5,376 KB
testcase_31	AC	22 ms; 5,376 KB
testcase_32	AC	5 ms; 5,376 KB

ソースコード

#pragma GCC optimize ( "O3" )
#pragma GCC target ( "avx" )
#include <bits/stdc++.h>
using namespace std;

using ll = long long;

#define TYPE_OF( VAR ) remove_const<remove_reference<decltype( VAR )>::type >::type
#define UNTIE ios_base::sync_with_stdio( false ); cin.tie( nullptr ) 
#define CEXPR( LL , BOUND , VALUE ) constexpr const LL BOUND = VALUE 
#define CIN( LL , A ) LL A; cin >> A 
#define ASSERT( A , MIN , MAX ) assert( MIN <= A && A <= MAX ) 
#define CIN_ASSERT( A , MIN , MAX ) CIN( TYPE_OF( MAX ) , A ); ASSERT( A , MIN , MAX ) 
#define FOR( VAR , INITIAL , FINAL_PLUS_ONE ) for( TYPE_OF( FINAL_PLUS_ONE ) VAR = INITIAL ; VAR < FINAL_PLUS_ONE ; VAR ++ ) 
#define FOREQ( VAR , INITIAL , FINAL ) for( TYPE_OF( FINAL ) VAR = INITIAL ; VAR <= FINAL ; VAR ++ ) 
#define QUIT return 0 
#define RETURN( ANSWER ) cout << ( ANSWER ) << "\n"; QUIT 

#define UPDATE					\
  ll& dp_curr = dp[i];				\
  dp_curr = dp_min + Ai;			\
  if( dp_min < dp_curr ){			\
    if( dp_min == dp[i - K] ){			\
      dp_min = dp_curr;				\
      FOR( j , i - K + 1 , i ){			\
	const ll& dp_j = dp[j];			\
	if( dp_min > dp_j ){			\
	  dp_min = dp_j;			\
	}					\
      }						\
    }						\
  } else if( dp_min  > dp_curr ){		\
    dp_min = dp_curr;				\
  }						\
						\


int main()
{
  UNTIE;
  CEXPR( int , bound_N , 200000 );
  CIN_ASSERT( N , 2 , bound_N );
  CIN_ASSERT( K , 2 , N );
  ll dp[bound_N + 2];
  dp[0] = 0;
  ll sum = 0;
  CEXPR( int , bound_Ai , 1000000000 );
  FOR( i , 1 , K ){
    CIN_ASSERT( Ai , 1 , bound_Ai );
    dp[i] = Ai;
    sum += Ai;
  }
  ll dp_min = 0;
  FOREQ( i , K , N ){
    CIN_ASSERT( Ai , 1 , bound_Ai );
    UPDATE;
    sum += Ai;
  }
  RETURN( sum - dp_min );
}