結果

問題 No.2755 行列の共役類
コンテスト
ユーザー 👑 p-adic
提出日時 2023-03-24 11:03:27
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 3,440 bytes
コンパイル時間 763 ms
コンパイル使用メモリ 75,656 KB
最終ジャッジ日時 2025-02-11 16:44:09
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16 RE * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

// C=1の時のみ有効
#pragma GCC optimize ( "O3" )
#pragma GCC optimize( "unroll-loops" )
#pragma GCC target ( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" )
#include <iostream>
#include <stdio.h>
#include <stdint.h>
#include <cassert>
#include <string>
using namespace std;

#define MAIN main
#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 QUIT return 0
#define COUT( ANSWER ) cout << ( ANSWER ) << "\n"
#define RETURN( ANSWER ) COUT( ANSWER ); QUIT

template <typename T> inline T Residue( const T& a , const T& p ){ return a >= 0 ? a % p : ( a % p ) + p; }

#define POWER_MOD( ANSWER , ARGUMENT , EXPONENT , MODULO )		\
  int ANSWER{ 1 };							\
  {									\
    int ARGUMENT_FOR_SQUARE_FOR_POWER = ( MODULO + ( ( ARGUMENT ) % MODULO ) ) % MODULO; \
    TYPE_OF( EXPONENT ) EXPONENT_FOR_SQUARE_FOR_POWER = ( EXPONENT );	\
    while( EXPONENT_FOR_SQUARE_FOR_POWER != 0 ){			\
      if( EXPONENT_FOR_SQUARE_FOR_POWER % 2 == 1 ){			\
	ANSWER = ( ANSWER * ARGUMENT_FOR_SQUARE_FOR_POWER ) % MODULO;	\
      }									\
      ARGUMENT_FOR_SQUARE_FOR_POWER = ( ARGUMENT_FOR_SQUARE_FOR_POWER * ARGUMENT_FOR_SQUARE_FOR_POWER ) % MODULO; \
      EXPONENT_FOR_SQUARE_FOR_POWER /= 2;				\
    }									\
  }									\

#define CONVERT1( i , P )			\
  int P ## 00 = i / B;			\
  coprime = true;				\
  FOR( n , 0 , num ){				\
    if( P ## 00 % prime[n] == 0 ){		\
      coprime = false;				\
      break;					\
    }						\
  }						\

#define CONVERT2( i , P )			\
  int P ## 01 = i % B;				\

int MAIN()
{
  UNTIE;
  CEXPR( int , bound_B , 600 );
  CIN_ASSERT( B , 1 , bound_B );
  CIN( int , C );
  // assert( 1 <= C && B % C == 0 );
  assert( C == 1 );
  int B_copy = B;
  int prime[4];
  int euler = 1;
  int num = 0;
  FOR( i , 2 , B_copy ){
    if( i * i > B_copy ){
      break;
    }
    if( B_copy % i == 0 ){
      prime[num++] = i;
      B_copy /= i;
      euler *= i - 1;
      while( B_copy % i == 0 ){
	B_copy /= i;
	euler *= i;
      }
    }
  }
  if( B_copy > 1 ){
    prime[num++] = B_copy;
    euler *= B_copy - 1;
  }
  int inv[bound_B];
  FOR( i , 1 , B ){
    POWER_MOD( i_inv , i , euler - 1 , B );
    inv[i] = i_inv;
  }
  CEXPR( int , bound_Bd , bound_B * bound_B );
  bool found[bound_Bd] = {};
  int B2 = B * B;
  int answer = 0;
  bool coprime;
  CEXPR( int , bound_answer , 100 );
  FOR( i , 0 , B2 ){
    if( ! found[i] ){
      CONVERT1( i , P );
      if( coprime ){
	answer++;
	if( answer > bound_answer ){
	  cout << bound_answer;
	  RETURN( "+" );
	}
	CONVERT2( i , P );
	FOR( j , 0 , B2 ){
	  CONVERT1( j , R );
	  if( coprime ){
	    CONVERT2( j , R );
	    int& detR_inv = inv[R00];
	    int Rinv01 = Residue( - detR_inv * R01 , B );
	    int Rinv00 = detR_inv;
	    int Q01 = ( Rinv00 * P01 + Rinv01 ) % B;
	    int Q00 = ( Rinv00 * P00 ) % B;
	    Q01 = ( Q00 * R01 + Q01 ) % B;
	    Q00 = ( Q00 * R00 ) % B;
	    int k = Q00 * B + Q01;
	    found[k] = true;
	  }
	}
      }
    }
  }
  RETURN( answer );
}
0