#pragma GCC optimize ( "O3" ) #pragma GCC target ( "avx" ) #include #include #include #include #include #include using namespace std; using ll = long long; #define TYPE_OF( VAR ) remove_const::type >::type #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 FOR_ITR( ARRAY , ITR , END ) for( auto ITR = ARRAY .begin() , END = ARRAY .end() ; ITR != END ; ITR ++ ) #define QUIT return 0 #define RETURN( ANSWER ) cout << ( ANSWER ) << "\n"; QUIT #include #define MAIN main // 通常の二分探索(単調関数-目的値が区間内に一意実数解を持つ場合にそれを超えない最大の整数を返す) #define BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , TARGET ) \ ll ANSWER = MAXIMUM; \ { \ ll VARIABLE_FOR_BINARY_SEARCH_L = MINIMUM; \ ll VARIABLE_FOR_BINARY_SEARCH_U = ANSWER; \ ll VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH = ( TARGET ) - ( EXPRESSION ); \ if( VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH == 0 ){ \ VARIABLE_FOR_BINARY_SEARCH_L = ANSWER; \ } else { \ ANSWER = ( VARIABLE_FOR_BINARY_SEARCH_L + VARIABLE_FOR_BINARY_SEARCH_U ) / 2; \ } \ while( VARIABLE_FOR_BINARY_SEARCH_L != ANSWER ){ \ VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH = ( TARGET ) - ( EXPRESSION ); \ if( VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH == 0 ){ \ break; \ } else { \ if( VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH > 0 ){ \ VARIABLE_FOR_BINARY_SEARCH_L = ANSWER; \ } else { \ VARIABLE_FOR_BINARY_SEARCH_U = ANSWER; \ } \ ANSWER = ( VARIABLE_FOR_BINARY_SEARCH_L + VARIABLE_FOR_BINARY_SEARCH_U ) / 2; \ } \ } \ } \ \ #define COMPUTE( VAR ) VAR ## 4 = VAR * VAR; VAR ## 3 = VAR ## 4 * VAR; VAR ## 4 *= VAR ## 4 #define COUNT_SOLUTION( NUM , VAR1 , VAR2 ) \ diff4 = bound_sum - VAR1 ## 4 - VAR2 ## 4; \ if( diff4 > 0 ){ \ BS( z_max , 1 , bound_xyz , z_max * z_max * z_max * z_max , diff4 ); \ diff3 = ( A - bound_A - VAR1 ## 3 - VAR2 ## 3 ) % bound_A + bound_A; \ vector& z_min_minus_curr = z_min_minus[diff3]; \ FOR_ITR( z_min_minus_curr , itr , end ){ \ answer ## NUM += ( z_max + *itr ) / bound_A; \ } \ } else { \ break; \ } \ int MAIN() { CEXPR( ll , bound_A , 333 ); CIN_ASSERT( A , 1 , bound_A ); CEXPR( ll , bound_xyz , 4444 ); CEXPR( ll , bound_sum , bound_xyz * bound_xyz * bound_xyz * bound_xyz ); static vector z_min_minus[bound_A] = {}; FOREQ( i , 1 , bound_A ){ z_min_minus[ ( i * i * i ) % bound_A ].push_back( int( bound_A - i ) ); } ll x = 1; ll y , x3 , y3 , x4 , y4 , diff3 , diff4; ll answer1 = 0; ll answer2 = 0; while( true ){ COMPUTE( x ); COUNT_SOLUTION( 1 , x , x ); y = x + 1; while( true ){ COMPUTE( y ); COUNT_SOLUTION( 2 , x , y ); y++; } x++; } RETURN( answer1 + answer2 * 2 ); }