#856086 (C++17) No.3105 Міжнародний підрядок саміт

提出ソース

結果

問題	No.3105 Міжнародний підрядок саміт
ユーザー	👑 p-adic
提出日時	2023-04-02 17:05:54
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	RE (最新) AC (最初)
実行時間	-
コード長	6,362 bytes
コンパイル時間	3,245 ms
コンパイル使用メモリ	226,316 KB
実行使用メモリ	180,540 KB
最終ジャッジ日時	2024-10-12 02:10:53
合計ジャッジ時間	7,167 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	90 ms 180,400 KB
testcase_01	AC	1,796 ms 180,436 KB
testcase_02	RE	-
testcase_03	RE	-
testcase_04	RE	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#pragma GCC optimize ( "O3" )
#pragma GCC optimize( "unroll-loops" )
#pragma GCC target ( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" )
#include <bits/stdc++.h>
using namespace std;

using ll = long long;

#define ATT __attribute__( ( target( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" ) ) )
#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 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 FOREQINV( VAR , INITIAL , FINAL ) for( TYPE_OF( INITIAL ) VAR = INITIAL ; VAR >= FINAL ; VAR -- )
#define REPEAT( HOW_MANY_TIMES ) FOR( VARIABLE_FOR_REPEAT , 0 , HOW_MANY_TIMES )
#define QUIT return 0
#define COUT( ANSWER ) cout << ( ANSWER ) << "\n"
#define RETURN( ANSWER ) COUT( ANSWER ); QUIT

inline CEXPR( int , bound_N , 13 );
inline CEXPR( int , lim_B , 1 << bound_N );

// O(2^N)
struct Card
{
  int m_val[lim_B];
  constexpr Card() : m_val()
  {
    int two_power = 1;
    FOR( d , 0 , bound_N ){
      FOR( B , 0 , two_power ){
	m_val[B | two_power] = m_val[B] + 1;
      }
      two_power <<= 1;
    }
  }
};

inline CEXPR( int , lim_x_shift , bound_N * ( bound_N - 1 ) + 1 );
inline CEXPR( int , bound_x , lim_x_shift >> 1 );
inline CEXPR( int , bound_three_power , 1594323 ); // 3^13

// O(3^N)
struct XBP
{
  int m_x[bound_three_power];
  int m_B[bound_three_power];
  int m_p[bound_three_power];
  // constexpr XBP() : m_x() , m_B() , m_p()
  inline XBP() : m_x() , m_B() , m_p()
  {
    m_x[0] = bound_x;
    int three_power = 1;
    int three_power2 = 2;
    int two_power = 1;
    FOR( d , 0 , bound_N ){
      FOR( i , 0 , three_power ){
	int& xi = m_x[i];
	int i_plus = i + three_power;
	int i_plus2 = i + three_power2;
	m_x[i_plus] = xi - d;
	m_x[i_plus2] = xi + d;
	m_B[i_plus] = m_B[i_plus2] = m_B[i] | two_power;
	m_p[i_plus2] = ( m_p[i_plus] = m_p[i] ) + 1;
      }
      three_power = three_power2 + three_power;
      three_power2 = three_power << 1;
      two_power <<= 1;
    }
  }
};

// O(3^N)
struct X
{
  bool m_val[lim_B][bound_N+1][lim_x_shift];
  inline X( const int ( &x )[bound_three_power] , const int ( &B )[bound_three_power] , const int ( &p )[bound_three_power] ) : m_val()
  {
    FOR( i , 1 , bound_three_power ){
      m_val[B[i]][p[i]][x[i]] = true;
    }
  }
};

// O(N 3^N)
struct Xlr
{
  int m_val[2][lim_B][bound_N+1][lim_x_shift];
  inline Xlr( const int ( &card )[lim_B] , const bool ( &x )[lim_B][bound_N+1][lim_x_shift] ) : m_val()
  {
    int ( &xl )[lim_B][bound_N+1][lim_x_shift] = m_val[0];
    int ( &xr )[lim_B][bound_N+1][lim_x_shift] = m_val[1];
    FOR( B , 1 , lim_B ){
      const bool ( &xB )[bound_N+1][lim_x_shift] = x[B];
      const int& B_card = card[B];
      int ( &xlB )[bound_N+1][lim_x_shift] = xl[B];
      int ( &xrB )[bound_N+1][lim_x_shift] = xr[B];
      FOREQ( p , 0 , B_card ){
	const bool ( &xBp )[lim_x_shift] = xB[p];
	int ( &xlBp )[lim_x_shift] = xlB[p];
	int y_prev = lim_x_shift - 1;
	FOREQINV( y , lim_x_shift - 1 , 0 ){
	  if( xBp[y] ){
	    FOREQINV( z , y_prev , y ){
	      xlBp[z] = y - bound_x;
	    }
	    y_prev = y - 1;
	  }
	}
	FOREQINV( z , y_prev , 0 ){
	  xlBp[z] = bound_x + 1;
	}
	int ( &xrBp )[lim_x_shift] = xrB[p];
	y_prev = 0;
	FOR( y , 0 , lim_x_shift ){
	  if( xBp[y] ){
	    FOREQ( z , y_prev , y ){
	      xrBp[z] = y - bound_x;
	    }
	    y_prev = y + 1;
	  }
	}
	FOR( z , y_prev , lim_x_shift ){
	  xrBp[z] = bound_x + 1;
	}
      }
    }
  }
};
  
struct CombSum
{
  int m_val[bound_N+1];
  constexpr CombSum() : m_val()
  {
    FOREQ( N , 1 , bound_N ){
      if( ( N & 1 ) == 1 ){
	m_val[N] = 1 << ( N - 1 );
      } else {
	int& m_val_N = m_val[N];
	int comb = 1;
	FOREQ( p , 1 , N ){
	  ( comb *= ( N - p + 1 ) ) /= p;
	  if( ( p & 1 ) == 1 ){
	    m_val_N += comb;
	  }
	}
      }
    }
  }
};

int main()
{
  UNTIE;
  CEXPR( int , bound_T , 6000 );
  CIN_ASSERT( T , 1 , bound_T );
  CEXPR( int , bound_Pl , 100000000 );
  CEXPR( int , bound_Pr , 1000000000 );
  CEXPR( ll , bound_Ai , 1000000000 );
  CEXPR( ll , bound_evenness , ll( 1 ) << 62 );
  constexpr Card card{};
  // constexpr XBP xbp{}; // コンパイル時間制限30[s]をオーバー
  static XBP xbp{};
  static X x{ xbp.m_x , xbp.m_B , xbp.m_p };
  static Xlr xlr( card.m_val , x.m_val );
  int ( &xl )[lim_B][bound_N+1][lim_x_shift] = xlr.m_val[0];
  int ( &xr )[lim_B][bound_N+1][lim_x_shift] = xlr.m_val[1];
  constexpr CombSum comb_sum{};
  REPEAT( T ){
    CIN_ASSERT( N , 1 , bound_N );
    CIN_ASSERT( P , bound_Pl , bound_Pr );
    CIN_ASSERT( A0 , 1 , bound_Ai );
    CIN_ASSERT( A1 , A0 , bound_Ai );
    ll d = A1 - A0;
    FOR( i , 2 , N ){
      cin >> A1;
    }
    ll answer;
    if( d == 0 ){
      answer = comb_sum.m_val[N] * A0;
    } else {
      answer = 0;
      if( d < 0 ){
	d *= -1;
	A0 -= d * ( N - 1 );
      }
      int power_N = 1 << N;
      FOR( B , 1 , power_N ){
	int ( &xlB )[bound_N+1][lim_x_shift] = xl[B];
	int ( &xrB )[bound_N+1][lim_x_shift] = xr[B];
	const int& B_card = card.m_val[B];
	ll evenness = bound_evenness;
	FOREQ( p , 0 , B_card ){
	  int ( &xlBp )[lim_x_shift] = xlB[p];
	  int ( &xrBp )[lim_x_shift] = xrB[p];
	  ll A0_factor = ( B_card - ( p << 1 ) ) * A0;
	  ll y = A0_factor / d - ( ( A0_factor < 0 && A0_factor % d != 0 ) ? 1 : 0 ) + bound_x;
	  if( y >= lim_x_shift ){
	    y = lim_x_shift - 1;
	  } else if( y < 0 ){
	    y = 0;
	  }
	  int& yl = xlBp[y];
	  if( yl <= bound_x ){
	    ll evenness_curr = -A0_factor + yl * d ;
	    if( evenness_curr < 0 ){
	      evenness_curr *= -1;
	    }
	    if( evenness > evenness_curr ){
	      evenness = evenness_curr;
	    }
	  }
	  int& yr = xrBp[y];
	  if( yr <= bound_x ){
	    ll evenness_curr = -A0_factor + yl * d ;
	    if( evenness_curr < 0 ){
	      evenness_curr *= -1;
	    }
	    if( evenness > evenness_curr ){
	      evenness = evenness_curr;
	    }
	  }
	}
	answer += evenness;
      }
    }
    COUT( answer % P );
  }
  QUIT;
}

yukicoder

結果

テストケース

ソースコード