結果

問題 No.2125 Inverse Sum
ユーザー 👑 p-adicp-adic
提出日時 2022-11-19 00:30:38
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 3,263 bytes
コンパイル時間 2,040 ms
コンパイル使用メモリ 206,468 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-20 04:46:29
合計ジャッジ時間 4,209 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 RE -
testcase_04 AC 2 ms
5,376 KB
testcase_05 WA -
testcase_06 AC 2 ms
5,376 KB
testcase_07 RE -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 2 ms
5,376 KB
testcase_11 WA -
testcase_12 AC 2 ms
5,376 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 AC 2 ms
5,376 KB
testcase_24 RE -
testcase_25 RE -
testcase_26 AC 2 ms
5,376 KB
testcase_27 RE -
testcase_28 RE -
testcase_29 WA -
testcase_30 RE -
testcase_31 RE -
testcase_32 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using uint = unsigned int;
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 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 RETURN( ANSWER ) cout << ( ANSWER ) << "\n"; QUIT 

class NM
{
public :
  ll m_N;
  ll m_M;
  inline NM( const ll& N = 0 , const ll& M = 0 ) : m_N( N ) , m_M( M ) {};
};

inline bool operator<( const NM& NM0 , const NM& NM1 ) { return NM0.m_N < NM1.m_N; }

void SetPrimeFactorisation( const uint& n , vector<uint>& P , vector<uint>& exponent )
{

  uint n_copy = n;
  uint p = 2;

  if( n_copy % p == 0 ){

    P.push_back( p );
    exponent.push_back( 1 );
    uint& exponent_back = exponent.back();
    n_copy /= p;
    
    while( n_copy % p == 0 ){

      exponent_back++;
      n_copy /= p;
      
    }

  }

  p++;

  while( n_copy != 1 ){

    if( n_copy % p == 0 ){

      P.push_back( p );
      exponent.push_back( 1 );
      uint& exponent_back = exponent.back();
      n_copy /= p;
    
      while( n_copy % p == 0 ){

	exponent_back++;
	n_copy /= p;
      
      }

    }

    p += 2;

    if( p * p > n_copy ){

      break;

    }

  }

  return;
  
}

int main()
{
  CEXPR( ll , bound_PQ , 1000000000 );
  CIN_ASSERT( P , 1 , bound_PQ );
  CIN_ASSERT( Q , 1 , bound_PQ );
  vector<uint> prime{};
  vector<uint> exponent{};
  if( Q == 1 ){
    prime.push_back( 1 );
    exponent.push_back( 0 );
  } else {
    SetPrimeFactorisation( Q , prime , exponent );
  }
  uint size = exponent.size();
  ll count = 1;
  FOR( i , 0 , size ){
    uint& exponent_i = exponent[i];
    exponent_i *= 2;
    count *= exponent_i + 1;
  }
  ll Q2 = Q * Q;
  vector<uint> curr( size );
  ll div_N , div_M;
  uint T = 0;
  CEXPR( uint , rQ , 31622 );
  NM answer[rQ];
  FOR( c , 0 , count ){
    div_N = 1;
    FOR( i , 0 , size ){
      uint& curr_i = curr[i];
      uint& prime_i = prime[i];
      REPEAT( curr_i ){
	div_N *= prime_i;
      }
      div_M = Q2 / div_N;
      div_N += Q;
      div_M += Q;
      if( div_N % P == 0 && div_M % P == 0 ){
	answer[T] = NM( div_N / P , div_M / P );
	T++;
      }
    }
    uint i = 0;
    bool updating = Q != 1;
    while( updating ){
      uint& curr_i = curr[i];
      if( curr_i < exponent[i] ){
	curr_i++;
	updating = false;
      } else {
	curr_i = 0;
	i++;
	updating = i < size;
      }
    }
  }
  cout << T << "\n";
  sort( answer , answer + T );
  FOR( t , 0 , T ){
    NM& NMt = answer[t];
    cout << NMt.m_N << " " << NMt.m_M << "\n";
  }
  QUIT;
}
0