結果

問題 No.2078 魔抜けなパープル
コンテスト
ユーザー 👑 p-adic
提出日時 2022-09-25 21:46:44
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,247 bytes
コンパイル時間 1,597 ms
コンパイル使用メモリ 192,336 KB
最終ジャッジ日時 2025-02-07 15:22:25
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 1 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

using uint = unsigned int;
using ll = long long;

#define UNTIE ios_base::sync_with_stdio( false ); cin.tie( nullptr ) 
#define CIN( LL , A ) LL A; cin >> A 
#define GETLINE( A ) string A; getline( cin , A ) 
#define GETLINE_SEPARATE( A , SEPARATOR ) string A; getline( cin , A , SEPARATOR ) 
#define FOR( VAR , INITIAL , FINAL_PLUS_ONE ) for( remove_const<remove_reference<decltype( FINAL_PLUS_ONE )>::type >::type VAR = INITIAL ; VAR < FINAL_PLUS_ONE ; VAR ++ ) 
#define FOREQ( VAR , INITIAL , FINAL ) for( remove_const<remove_reference<decltype( FINAL )>::type >::type VAR = INITIAL ; VAR <= FINAL ; VAR ++ )  
#define FOR_ITR( ARRAY , ITR , END ) for( auto ITR = ARRAY .begin() , END = ARRAY .end() ; ITR != END ; ITR ++ ) 
#define REPEAT( HOW_MANY_TIMES ) FOR( VARIABLE_FOR_REPEAT , 0 , HOW_MANY_TIMES ) 
#define QUIT return 0 
#define RETURN( ANSWER ) cout << ( ANSWER ) << "\n"; QUIT 
#define DOUBLE( PRECISION , ANSWER ) cout << fixed << setprecision( PRECISION ) << ( ANSWER ) << "\n"; QUIT 
#define MIN( A , B ) ( A < B ? A : B )
#define MAX( A , B ) ( A < B ? B : A )

template <typename T> inline T Absolute( const T& a ){ return a > 0 ? a : - a; }

int main()
{
  UNTIE;
  CIN( int , T );
  ll X;
  ll A;
  ll A_q;
  ll A_r;
  ll K;
  ll answer0;
  ll answer1;
  REPEAT( T ){
    cin >> X >> A;
    if( X == 0 ){
      cout << A << "\n";
    } else if( X == 1 ){
      cout << 2 * A << "\n";
    } else {
      K = 0;
      while( K * K < X ){
	K++;
      }
      K--;
      K--;
      answer0 = A * A + X;
      FOR( i , 0 , 3 ){
	if( K > 0 ){
	  A_q = A / K;
	  A_r = A % K;
	  answer1 = A_q * ( K * K + X ) + ( A_r == 0 ? 0 : A_r * A_r + X );
	  answer0 = MIN( answer0 , answer1 );
	  if( A_r <= A_q ){
	    answer1 = A_r * ( ( K + 1 ) * ( K + 1 ) + X ) + ( A_q - A_r ) * ( K * K + X );
	    answer0 = MIN( answer0 , answer1 );
	  } else {
	    answer1 = A_q * ( ( K + 1 ) * ( K + 1 ) + X ) + ( A_r - A_q ) * ( A_r - A_q ) + X;
	    answer0 = MIN( answer0 , answer1 );
	  }
	  if( A_q > 0 ){
	    A_q--;
	    A_r += K;
	    answer1 = A_q * ( K * K + X ) + A_r * A_r + X;
	    answer0 = MIN( answer0 , answer1 );
	  }
	}
	K++;
      }
      cout << answer0 << "\n";
    }
  }
  QUIT;
}
0