#ifndef INCLUDE_MODE #define INCLUDE_MODE // #define REACTIVE // #define USE_GETLINE #endif #ifdef INCLUDE_MAIN IN VO Solve() { DEXPR( int , bound_Ai , 1e6 , 1e2 ); LeastDivisor least_divisor{}; CIN( int , N ); MultipleZetaTransform zt{ least_divisor , MP(1) , bound_Ai + 1 }; REPEAT( N ){ CIN_ASSERT( Ai , 1 , bound_Ai ); auto div = EnumerateDivisor( least_divisor , Ai ); auto range = [&]( const int& ) -> const vector& { return div; }; zt.Add( Ai , 1 + zt.InitialSegmentSum( 1 ) - zt.InverseImageSum( Id , range , 1 ) ); } RETURN( zt.InitialSegmentSum( 1 ) ); } REPEAT_MAIN(1); #else // INCLUDE_MAIN #ifdef INCLUDE_SUB // COMPAREに使用。圧縮時は削除する。 ll Naive( ll N , ll M , ll K ) { ll answer = N + M + K; return answer; } // COMPAREに使用。圧縮時は削除する。 ll Answer( ll N , ll M , ll K ) { // START_WATCH; ll answer = N + M + K; // // TLに準じる乱択や全探索。デフォルトの猶予は100.0[ms]。 // CEXPR( double , TL , 2000.0 ); // while( CHECK_WATCH( TL ) ){ // } return answer; } // 圧縮時は中身だけ削除する。 IN VO Experiment() { // CEXPR( int , bound , 10 ); // FOREQ( N , 0 , bound ){ // FOREQ( M , 0 , bound ){ // FOREQ( K , 0 , bound ){ // COUT( N , M , K , ":" , Naive( N , M , K ) ); // } // } // // cout << Naive( N ) << ",\n"[N==bound]; // } } // 圧縮時は中身だけ削除する。 IN VO SmallTest() { // CEXPR( int , bound , 10 ); // FOREQ( N , 0 , bound ){ // FOREQ( M , 0 , bound ){ // FOREQ( K , 0 , bound ){ // COMPARE( N , M , K ); // } // } // } } // 圧縮時は中身だけ削除する。 IN VO RandomTest() { // CEXPR( int , bound_N , 1e5 ); CIN_ASSERT( N , 1 , bound_N ); // CEXPR( ll , bound_M , 1e18 ); CIN_ASSERT( M , 1 , bound_M ); // CEXPR( ll , bound_K , 1e9 ); CIN_ASSERT( K , 1 , bound_K ); // COMPARE( N , M , N ); } #define INCLUDE_MAIN #include __FILE__ #else // INCLUDE_SUB #ifdef INCLUDE_LIBRARY /* C-x 3 C-x o C-x C-fによるファイル操作用 BFS (5KB) c:/Users/user/Documents/Programming/Mathematics/Geometry/Graph/BreadthFirstSearch/compress.txt CoordinateCompress (3KB) c:/Users/user/Documents/Programming/Mathematics/SetTheory/DirectProduct/CoordinateCompress/compress.txt DFSOnTree (11KB) c:/Users/user/Documents/Programming/Mathematics/Geometry/Graph/DepthFirstSearch/Tree/a.hpp Divisor (4KB) c:/Users/user/Documents/Programming/Mathematics/Arithmetic/Prime/Divisor/compress.txt IntervalAddBIT (9KB) c:/Users/user/Documents/Programming/Mathematics/SetTheory/DirectProduct/AffineSpace/BIT/IntervalAdd/compress.txt Polynomial (21KB) c:/Users/user/Documents/Programming/Mathematics/Polynomial/compress.txt UnionFind (3KB) c:/Users/user/Documents/Programming/Mathematics/Geometry/Graph/UnionFindForest/compress.txt */ // VVV 常設でないライブラリは以下に挿入する。 #ifdef DEBUG #include "c:/Users/user/Documents/Programming/Mathematics/Combinatorial/ZetaTransform/Divisor/a_Body.hpp" #else TE CL VirtualRSet:VI PU UnderlyingSet{PU:VI U Action(CO R& r,U u)= 0;IN U PW(U u,CO R& r);IN U ScalarProduct(CO R& r,U u);};TE CL RegularRSet:VI PU VirtualRSet,PU MAGMA{PU:IN RegularRSet(MAGMA magma);IN U Action(CO U& r,U u);};TE RegularRSet(MAGMA magma)-> RegularRSet,MAGMA>;TE CL AbstractRSet:VI PU VirtualRSet{PU:O_U m_o_U;IN AbstractRSet(CO R& dummy0,CO U& dummy1,O_U o_U);IN U Action(CO R& r,U u);};TE CL AbstractModule:PU AbstractRSet,PU GROUP{PU:IN AbstractModule(CO R& dummy,O_U o_U,GROUP M);};TE AbstractModule(CO R& dummy,O_U o_U,GROUP M)-> AbstractModule,O_U,GROUP>;TE CL Module:VI PU VirtualRSet,PU AdditiveGroup{PU:IN U Action(CO R& r,U u);}; TE IN RegularRSet::RegularRSet(MAGMA magma):MAGMA(MO(magma)){}TE IN AbstractRSet::AbstractRSet(CO R& dummy0,CO U& dummy1,O_U o_U):m_o_U(MO(o_U)){ST_AS(is_invocable_r_v);}TE IN AbstractModule::AbstractModule(CO R& dummy,O_U o_U,GROUP M):AbstractRSet(dummy,M.One(),MO(o_U)),GROUP(MO(M)){ST_AS(is_same_v>);}TE IN U RegularRSet::Action(CO U& r,U u){RE TH->Product(r,MO(u));}TE IN U AbstractRSet::Action(CO R& r,U u){RE m_o_U(r,MO(u));}TE IN U Module::Action(CO R& r,U u){RE MO(u *= r);}TE IN U VirtualRSet::PW(U u,CO R& r){RE Action(r,MO(u));}TE IN U VirtualRSet::ScalarProduct(CO R& r,U u){RE Action(r,MO(u));} TE CL VirtualSemirng{PU:VI U Sum(CO U& u0,CO U& u1)= 0;VI CO U& Zero()CO NE = 0;VI U Product(CO U& u0,CO U& u1)= 0;VI MONOID& AdditiveMonoid()NE = 0;VI SEMIGROUP& MultiplicativeSemigroup()NE = 0;US type = U;};TE CL AbstractSemirng:VI PU VirtualSemirng{PU:MONOID m_R0;SEMIGROUP m_R1;IN AbstractSemirng(MONOID R0,SEMIGROUP R1);IN U Sum(CO U& u0,CO U& u1);IN CO U& Zero()CO NE;IN U Product(CO U& u0,CO U& u1);IN MONOID& AdditiveMonoid()NE;IN SEMIGROUP& MultiplicativeSemigroup()NE;};TE CL Semirng:PU AbstractSemirng,MultiplicativeMagma>{PU:IN Semirng();}; TE IN AbstractSemirng::AbstractSemirng(MONOID R0,SEMIGROUP R1):m_R0(MO(R0)),m_R1(MO(R1)){}TE IN Semirng::Semirng():AbstractSemirng,MultiplicativeMagma>(AdditiveMonoid(),MultiplicativeMagma()){}TE IN U AbstractSemirng::Sum(CO U& u0,CO U& u1){RE m_R0.Sum(u0,u1);}TE IN CO U& AbstractSemirng::Zero()CO NE{RE m_R0.Zero();}TE IN U AbstractSemirng::Product(CO U& u0,CO U& u1){RE m_R1.Product(u0,u1);}TE IN MONOID& AbstractSemirng::AdditiveMonoid()NE{RE m_R0;}TE IN SEMIGROUP& AbstractSemirng::MultiplicativeSemigroup()NE{RE m_R1;} TE CL VirtualRing:VI PU VirtualSemirng{PU:VI U Inverse(CO U& u)= 0;VI CO U& One()CO NE = 0;IN GROUP& AdditiveGroup()NE;IN MONOID& MultiplicativeMonoid()NE;};TE CL AbstractRing:VI PU VirtualRing,PU AbstractSemirng{PU:IN AbstractRing(GROUP R0,MONOID R1);IN U Inverse(CO U& u);IN CO U& One()CO NE;};TE CL Ring:PU AbstractRing,MultiplicativeMonoid>{PU:IN Ring(CO U& one_U);}; TE IN AbstractRing::AbstractRing(GROUP R0,MONOID R1):AbstractSemirng(MO(R0),MO(R1)){}TE IN Ring::Ring(CO U& one_U):AbstractRing,MultiplicativeMonoid>(AdditiveGroup(),MultiplicativeMonoid(one_U)){}TE IN U AbstractRing::Inverse(CO U& u){RE TH->m_R0.Inverse(u);}TE IN CO U& AbstractRing::One()CO NE{RE TH->m_R1.One();}TE IN GROUP& VirtualRing::AdditiveGroup()NE{RE TH->AdditiveMonoid();}TE IN MONOID& VirtualRing::MultiplicativeMonoid()NE{RE TH->MultiplicativeSemigroup();} TE CL Algebra:VI PU VirtualRSet,PU Ring{PU:IN Algebra(U one);US Ring::type;IN U Action(CO R& r,U u);IN U PW(U u,CO R& r)= delete;}; TE IN Algebra::Algebra(U one):Ring(MO(one)){}TE IN U Algebra::Action(CO R& r,U u){RE MO(u *= r);} TE CL VirtualZetaTransform{PU:GRAPH m_G;GRAPH_INV m_G_inv;Z_ALG m_R;VE m_val;IN VirtualZetaTransform(GRAPH G,GRAPH_INV G_inv,Z_ALG R);IN VirtualZetaTransform(GRAPH G,GRAPH_INV G_inv,Z_ALG R,VE a,CO bool& transformed = false);TE IN VO Initialise(Args&&... args);IN VO Add(CO T& t,CO U& u);IN VO TotalAdd(CO U& u);IN VirtualZetaTransform& OP+=(CO VirtualZetaTransform& a);IN VO TotalMultiply(CO U& u);IN VirtualZetaTransform& OP*=(CO VirtualZetaTransform& a);U OP[](CO T& t);IN U Get(CO T& t);IN CO U& InitialSegmentSum(CO T& t);TE U InverseImageSum(F_INV_MAX&& f_inv_max,RANGE&& range,CO S& s);TE IN CO U& InitialSegmentInverseImageSum(F_INV_MAX&& f_inv_max,CO S& s);VI int Moevius(CO T& t0,CO T& t1);};TE VirtualZetaTransform(GRAPH&,GRAPH_INV&,Z_ALG)-> VirtualZetaTransform,GRAPH,GRAPH_INV,inner_t,Z_ALG>;TE CL AbstractZetaTransform:PU VirtualZetaTransform{PU:MU m_mu;TE IN AbstractZetaTransform(GRAPH G,GRAPH_INV G_inv,Z_ALG R,MU mu,Args&&... args);IN int Moevius(CO T& t0,CO T& t1);};TE AbstractZetaTransform(GRAPH&,GRAPH_INV&,Z_ALG,MU)-> AbstractZetaTransform,GRAPH,GRAPH_INV,inner_t,Z_ALG,MU>; TE IN VirtualZetaTransform::VirtualZetaTransform(GRAPH G,GRAPH_INV G_inv,Z_ALG R):m_G(MO(G)),m_G_inv(MO(G_inv)),m_R(MO(R)),m_val(m_G.SZ(),m_R.Zero()){ST_AS(is_same_v> && is_same_v> && is_same_v>);AS(m_G_inv.SZ()== m_G.SZ());}TE IN VirtualZetaTransform::VirtualZetaTransform(GRAPH G,GRAPH_INV G_inv,Z_ALG R,VE a,CO bool& transformed):m_G(MO(G)),m_G_inv(MO(G_inv)),m_R(MO(R)),m_val(MO(a)){ST_AS(is_same_v> && is_same_v> && is_same_v>);CRI SZ = m_G.SZ();AS(m_G_inv.SZ()== SZ && int(m_val.SZ())== SZ);if(!transformed){a = m_val;for(int i = 0;i < SZ;i++){U& m_val_i = m_val[i];auto&& sub_i = m_G.Edge(i);for(auto& j:sub_i){j == i?m_val_i:m_val_i = m_R.Sum(MO(m_val_i),a[j]);}}}}TE TE IN AbstractZetaTransform::AbstractZetaTransform(GRAPH G,GRAPH_INV G_inv,Z_ALG R,MU mu,Args&&... args):VirtualZetaTransform(MO(G),MO(G_inv),MO(R),forward(args)...),m_mu(MO(mu)){ST_AS(is_invocable_r_v);}TE TE VO VirtualZetaTransform::Initialise(Args&&... args){VirtualZetaTransform temp{m_G,m_G_inv,m_R,forward(args)...};m_val = MO(temp.m_val);}TE VO VirtualZetaTransform::Add(CO T& t,CO U& u){auto&& sup = m_G_inv.Edge(t);for(auto& s:sup){U& m_val_i = m_val[m_G.Enumeration_inv(s)];m_val_i = m_R.Sum(MO(m_val_i),u);}}TE IN VO VirtualZetaTransform::TotalAdd(CO U& u){CRI SZ = m_G.SZ();for(int i = 0;i < SZ;i++){U& m_val_i = m_val[i];m_val_i = m_R.Sum(MO(m_val_i),m_R.ScalarProduct(m_G.Edge(m_G.Enumeration(i)).SZ(),u));}}TE IN VirtualZetaTransform& VirtualZetaTransform::OP+=(CO VirtualZetaTransform& a){CRI SZ = m_G.SZ();for(int i = 0;i < SZ;i++){U& m_val_i = m_val[i];m_val_i = Sum(MO(m_val_i),a.m_val[i]);}RE *TH;}TE IN VO VirtualZetaTransform::TotalMultiply(CO U& u){CRI SZ = m_G.SZ();for(int i = 0;i < SZ;i++){U& m_val_i = m_val[i];m_val_i = m_R.Product(MO(m_val_i),u);}}TE IN VirtualZetaTransform& VirtualZetaTransform::OP*=(CO VirtualZetaTransform& a){CRI SZ = m_G.SZ();for(int i = 0;i < SZ;i++){U& m_val_i = m_val[i];m_val_i = m_R.Product(MO(m_val_i),a.m_val[i]);}RE *TH;}TE U VirtualZetaTransform::OP[](CO T& t){auto&& sub = m_G.Edge(t);U AN = m_R.Zero();CRI SZ = m_G.SZ();for(auto& s:sub){auto&& i = m_G.Enumeration_inv(s);AS(i < SZ);AN = m_R.Sum(MO(AN),m_R.ScalarProduct(Moevius(s,t),m_val[i]));}RE AN;}TE IN U VirtualZetaTransform::Get(CO T& t){RE OP[](t);}TE IN CO U& VirtualZetaTransform::InitialSegmentSum(CO T& t){auto&& i = m_G.Enumeration_inv(t);AS(i < m_G.SZ());RE m_val[i];}TE TE U VirtualZetaTransform::InverseImageSum(F_INV_MAX&& f_inv_max,RANGE&& range,CO S& s){ST_AS(is_invocable_r_v && is_invocable_r_v,RANGE,CO S&>);auto&& t = f_inv_max(s);auto&& sub = range(s);U AN = m_R.Zero();CRI SZ = m_G.SZ();for(auto& s_sub:sub){auto&& t_sub = f_inv_max(s_sub);auto&& i = m_G.Enumeration_inv(t_sub);AS(i < SZ);AN = m_R.Sum(MO(AN),m_R.ScalarProduct(Moevius(t_sub,t),m_val[i]));}RE AN;}TE TE IN CO U& VirtualZetaTransform::InitialSegmentInverseImageSum(F_INV_MAX&& f_inv_max,CO S& s){RE m_val[m_G.Enumeration_inv(f_inv_max(s))];}TE int VirtualZetaTransform::Moevius(CO T& t0,CO T& t1){ST VE> memory(m_G.SZ());auto&& i = m_G.Enumeration_inv(t0);auto&& j = m_G.Enumeration_inv(t1);unordered_map& memory_t0 = memory[i];CO bool found = memory_t0.count(j)== 1;int& AN = memory_t0[j];if(! found){if(i == j){AN = 1;}else{AN = 0;auto&& sub = m_G.Edge(t1);for(auto& s:sub){s == t1?AN:AN -= Moevius(t0,s);}}}RE AN;}TE IN int AbstractZetaTransform::Moevius(CO T& t0,CO T& t1){RE m_mu(t0,t1);} TE CL LeastDivisor{PU:INT m_val[val_limit];CE LeastDivisor()NE;IN CO INT& OP[](CRI i)CO;CE CO INT& Get(CRI i)CO;}; TE CE LeastDivisor::LeastDivisor()NE:m_val{}{for(int d = 2;d < val_limit;d++){if(m_val[d]== 0){for(int n = d;n < val_limit;n += d){m_val[n]== 0?m_val[n]= d:d;}}}}TE IN CO INT& LeastDivisor::OP[](CRI i)CO{AS(0 <= i && i < val_limit);RE m_val[i];}TE CE CO INT& LeastDivisor::Get(CRI i)CO{RE m_val[i];} TE INT CountDivisorBody(VE& EX)NE{CO int LE = EX.SZ();INT AN = 1;for(int i = 0;i < LE;i++){AN *= ++EX[i];}RE AN;}TE VE EnumerateDivisorBody(CO VE& P,VE& EX){CO int LE = P.SZ();VE AN(CountDivisorBody(EX),1);int SZ = 1;for(int i = 0;i < LE;i++){CO INT& P_i = P[i];CRI EX_i = EX[i];INT PW = 1;int j_shift = 0;for(int e = 1;e < EX_i;e++){PW *= P_i;j_shift += SZ;for(int j = 0;j < SZ;j++){AN[j + j_shift]= AN[j]* PW;}}SZ *= EX_i;}RE AN;}TE VE EnumerateDivisor(CO LeastDivisor& ld,INT2 n){VE P{},EX{};WH(n > 1){auto& p = ld[n];P.push_back(p);INT2 e = 1;WH((n /= p)% p == 0){e++;}EX.push_back(e);}RE EnumerateDivisorBody(P,EX);} TE CL MoeviusFunction{PU:INT m_val[val_limit];CE MoeviusFunction(CO LeastDivisor& ld)NE;IN CO INT& OP()(CRI n)CO;}; TE CE MoeviusFunction::MoeviusFunction(CO LeastDivisor& ld)NE:m_val{0,1}{for(int i = 2;i < val_limit;i++){auto& p = ld[i];CO int j = i / p;m_val[i]= j % p == 0?0:-m_val[j];}}TE IN CO INT& MoeviusFunction::OP()(CRI n)CO{AS(0 <= n && n < val_limit);RE m_val[n];} TE CL DivisorMoeviusFunction{PU:CO MU* m_p_mu;CE DivisorMoeviusFunction(CO MU& mu);IN int OP()(CRI t0,CRI t1);};TE CL MultipleMoeviusFunction{PU:CO MU* m_p_mu;CE MultipleMoeviusFunction(CO MU& mu);IN int OP()(CRI t0,CRI t1);};TE CL DivisorEdge{PU:CO LD* m_p_ld;IN DivisorEdge(CO LD& ld);IN VE OP()(CRI t);};CL MultipleEdge{PU:int m_SZ;IN MultipleEdge(CRI SZ);IN VE OP()(CRI t);};TE CL AbstractDivisorZetaTransform:PU AbstractZetaTransform>,Graph,U,Z_ALG,DivisorMoeviusFunction>{PU:MU m_mu;IN AbstractDivisorZetaTransform(CO LD& ld,Z_ALG R,CRI SZ = 0);IN AbstractDivisorZetaTransform(CO LD& ld,Z_ALG R,VE a,CO bool& transformed = false);IN AbstractDivisorZetaTransform(CO LD& ld,Z_ALG R,CRI SZ,VE& a,CO bool& transformed);};TE AbstractDivisorZetaTransform(CO LeastDivisor&,Z_ALG,Args&&...)-> AbstractDivisorZetaTransform,MoeviusFunction,inner_t,Z_ALG>;TE CL DivisorZetaTransform:PU AbstractDivisorZetaTransform,MoeviusFunction,U,Algebra>{PU:TE IN DivisorZetaTransform(CO LeastDivisor& ld,CO U& one,Args&&... args);};TE CL AbstractMultipleZetaTransform:PU AbstractZetaTransform,Graph>,U,Z_ALG,MultipleMoeviusFunction>{PU:MU m_mu;IN AbstractMultipleZetaTransform(CO LD& ld,Z_ALG R,CRI SZ = 0);IN AbstractMultipleZetaTransform(CO LD& ld,Z_ALG R,VE a,CO bool& transformed = false);IN AbstractMultipleZetaTransform(CO LD& ld,Z_ALG R,CRI SZ,VE& a,CO bool& transformed);};TE AbstractMultipleZetaTransform(CO LeastDivisor&,Z_ALG,Args&&...)-> AbstractMultipleZetaTransform,MoeviusFunction,inner_t,Z_ALG>;TE CL MultipleZetaTransform:PU AbstractMultipleZetaTransform,MoeviusFunction,U,Algebra>{PU:TE IN MultipleZetaTransform(CO LeastDivisor& ld,CO U& one,Args&&... args);}; TE CE DivisorMoeviusFunction::DivisorMoeviusFunction(CO MU& mu):m_p_mu(&mu){}TE CE MultipleMoeviusFunction::MultipleMoeviusFunction(CO MU& mu):m_p_mu(&mu){}TE IN int DivisorMoeviusFunction::OP()(CRI t0,CRI t1){RE(*m_p_mu)(t1 / t0);}TE IN int MultipleMoeviusFunction::OP()(CRI t0,CRI t1){RE(*m_p_mu)(t0 / t1);}TE IN DivisorEdge::DivisorEdge(CO LD& ld):m_p_ld(&ld){}IN MultipleEdge::MultipleEdge(CRI SZ):m_SZ(SZ){}TE IN VE DivisorEdge::OP()(CRI t){AS(0 <= t);VE AN{};RE t == 0?VE(1):EnumerateDivisor(*m_p_ld,t);}IN VE MultipleEdge::OP()(CRI t){AS(0 <= t && t < m_SZ);CO int SZ = t == 0?0:(m_SZ - 1)/ t;VE AN(SZ);for(int i = 0;i < SZ;i++){AN[i]= t *(i + 1);}RE AN;}TE IN AbstractDivisorZetaTransform::AbstractDivisorZetaTransform(CO LD& ld,Z_ALG R,CRI SZ):AbstractZetaTransform>,Graph,U,Z_ALG,DivisorMoeviusFunction>(Graph(SZ,DivisorEdge(ld)),Graph(SZ,MultipleEdge(SZ)),MO(R),DivisorMoeviusFunction(m_mu)),m_mu(ld){}TE IN AbstractDivisorZetaTransform::AbstractDivisorZetaTransform(CO LD& ld,Z_ALG R,VE a,CO bool& transformed):AbstractDivisorZetaTransform(ld,MO(R),a.SZ(),a,transformed){}TE IN AbstractDivisorZetaTransform::AbstractDivisorZetaTransform(CO LD& ld,Z_ALG R,CRI SZ,VE& a,CO bool& transformed):AbstractZetaTransform>,Graph,U,Z_ALG,DivisorMoeviusFunction>(Graph(SZ,DivisorEdge(ld)),Graph(SZ,MultipleEdge(SZ)),MO(R),DivisorMoeviusFunction(m_mu),MO(a),transformed),m_mu(ld){}TE TE IN DivisorZetaTransform::DivisorZetaTransform(CO LeastDivisor& ld,CO U& one,Args&&... args):AbstractDivisorZetaTransform,MoeviusFunction,U,Algebra>(ld,Algebra(one),forward(args)...){}TE IN AbstractMultipleZetaTransform::AbstractMultipleZetaTransform(CO LD& ld,Z_ALG R,CRI SZ):AbstractZetaTransform,Graph>,U,Z_ALG,MultipleMoeviusFunction>(Graph(SZ,MultipleEdge(SZ)),Graph(SZ,DivisorEdge(ld)),MO(R),MultipleMoeviusFunction(m_mu)),m_mu(ld){}TE IN AbstractMultipleZetaTransform::AbstractMultipleZetaTransform(CO LD& ld,Z_ALG R,VE a,CO bool& transformed):AbstractMultipleZetaTransform(ld,MO(R),a.SZ(),a,transformed){}TE IN AbstractMultipleZetaTransform::AbstractMultipleZetaTransform(CO LD& ld,Z_ALG R,CRI SZ,VE& a,CO bool& transformed):AbstractZetaTransform,Graph>,U,Z_ALG,MultipleMoeviusFunction>(Graph(SZ,MultipleEdge(ld)),Graph(SZ,DivisorEdge(ld)),MO(R),MultipleMoeviusFunction(m_mu),MO(a),transformed),m_mu(ld){}TE TE IN MultipleZetaTransform::MultipleZetaTransform(CO LeastDivisor& ld,CO U& one,Args&&... args):AbstractMultipleZetaTransform,MoeviusFunction,U,Algebra>(ld,Algebra(one),forward(args)...){} #endif // AAA 常設でないライブラリは以上に挿入する。 #define INCLUDE_SUB #include __FILE__ #else // INCLUDE_LIBRARY #ifdef DEBUG #define _GLIBCXX_DEBUG #define REPEAT_MAIN( BOUND ) START_MAIN; signal( SIGABRT , &AlertAbort ); AutoCheck( exec_mode , use_getline ); CEXPR( int , bound_test_case_num , BOUND ); int test_case_num = 1; if( exec_mode == solve_mode ){ if CE( bound_test_case_num > 1 ){ CERR( "テストケースの個数を入力してください。" ); SET_ASSERT( test_case_num , 1 , bound_test_case_num ); } } else { if( exec_mode == experiment_mode ){ Experiment(); } else if( exec_mode == small_test_mode ){ SmallTest(); } else if( exec_mode == random_test_mode ){ CERR( "ランダムテストを行う回数を指定してください。" ); SET_LL( test_case_num ); REPEAT( test_case_num ){ RandomTest(); } } RE 0; } FINISH_MAIN #define DEXPR( LL , BOUND , VALUE1 , VALUE2 ) CEXPR( LL , BOUND , VALUE2 ) #define ASSERT( A , MIN , MAX ) CERR( "ASSERTチェック: " , ( MIN ) , ( ( MIN ) <= A ? "<=" : ">" ) , A , ( A <= ( MAX ) ? "<=" : ">" ) , ( MAX ) ); AS( ( MIN ) <= A && A <= ( MAX ) ) #define SET_ASSERT( A , MIN , MAX ) if( exec_mode == solve_mode ){ SET_LL( A ); ASSERT( A , MIN , MAX ); } else if( exec_mode == random_test_mode ){ CERR( #A , " = " , ( A = GetRand( MIN , MAX ) ) ); } else { AS( false ); } #define SOLVE_ONLY ST_AS( __FUNCTION__[0] == 'S' ) #define CERR( ... ) VariadicCout( cerr , __VA_ARGS__ ) << endl #define COUT( ... ) VariadicCout( cout << "出力: " , __VA_ARGS__ ) << endl #define CERR_A( A , N ) OUTPUT_ARRAY( cerr , A , N ) << endl #define COUT_A( A , N ) cout << "出力: "; OUTPUT_ARRAY( cout , A , N ) << endl #define CERR_ITR( A ) OUTPUT_ITR( cerr , A ) << endl #define COUT_ITR( A ) cout << "出力: "; OUTPUT_ITR( cout , A ) << endl #else #pragma GCC optimize ( "O3" ) #pragma GCC optimize ( "unroll-loops" ) #pragma GCC target ( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" ) #define REPEAT_MAIN( BOUND ) START_MAIN; CEXPR( int , bound_test_case_num , BOUND ); int test_case_num = 1; if CE( bound_test_case_num > 1 ){ SET_ASSERT( test_case_num , 1 , bound_test_case_num ); } FINISH_MAIN #define DEXPR( LL , BOUND , VALUE1 , VALUE2 ) CEXPR( LL , BOUND , VALUE1 ) #define ASSERT( A , MIN , MAX ) AS( ( MIN ) <= A && A <= ( MAX ) ) #define SET_ASSERT( A , MIN , MAX ) SET_LL( A ); ASSERT( A , MIN , MAX ) #define SOLVE_ONLY #define CERR( ... ) #define COUT( ... ) VariadicCout( cout , __VA_ARGS__ ) << ENDL #define CERR_A( A , N ) #define COUT_A( A , N ) OUTPUT_ARRAY( cout , A , N ) << ENDL #define CERR_ITR( A ) #define COUT_ITR( A ) OUTPUT_ITR( cout , A ) << ENDL #endif #ifdef REACTIVE #define ENDL endl #else #define ENDL "\n" #endif #ifdef USE_GETLINE #define SET_LL( A ) { GETLINE( A ## _str ); A = stoll( A ## _str ); } #define GETLINE_SEPARATE( SEPARATOR , ... ) SOLVE_ONLY; string __VA_ARGS__; VariadicGetline( cin , SEPARATOR , __VA_ARGS__ ) #define GETLINE( ... ) SOLVE_ONLY; GETLINE_SEPARATE( '\n' , __VA_ARGS__ ) #else #define SET_LL( A ) cin >> A #define CIN( LL , ... ) SOLVE_ONLY; LL __VA_ARGS__; VariadicCin( cin , __VA_ARGS__ ) #define SET_A( A , N ) SOLVE_ONLY; FOR( VARIABLE_FOR_SET_A , 0 , N ){ cin >> A[VARIABLE_FOR_SET_A]; } #define CIN_A( LL , A , N ) VE A( N ); SET_A( A , N ); #endif #include using namespace std; #define ATT __attribute__( ( target( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" ) ) ) #define START_MAIN int main(){ ios_base::sync_with_stdio( false ); cin.tie( nullptr ) #define FINISH_MAIN REPEAT( test_case_num ){ if CE( bound_test_case_num > 1 ){ CERR( "testcase " , VARIABLE_FOR_REPEAT_test_case_num , ":" ); } Solve(); CERR( "" ); } } #define START_WATCH chrono::system_clock::time_point watch = chrono::system_clock::now() #define CURRENT_TIME static_cast( chrono::duration_cast( chrono::system_clock::now() - watch ).count() / 1000.0 ) #define CHECK_WATCH( TL_MS ) ( CURRENT_TIME < TL_MS - 100.0 ) #define CEXPR( LL , BOUND , VALUE ) CE LL BOUND = VALUE #define SET_A_ASSERT( A , N , MIN , MAX ) FOR( VARIABLE_FOR_SET_A , 0 , N ){ SET_ASSERT( A[VARIABLE_FOR_SET_A] , MIN , MAX ); } #define CIN_ASSERT( A , MIN , MAX ) decldecay_t( MAX ) A; SET_ASSERT( A , MIN , MAX ) #define CIN_A_ASSERT( A , N , MIN , MAX ) vector A( N ); SET_A_ASSERT( A , N , MIN , MAX ) #define FOR( VAR , INITIAL , FINAL_PLUS_ONE ) for( decldecay_t( FINAL_PLUS_ONE ) VAR = INITIAL ; VAR < FINAL_PLUS_ONE ; VAR ++ ) #define FOREQ( VAR , INITIAL , FINAL ) for( decldecay_t( FINAL ) VAR = INITIAL ; VAR <= FINAL ; VAR ++ ) #define FOREQINV( VAR , INITIAL , FINAL ) for( decldecay_t( INITIAL ) VAR = INITIAL ; VAR + 1 > FINAL ; VAR -- ) #define AUTO_ITR( ARRAY ) auto itr_ ## ARRAY = ARRAY .BE() , end_ ## ARRAY = ARRAY .EN() #define FOR_ITR( ARRAY ) for( AUTO_ITR( ARRAY ) , itr = itr_ ## ARRAY ; itr_ ## ARRAY != end_ ## ARRAY ; itr_ ## ARRAY ++ , itr++ ) #define REPEAT( HOW_MANY_TIMES ) FOR( VARIABLE_FOR_REPEAT_ ## HOW_MANY_TIMES , 0 , HOW_MANY_TIMES ) #define SET_PRECISION( DECIMAL_DIGITS ) cout << fixed << setprecision( DECIMAL_DIGITS ) #define OUTPUT_ARRAY( OS , A , N ) FOR( VARIABLE_FOR_OUTPUT_ARRAY , 0 , N ){ OS << A[VARIABLE_FOR_OUTPUT_ARRAY] << (VARIABLE_FOR_OUTPUT_ARRAY==N-1?"":" "); } OS #define OUTPUT_ITR( OS , A ) { auto ITERATOR_FOR_OUTPUT_ITR = A.BE() , EN_FOR_OUTPUT_ITR = A.EN(); bool VARIABLE_FOR_OUTPUT_ITR = ITERATOR_FOR_COUT_ITR != END_FOR_COUT_ITR; WH( VARIABLE_FOR_OUTPUT_ITR ){ OS << *ITERATOR_FOR_COUT_ITR; ( VARIABLE_FOR_OUTPUT_ITR = ++ITERATOR_FOR_COUT_ITR != END_FOR_COUT_ITR ) ? OS : OS << " "; } } OS #define RETURN( ... ) SOLVE_ONLY; COUT( __VA_ARGS__ ); RE #define COMPARE( ... ) auto naive = Naive( __VA_ARGS__ ); auto answer = Answer( __VA_ARGS__ ); bool match = naive == answer; COUT( "(" , #__VA_ARGS__ , ") == (" , __VA_ARGS__ , ") : Naive == " , naive , match ? "==" : "!=" , answer , "== Answer" ); if( !match ){ RE; } // 圧縮用 #define TE template #define TY typename #define US using #define ST static #define AS assert #define IN inline #define CL class #define PU public #define OP operator #define CE constexpr #define CO const #define NE noexcept #define RE return #define WH while #define VO void #define VE vector #define LI list #define BE begin #define EN end #define SZ size #define LE length #define PW Power #define MO move #define TH this #define CRI CO int& #define CRUI CO uint& #define CRL CO ll& #define VI virtual #define ST_AS static_assert #define reMO_CO remove_const #define is_COructible_v is_constructible_v #define rBE rbegin #define reSZ resize // 型のエイリアス #define decldecay_t(VAR)decay_t TE US ret_t = decltype(declval()(declval()...)); TE US inner_t = TY T::type; US uint = unsigned int; US ll = long long; US ull = unsigned long long; US ld = long double; US lld = __float128; TE US T2 = pair; TE US T3 = tuple; TE US T4 = tuple; US path = pair; // 入出力用 #ifndef DEBUG #define DF_OF_COUT_FOR_VE(V)TE IN basic_ostream& OP<<(basic_ostream& os,CO V& arg){auto BE = arg.BE(),EN = arg.EN();auto IT = BE;WH(IT != EN){(IT == BE?os:os << " ")<< *IT;IT++;}RE os;} TE IN basic_istream& VariadicCin(basic_istream& is){RE is;} TE IN basic_istream& VariadicCin(basic_istream& is,Arg& arg,ARGS&... args){RE VariadicCin(is >> arg,args...);} TE IN basic_istream& VariadicGetline(basic_istream& is,CO char& separator){RE is;} TE IN basic_istream& VariadicGetline(basic_istream& is,CO char& separator,Arg& arg,ARGS&... args){RE VariadicGetline(getline(is,arg,separator),separator,args...);} DF_OF_COUT_FOR_VE(VE); DF_OF_COUT_FOR_VE(LI); DF_OF_COUT_FOR_VE(set); DF_OF_COUT_FOR_VE(unordered_set); TE IN basic_ostream& OP<<(basic_ostream& os,CO pair& arg){RE os << arg.first << " " << arg.second;} TE IN basic_ostream& VariadicCout(basic_ostream& os,CO Arg& arg){RE os << arg;} TE IN basic_ostream& VariadicCout(basic_ostream& os,CO Arg1& arg1,CO Arg2& arg2,CO ARGS&... args){RE VariadicCout(os << arg1 << " ",arg2,args...);} #endif // 算術用 TE CE T PositiveBaseModulo(T a,CO T& p){RE MO(a < 0?((((++a)*= -1)%= p)*= -1)+= p - 1:a < p?a:a %= p);} TE CE T Modulo(T a,CO T& p){RE PositiveBaseRS(MO(a),p < 0?-p:p);} TE CE T PositiveBaseQuotient(CO T& a,CO T& p){RE(a - PositiveBaseModulo(a,p))/ p;} TE CE T Quotient(CO T& a,CO T& p){RE p < 0?PositiveBaseQuotient(-a,-p):PositiveBaseQuotient(a,p);} #ifdef DEBUG #include "c:/Users/user/Documents/Programming/Utility/BinarySearch/a_Body.hpp" #include "c:/Users/user/Documents/Programming/Utility/TwoPointerapproach/a_Body.hpp" #else // 二分探索用 // EXPRESSIONがANSWERの広義単調関数の時、EXPRESSION >= CO_TARGETの整数解を格納。 #define BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , DESIRED_INEQUALITY , CO_TARGET , INEQUALITY_FOR_CHECK , UPDATE_U , UPDATE_L , UPDATE_ANSWER ) \ ST_AS( ! is_same::value && ! is_same::value ); \ ll ANSWER = MINIMUM; \ { \ ll L_BS = MINIMUM; \ ll U_BS = MAXIMUM; \ ANSWER = UPDATE_ANSWER; \ ll EXPRESSION_BS; \ CO ll CO_TARGET_BS = ( CO_TARGET ); \ ll DIFFERENCE_BS; \ WH( L_BS < U_BS ){ \ DIFFERENCE_BS = ( EXPRESSION_BS = ( EXPRESSION ) ) - CO_TARGET_BS; \ CERR( "二分探索中:" , "L_BS =" , L_BS , "<=" , #ANSWER , "=" , ANSWER , "<=" , U_BS , "= U_BS :" , #EXPRESSION , "=" , EXPRESSION_BS , DIFFERENCE_BS > 0 ? ">" : DIFFERENCE_BS < 0 ? "<" : "=" , CO_TARGET_BS , "=" , #CO_TARGET ); \ if( DIFFERENCE_BS INEQUALITY_FOR_CHECK 0 ){ \ U_BS = UPDATE_U; \ } else { \ L_BS = UPDATE_L; \ } \ ANSWER = UPDATE_ANSWER; \ } \ if( L_BS > U_BS ){ \ CERR( "二分探索失敗:" , "L_BS =" , L_BS , ">" , U_BS , "= U_BS :" , #ANSWER , ":=" , #MAXIMUM , "+ 1 =" , MAXIMUM + 1 ); \ CERR( "二分探索マクロにミスがある可能性があります。変更前の版に戻してください。" ); \ ANSWER = MAXIMUM + 1; \ } else { \ CERR( "二分探索終了:" , "L_BS =" , L_BS , "<=" , #ANSWER , "=" , ANSWER , "<=" , U_BS , "= U_BS" ); \ CERR( "二分探索が成功したかを確認するために" , #EXPRESSION , "を計算します。" ); \ CERR( "成功判定が不要な場合はこの計算を削除しても構いません。" ); \ EXPRESSION_BS = ( EXPRESSION ); \ CERR( "二分探索結果:" , #EXPRESSION , "=" , EXPRESSION_BS , ( EXPRESSION_BS > CO_TARGET_BS ? ">" : EXPRESSION_BS < CO_TARGET_BS ? "<" : "=" ) , CO_TARGET_BS ); \ if( EXPRESSION_BS DESIRED_INEQUALITY CO_TARGET_BS ){ \ CERR( "二分探索成功:" , #ANSWER , ":=" , ANSWER ); \ } else { \ CERR( "二分探索失敗:" , #ANSWER , ":=" , #MAXIMUM , "+ 1 =" , MAXIMUM + 1 ); \ CERR( "単調でないか、単調増加性と単調減少性を逆にしてしまったか、探索範囲内に解が存在しません。" ); \ ANSWER = MAXIMUM + 1; \ } \ } \ } \ // 単調増加の時にEXPRESSION >= CO_TARGETの最小解を格納。 #define BS1( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , CO_TARGET ) BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , >= , CO_TARGET , >= , ANSWER , ANSWER + 1 , ( L_BS + U_BS ) / 2 ) // 単調増加の時にEXPRESSION <= CO_TARGETの最大解を格納。 #define BS2( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , CO_TARGET ) BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , <= , CO_TARGET , > , ANSWER - 1 , ANSWER , ( L_BS + 1 + U_BS ) / 2 ) // 単調減少の時にEXPRESSION >= CO_TARGETの最大解を格納。 #define BS3( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , CO_TARGET ) BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , >= , CO_TARGET , < , ANSWER - 1 , ANSWER , ( L_BS + 1 + U_BS ) / 2 ) // 単調減少の時にEXPRESSION <= CO_TARGETの最小解を格納。 #define BS4( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , CO_TARGET ) BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , <= , CO_TARGET , <= , ANSWER , ANSWER + 1 , ( L_BS + U_BS ) / 2 ) // t以下の値が存在すればその最大値のiterator、存在しなければend()を返す。 TE IN TY set::iterator MaximumLeq(set& S,CO T& t){CO auto EN = S.EN();if(S.empty()){RE EN;}auto IT = S.upper_bound(t);RE IT == EN?S.find(*(S.rBE())):IT == S.BE()?EN:--IT;} // t未満の値が存在すればその最大値のiterator、存在しなければend()を返す。 TE IN TY set::iterator MaximumLt(set& S,CO T& t){CO auto EN = S.EN();if(S.empty()){RE EN;}auto IT = S.lower_bound(t);RE IT == EN?S.find(*(S.rBE())):IT == S.BE()?EN:--IT;} // t以上の値が存在すればその最小値のiterator、存在しなければend()を返す。 TE IN TY set::iterator MinimumGeq(set& S,CO T& t){RE S.lower_bound(t);} // tより大きい値が存在すればその最小値のiterator、存在しなければend()を返す。 TE IN TY set::iterator MinimumGt(set& S,CO T& t){RE S.upper_bound(t);} // 尺取り法用 // VAR_TPA_LとVAR_TPA_RをINITで初期化し、VAR_TPA_RがCONTINUE_CONDITIONを満たす限り、 // 閉区間[VAR_TPA_L,VAR_TPA_R]が条件ON_CONDITIONを満たすか否かを判定し、 // trueになるかVAR_TAR_LがVAR_TAR_Rに追い付くまでVAR_TPA_Lの更新操作UPDATE_Lを繰り返し、 // その後VAR_TPA_Rの更新操作UPDATE_Rを行う。 // ON_CONDITIONがtrueとなる極大閉区間とその時点でのINFOをANSWERに格納する。 #define TPA( ANSWER , VAR_TPA , INIT , CONTINUE_CONDITION , UPDATE_L , UPDATE_R , ON_CONDITION , INFO ) \ VE> ANSWER{}; \ { \ auto init_TPA = INIT; \ decldecay_t( ANSWER.front() ) ANSWER ## _temp = { init_TPA , init_TPA , INFO }; \ auto ANSWER ## _prev = ANSWER ## _temp; \ auto& VAR_TPA ## _L = get<0>( ANSWER ## _temp ); \ auto& VAR_TPA ## _R = get<1>( ANSWER ## _temp ); \ auto& VAR_TPA ## _info = get<2>( ANSWER ## _temp ); \ bool on_TPA_prev = false; \ WH( true ){ \ bool continuing = CONTINUE_CONDITION; \ bool on_TPA = continuing && ( ON_CONDITION ); \ CERR( continuing ? "尺取り中" : "尺取り終了" , ": [L,R] = [" , VAR_TPA ## _L , "," , VAR_TPA ## _R , "] ," , on_TPA_prev ? "on" : "off" , "->" , on_TPA ? "on" : "off" , ", info =" , VAR_TPA ## _info ); \ if( on_TPA_prev && ! on_TPA ){ \ ANSWER.push_back( ANSWER ## _prev ); \ } \ if( continuing ){ \ if( on_TPA || VAR_TPA ## _L == VAR_TPA ## _R ){ \ ANSWER ## _prev = ANSWER ## _temp; \ UPDATE_R; \ } else { \ UPDATE_L; \ } \ } else { \ break; \ } \ on_TPA_prev = on_TPA; \ } \ } \ #endif // データ構造用 TE TY V> IN auto OP+(CO V& a0,CO V& a1)-> decldecay_t((declval>().push_back(declval()),a0)){if(a0.empty()){RE a1;}if(a1.empty()){RE a0;}AS(a0.SZ()== a1.SZ());V AN{};for(auto IT0 = a0.BE(),IT1 = a1.BE(),EN0 = a0.EN();IT0 != EN0;IT0++,IT1++){AN.push_back(*IT0 + *IT1);}RE AN;} TE IN pair OP+(CO pair& t0,CO pair& t1){RE{t0.first + t1.first,t0.second + t1.second};} TE IN tuple OP+(CO tuple& t0,CO tuple& t1){RE{get<0>(t0)+ get<0>(t1),get<1>(t0)+ get<1>(t1),get<2>(t0)+ get<2>(t1)};} TE IN tuple OP+(CO tuple& t0,CO tuple& t1){RE{get<0>(t0)+ get<0>(t1),get<1>(t0)+ get<1>(t1),get<2>(t0)+ get<2>(t1),get<3>(t0)+ get<3>(t1)};} TE IN T Addition(CO T& t0,CO T& t1){RE t0 + t1;} TE IN T Xor(CO T& t0,CO T& t1){RE t0 ^ t1;} TE IN T MU(CO T& t0,CO T& t1){RE t0 * t1;} TE IN CO T& Zero(){ST CO T z{};RE z;} TE IN CO T& One(){ST CO T o = 1;RE o;}TE IN T AdditionInv(CO T& t){RE -t;} TE IN T Id(CO T& v){RE v;} TE IN T Min(CO T& a,CO T& b){RE a < b?a:b;} TE IN T Max(CO T& a,CO T& b){RE a < b?b:a;} // グラフ用 TE TY V> IN auto Get(CO V& a){RE[&](CRI i = 0){RE a[i];};} TE IN VE id(CRI SZ){VE AN(SZ);FOR(i,0,SZ){AN[i]= i;}RE AN;} // グリッド問題用 int H,W,H_minus,W_minus,HW; VE wall_str;VE > non_wall; char walkable = '.',unwalkable = '#'; IN T2 EnumHW(CRI v){RE{v / W,v % W};} IN int EnumHW_inv(CO T2& ij){auto&[i,j]= ij;RE i * W + j;} CO string direction[4]={"U","R","D","L"}; IN int DirectionNumberOnGrid(CRI i,CRI j,CRI k,CRI h){RE ik?0:jh?3:(AS(false),-1);} IN int DirectionNumberOnGrid(CRI v,CRI w){auto[i,j]=EnumHW(v);auto[k,h]=EnumHW(w);RE DirectionNumberOnGrid(i,j,k,h);} IN int ReverseDirectionNumberOnGrid(CRI n){AS(0<=n&&n<4);RE(n+2)%4;} IN VE EdgeOnGrid(CRI v){VEAN{};auto[i,j]=EnumHW(v);if(i>0&&wall_str[i-1][j]==walkable){AN.push_back(EnumHW_inv({i-1,j}));}if(i+10&&wall_str[i][j-1]==walkable){AN.push_back(EnumHW_inv({i,j-1}));}if(j+1 WeightedEdgeOnGrid(CRI v){VEAN{};auto[i,j]=EnumHW(v);if(i>0&&wall_str[i-1][j]==walkable){AN.push_back({EnumHW_inv({i-1,j}),1});}if(i+10&&wall_str[i][j-1]==walkable){AN.push_back({EnumHW_inv({i,j-1}),1});}if(j+1& S){if(S.empty()){S.reSZ(H);}cin>>S[i];AS(int(S[i].SZ())==W);} IN VO SetWallOnGrid(CRI i,VE>& b){if(b.empty()){b.reSZ(H,VE(W));}auto&S_i=wall_str[i];auto&b_i=b[i];FOR(j,0,W){b_i[j]=S_i[j]==walkable?false:(AS(S_i[j]==unwalkable),true);}} // デバッグ用 #ifdef DEBUG IN VO AlertAbort( int n ) { CERR( "abort関数が呼ばれました。assertマクロのメッセージが出力されていない場合はオーバーフローの有無を確認をしてください。" ); } VO AutoCheck( int& exec_mode , CO bool& use_getline ); IN VO Solve(); IN VO Experiment(); IN VO SmallTest(); IN VO RandomTest(); ll GetRand( CRL Rand_min , CRL Rand_max ); IN VO BreakPoint( CRI LINE ) {} int exec_mode; CEXPR( int , solve_mode , 0 ); CEXPR( int , sample_debug_mode , 1 ); CEXPR( int , submission_debug_mode , 2 ); CEXPR( int , library_search_mode , 3 ); CEXPR( int , experiment_mode , 4 ); CEXPR( int , small_test_mode , 5 ); CEXPR( int , random_test_mode , 6 ); #ifdef USE_GETLINE CEXPR( bool , use_getline , true ); #else CEXPR( bool , use_getline , false ); #endif #else ll GetRand( CRL Rand_min , CRL Rand_max ) { ll answer = time( NULL ); RE answer * rand() % ( Rand_max + 1 - Rand_min ) + Rand_min; } #endif // VVV 常設ライブラリは以下に挿入する。 #ifdef DEBUG #include "c:/Users/user/Documents/Programming/Utility/Set/a_Body.hpp" #include "c:/Users/user/Documents/Programming/Mathematics/Algebra/Monoid/Group/a_Body.hpp" #include "c:/Users/user/Documents/Programming/Mathematics/Geometry/Graph/a_Body.hpp" #include "c:/Users/user/Documents/Programming/Mathematics/Arithmetic/Mod/ConstexprModulo/Debug/a_Body.hpp" CEXPR(uint,P,998244353);US MP = Mod

; #else // Set (1KB) CL is_ordered{PU:is_ordered()= delete;TE ST CE auto Check(CO T& t)-> decltype(t < t,true_type());ST CE false_type Check(...);TE ST CE CO bool value = is_same_v< decltype(Check(declval())),true_type >;};TE TY MOD>struct hash>{IN size_t OP()(CO MOD& n)CO;};TE TY PAIR>struct hash>{IN size_t OP()(CO PAIR& n)CO;};TE struct hash>{IN size_t OP()(CO tuple& n)CO;}; TE US Set = conditional_t>,unordered_set,conditional_t,set,VO>>;TE US Map = conditional_t>,unordered_map,conditional_t,map,VO>>; TE TY MOD> IN size_t hash>::OP()(CO MOD& n)CO{ST CO hash h;RE h(n.RP());}TE TY PAIR> IN size_t hash>::OP()(CO PAIR& n)CO{ST CO size_t seed = GetRand(1e3,1e8);ST CO hash h0;ST CO hash h1;RE(h0(get<0>(n))+ seed)^ h1(get<1>(n));}TE IN size_t hash>::OP()(CO tuple& n)CO{ST CO size_t seed = GetRand(1e3,1e8);ST CO hash> h01;ST CO hash h2;RE(h01({get<0>(n),get<1>(n)})+ seed)^ h2(get<2>(n));} // Algebra (4KB) #define DC_OF_CPOINT(POINT)IN CO U& POINT()CO NE #define DC_OF_POINT(POINT)IN U& POINT()NE #define DF_OF_CPOINT(POINT)TE IN CO U& VirtualPointedSet::POINT()CO NE{RE Point();} #define DF_OF_POINT(POINT)TE IN U& VirtualPointedSet::POINT()NE{RE Point();} TE CL UnderlyingSet{PU:US type = U;};TE CL VirtualPointedSet:VI PU UnderlyingSet{PU:VI CO U& Point()CO NE = 0;VI U& Point()NE = 0;DC_OF_CPOINT(Unit);DC_OF_CPOINT(Zero);DC_OF_CPOINT(One);DC_OF_CPOINT(Infty);DC_OF_POINT(init);DC_OF_POINT(root);};TE CL PointedSet:VI PU VirtualPointedSet{PU:U m_b_U;IN PointedSet(U b_u = U());IN CO U& Point()CO NE;IN U& Point()NE;};TE CL VirtualNSet:VI PU UnderlyingSet{PU:VI U Transfer(CO U& u)= 0;IN U Inverse(CO U& u);};TE CL AbstractNSet:VI PU VirtualNSet{PU:F_U m_f_U;IN AbstractNSet(F_U f_U);IN U Transfer(CO U& u);};TE CL VirtualMagma:VI PU UnderlyingSet{PU:VI U Product(U u0,CO U& u1)= 0;IN U Sum(U u0,CO U& u1);};TE CL AdditiveMagma:VI PU VirtualMagma{PU:IN U Product(U u0,CO U& u1);};TE CL MultiplicativeMagma:VI PU VirtualMagma{PU:IN U Product(U u0,CO U& u1);};TE CL AbstractMagma:VI PU VirtualMagma{PU:M_U m_m_U;IN AbstractMagma(M_U m_U);IN U Product(U u0,CO U& u1);}; TE IN PointedSet::PointedSet(U b_U):m_b_U(MO(b_U)){}TE IN CO U& PointedSet::Point()CO NE{RE m_b_U;}TE IN U& PointedSet::Point()NE{RE m_b_U;}DF_OF_CPOINT(Unit);DF_OF_CPOINT(Zero);DF_OF_CPOINT(One);DF_OF_CPOINT(Infty);DF_OF_POINT(init);DF_OF_POINT(root);TE IN AbstractNSet::AbstractNSet(F_U f_U):m_f_U(MO(f_U)){ST_AS(is_invocable_r_v);}TE IN U AbstractNSet::Transfer(CO U& u){RE m_f_U(u);}TE IN U VirtualNSet::Inverse(CO U& u){RE Transfer(u);}TE IN AbstractMagma::AbstractMagma(M_U m_U):m_m_U(MO(m_U)){ST_AS(is_invocable_r_v);}TE IN U AdditiveMagma::Product(U u0,CO U& u1){RE MO(u0 += u1);}TE IN U MultiplicativeMagma::Product(U u0,CO U& u1){RE MO(u0 *= u1);}TE IN U AbstractMagma::Product(U u0,CO U& u1){RE m_m_U(MO(u0),u1);}TE IN U VirtualMagma::Sum(U u0,CO U& u1){RE Product(MO(u0),u1);}TE CL VirtualMonoid:VI PU VirtualMagma,VI PU VirtualPointedSet{};TE CL AdditiveMonoid:VI PU VirtualMonoid,PU AdditiveMagma,PU PointedSet{};TE CL MultiplicativeMonoid:VI PU VirtualMonoid,PU MultiplicativeMagma,PU PointedSet{PU:IN MultiplicativeMonoid(U e_U);};TE CL AbstractMonoid:VI PU VirtualMonoid,PU AbstractMagma,PU PointedSet{PU:IN AbstractMonoid(M_U m_U,U e_U);};TE IN MultiplicativeMonoid::MultiplicativeMonoid(U e_U):PointedSet(MO(e_U)){}TE IN AbstractMonoid::AbstractMonoid(M_U m_U,U e_U):AbstractMagma(MO(m_U)),PointedSet(MO(e_U)){}TE CL VirtualGroup:VI PU VirtualMonoid,VI PU VirtualPointedSet,VI PU VirtualNSet{};TE CL AdditiveGroup:VI PU VirtualGroup,PU AdditiveMonoid{PU:IN U Transfer(CO U& u);};TE CL AbstractGroup:VI PU VirtualGroup,PU AbstractMonoid,PU AbstractNSet{PU:IN AbstractGroup(M_U m_U,U e_U,I_U i_U);};TE IN AbstractGroup::AbstractGroup(M_U m_U,U e_U,I_U i_U):AbstractMonoid(MO(m_U),MO(e_U)),AbstractNSet(MO(i_U)){}TE IN U AdditiveGroup::Transfer(CO U& u){RE -u;} // Graph (5KB) TE CL VirtualGraph:VI PU UnderlyingSet{PU:VI R1 Enumeration(CRI i)= 0;IN R2 Enumeration_inv(CO T& t);TE IN R2 Enumeration_inv(CO PATH& p);IN VO Reset();VI CRI SZ()CO NE = 0;VI E& edge()NE = 0;VI ret_t Edge(CO T& t)= 0;VI R2 Enumeration_inv_Body(CO T& t)= 0;};TE CL EdgeImplimentation:VI PU VirtualGraph{PU:int m_SZ;E m_edge;IN EdgeImplimentation(CRI SZ,E edge);IN CRI SZ()CO NE;IN E& edge()NE;IN ret_t Edge(CO T& t);};TE CL Graph:PU EdgeImplimentation{PU:IN Graph(CRI SZ,E edge);IN CRI Enumeration(CRI i);TE IN Graph GetGraph(F edge)CO;IN CRI Enumeration_inv_Body(CRI t);};TE CL EnumerationGraph:PU EdgeImplimentation,ret_t,E>{PU:Enum_T m_enum_T;Enum_T_inv m_enum_T_inv;IN EnumerationGraph(CRI SZ,Enum_T enum_T,Enum_T_inv enum_T_inv,E edge);IN ret_t Enumeration(CRI i);TE IN EnumerationGraph GetGraph(F edge)CO;IN ret_t Enumeration_inv_Body(CO T& t);};TE EnumerationGraph(CRI SZ,Enum_T enum_T,Enum_T_inv enum_T_inv,E edge)-> EnumerationGraph()(0)),Enum_T,Enum_T_inv,E>;TE CL MemorisationGraph:PU EdgeImplimentation{PU:int m_LE;VE m_memory;Map m_memory_inv;IN MemorisationGraph(CRI SZ,E edge);IN T Enumeration(CRI i);IN VO Reset();TE IN MemorisationGraph GetGraph(F edge)CO;IN CRI Enumeration_inv_Body(CO T& t);};TE MemorisationGraph(CRI SZ,E edge)-> MemorisationGraph()().back()),E>;TE MemorisationGraph(CRI SZ,E edge)-> MemorisationGraph(declval()().back())),E>; TE IN EdgeImplimentation::EdgeImplimentation(CRI SZ,E edge):m_SZ(SZ),m_edge(MO(edge)){ST_AS(is_COructible_v && is_COructible_v && is_invocable_v);}TE IN Graph::Graph(CRI SZ,E edge):EdgeImplimentation(SZ,MO(edge)){}TE IN EnumerationGraph::EnumerationGraph(CRI SZ,Enum_T enum_T,Enum_T_inv enum_T_inv,E edge):EdgeImplimentation,ret_t,E>(SZ,MO(edge)),m_enum_T(MO(enum_T)),m_enum_T_inv(MO(enum_T_inv)){}TE IN MemorisationGraph::MemorisationGraph(CRI SZ,E edge):EdgeImplimentation(SZ,MO(edge)),m_LE(),m_memory(),m_memory_inv(){ST_AS(is_invocable_v && is_invocable_v);}TE IN CRI Graph::Enumeration(CRI i){RE i;}TE IN ret_t EnumerationGraph::Enumeration(CRI i){RE m_enum_T(i);}TE IN T MemorisationGraph::Enumeration(CRI i){AS(0 <= i && i < m_LE);RE m_memory[i];}TE IN R2 VirtualGraph::Enumeration_inv(CO T& t){RE Enumeration_inv_Body(t);}TE TE IN R2 VirtualGraph::Enumeration_inv(CO PATH& p){RE Enumeration_inv_Body(get<0>(p));}TE IN CRI Graph::Enumeration_inv_Body(CRI i){RE i;}TE IN ret_t EnumerationGraph::Enumeration_inv_Body(CO T& t){RE m_enum_T_inv(t);}TE IN CRI MemorisationGraph::Enumeration_inv_Body(CO T& t){if(m_memory_inv.count(t)== 0){AS(m_LE < TH->SZ());m_memory.push_back(t);RE m_memory_inv[t]= m_LE++;}RE m_memory_inv[t];}TE VO VirtualGraph::Reset(){}TE IN VO MemorisationGraph::Reset(){m_LE = 0;m_memory.clear();m_memory_inv.clear();}TE IN CRI EdgeImplimentation::SZ()CO NE{RE m_SZ;}TE IN E& EdgeImplimentation::edge()NE{RE m_edge;}TE IN ret_t EdgeImplimentation::Edge(CO T& t){RE m_edge(t);}TE TE IN Graph Graph::GetGraph(F edge)CO{RE Graph(TH->SZ(),MO(edge));}TE TE IN EnumerationGraph EnumerationGraph::GetGraph(F edge)CO{RE EnumerationGraph(TH->SZ(),m_enum_T,m_enum_T_inv,MO(edge));}TE TE IN MemorisationGraph MemorisationGraph::GetGraph(F edge)CO{RE MemorisationGraph(TH->SZ(),MO(edge));} // ConstexprModulo (7KB) CEXPR(uint,P,998244353); #define RP Represent #define DeRP Derepresent TE CE INT RS(INT n)NE{RE MO(n < 0?((((++n)*= -1)%= M)*= -1)+= M - 1:n < INT(M)?n:n %= M);}TE CE INT& RSP(INT& n)NE{CE CO uint trunc =(1 << 23)- 1;INT n_u = n >> 23;n &= trunc;INT n_uq =(n_u / 7)/ 17;n_u -= n_uq * 119;n += n_u << 23;RE n < n_uq?n += P - n_uq:n -= n_uq;} TE CL Mod;TE CL COantsForMod{PU:COantsForMod()= delete;ST CE CO uint g_memory_bound = 1e6;ST CE CO uint g_memory_LE = M < g_memory_bound?M:g_memory_bound;ST CE uint g_M_minus = M - 1;ST CE int g_order_minus_1 = M - 2;ST CE int g_order_minus_1_neg = -g_order_minus_1;}; #define DC_OF_CM_FOR_MOD(OPR)CE bool OP OPR(CO Mod& n)CO NE #define DC_OF_AR_FOR_MOD(OPR,EX)CE Mod OP OPR(Mod n)CO EX; #define DF_OF_CM_FOR_MOD(OPR)TE CE bool Mod::OP OPR(CO Mod& n)CO NE{RE m_n OPR n.m_n;} #define DF_OF_AR_FOR_MOD(OPR,EX,LEFT,OPR2)TE CE Mod Mod::OP OPR(Mod n)CO EX{RE MO(LEFT OPR2 ## = *TH);}TE CE Mod OP OPR(T n0,CO Mod& n1)EX{RE MO(Mod(MO(n0))OPR ## = n1);} TE CL Mod{PU:uint m_n;CE Mod()NE;CE Mod(CO Mod& n)NE;CE Mod(Mod&& n)NE;TE CE Mod(T n)NE;CE Mod& OP=(Mod n)NE;CE Mod& OP+=(CO Mod& n)NE;CE Mod& OP-=(CO Mod& n)NE;CE Mod& OP*=(CO Mod& n)NE;IN Mod& OP/=(Mod n);TE CE Mod& OP<<=(INT n);TE CE Mod& OP>>=(INT n);CE Mod& OP++()NE;CE Mod OP++(int)NE;CE Mod& OP--()NE;CE Mod OP--(int)NE;DC_OF_CM_FOR_MOD(==);DC_OF_CM_FOR_MOD(!=);DC_OF_CM_FOR_MOD(<);DC_OF_CM_FOR_MOD(<=);DC_OF_CM_FOR_MOD(>);DC_OF_CM_FOR_MOD(>=);DC_OF_AR_FOR_MOD(+,NE);DC_OF_AR_FOR_MOD(-,NE);DC_OF_AR_FOR_MOD(*,NE);DC_OF_AR_FOR_MOD(/,);TE CE Mod OP^(INT EX)CO;TE CE Mod OP<<(INT n)CO;TE CE Mod OP>>(INT n)CO;CE Mod OP-()CO NE;CE Mod& SignInvert()NE;IN Mod& Invert();TE CE Mod& PW(INT EX);CE VO swap(Mod& n)NE;CE CRUI RP()CO NE;ST CE Mod DeRP(uint n)NE;ST IN CO Mod& Inverse(CRUI n);ST IN CO Mod& Factorial(CRUI n);ST IN CO Mod& FactorialInverse(CRUI n);ST IN Mod Combination(CRUI n,CRUI i);ST IN CO Mod& zero()NE;ST IN CO Mod& one()NE;TE CE Mod& PositivePW(INT EX)NE;TE CE Mod& NonNegativePW(INT EX)NE;US COants = COantsForMod;}; US MP = Mod

; TE CE Mod::Mod()NE:m_n(){}TE CE Mod::Mod(CO Mod& n)NE:m_n(n.m_n){}TE CE Mod::Mod(Mod&& n)NE:m_n(MO(n.m_n)){}TE TE CE Mod::Mod(T n)NE:m_n(RS(MO(n))){ST_AS(is_COructible_v >);}TE CE Mod& Mod::OP=(Mod n)NE{m_n = MO(n.m_n);RE *TH;}TE CE Mod& Mod::OP+=(CO Mod& n)NE{(m_n += n.m_n)< M?m_n:m_n -= M;RE *TH;}TE CE Mod& Mod::OP-=(CO Mod& n)NE{m_n < n.m_n?(m_n += M)-= n.m_n:m_n -= n.m_n;RE *TH;}TE CE Mod& Mod::OP*=(CO Mod& n)NE{m_n = MO(ull(m_n)* n.m_n)% M;RE *TH;}TE <> CE MP& MP::OP*=(CO MP& n)NE{ull m_n_copy = m_n;m_n = MO((m_n_copy *= n.m_n)< P?m_n_copy:RSP(m_n_copy));RE *TH;}TE IN Mod& Mod::OP/=(Mod n){RE OP*=(n.Invert());}TE TE CE Mod& Mod::OP<<=(INT n){AS(n >= 0);RE *TH *= Mod(2).NonNegativePW(MO(n));}TE TE CE Mod& Mod::OP>>=(INT n){AS(n >=0);WH(n-- > 0){((m_n & 1)== 0?m_n:m_n += M)>>= 1;}RE *TH;}TE CE Mod& Mod::OP++()NE{m_n < COants::g_M_minus?++m_n:m_n = 0;RE *TH;}TE CE Mod Mod::OP++(int)NE{Mod n{*TH};OP++();RE n;}TE CE Mod& Mod::OP--()NE{m_n == 0?m_n = COants::g_M_minus:--m_n;RE *TH;}TE CE Mod Mod::OP--(int)NE{Mod n{*TH};OP--();RE n;}DF_OF_CM_FOR_MOD(==);DF_OF_CM_FOR_MOD(!=);DF_OF_CM_FOR_MOD(>);DF_OF_CM_FOR_MOD(>=);DF_OF_CM_FOR_MOD(<);DF_OF_CM_FOR_MOD(<=);DF_OF_AR_FOR_MOD(+,NE,n,+);DF_OF_AR_FOR_MOD(-,NE,n.SignInvert(),+);DF_OF_AR_FOR_MOD(*,NE,n,*);DF_OF_AR_FOR_MOD(/,,n.Invert(),*);TE TE CE Mod Mod::OP^(INT EX)CO{RE MO(Mod(*TH).PW(MO(EX)));}TE TE CE Mod Mod::OP<<(INT n)CO{RE MO(Mod(*TH)<<= MO(n));}TE TE CE Mod Mod::OP>>(INT n)CO{RE MO(Mod(*TH)>>= MO(n));}TE CE Mod Mod::OP-()CO NE{RE MO(Mod(*TH).SignInvert());}TE CE Mod& Mod::SignInvert()NE{m_n > 0?m_n = M - m_n:m_n;RE *TH;}TE IN Mod& Mod::Invert(){AS(m_n != 0);uint m_n_neg;RE m_n < COants::g_memory_LE?(m_n = Inverse(m_n).m_n,*TH):((m_n_neg = M - m_n)< COants::g_memory_LE)?(m_n = M - Inverse(m_n_neg).m_n,*TH):NonNegativePW(COants::g_order_minus_1);}TE TE CE Mod& Mod::PositivePW(INT EX)NE{Mod PW{*TH};EX--;WH(EX != 0){(EX & 1)== 1?*TH *= PW:*TH;EX >>= 1;PW *= PW;}RE *TH;}TE TE CE Mod& Mod::NonNegativePW(INT EX)NE{RE EX == 0?(m_n = 1,*TH):PositivePW(MO(EX));}TE TE CE Mod& Mod::PW(INT EX){bool neg = EX < 0;AS(!(neg && m_n == 0));RE neg?PositivePW(MO(EX *= COants::g_order_minus_1_neg)):NonNegativePW(MO(EX));}TE CE VO Mod::swap(Mod& n)NE{std::swap(m_n,n.m_n);}TE IN CO Mod& Mod::Inverse(CRUI n){AS(n < COants::g_memory_LE);ST Mod memory[COants::g_memory_LE]={zero(),one()};ST uint LE_curr = 2;WH(LE_curr <= n){memory[LE_curr].m_n = M - memory[M % LE_curr].m_n * ull(M / LE_curr)% M;LE_curr++;}RE memory[n];}TE IN CO Mod& Mod::Factorial(CRUI n){if(M <= n){RE zero();}AS(n < COants::g_memory_LE);ST Mod memory[COants::g_memory_LE]={one(),one()};ST uint LE_curr = 2;WH(LE_curr <= n){(memory[LE_curr]= memory[LE_curr - 1])*= LE_curr;LE_curr++;}RE memory[n];}TE IN CO Mod& Mod::FactorialInverse(CRUI n){ST Mod memory[COants::g_memory_LE]={one(),one()};ST uint LE_curr = 2;WH(LE_curr <= n){(memory[LE_curr]= memory[LE_curr - 1])*= Inverse(LE_curr);LE_curr++;}RE memory[n];}TE IN Mod Mod::Combination(CRUI n,CRUI i){RE i <= n?Factorial(n)* FactorialInverse(i)* FactorialInverse(n - i):zero();}TE CE CRUI Mod::RP()CO NE{RE m_n;}TE CE Mod Mod::DeRP(uint n)NE{Mod n_copy{};n_copy.m_n = MO(n);RE n_copy;}TE IN CO Mod& Mod::zero()NE{ST CE CO Mod z{};RE z;}TE IN CO Mod& Mod::one()NE{ST CE CO Mod o{1};RE o;}TE IN Mod Inverse(CO Mod& n){RE MO(Mod(n).Invert());}TE CE Mod PW(Mod n,INT EX){RE MO(n.PW(MO(EX)));}TE CE VO swap(Mod& n0,Mod& n1)NE{n0.swap(n1);}TE IN string to_string(CO Mod& n)NE{RE to_string(n.RP())+ " + " + to_string(M)+ "Z";}TE IN basic_istream& OP>>(basic_istream& is,Mod& n){ll m;is >> m;n = m;RE is;}TE IN basic_ostream& OP<<(basic_ostream& os,CO Mod& n){RE os << n.RP();} #endif // AAA 常設ライブラリは以上に挿入する。 #define INCLUDE_LIBRARY #include __FILE__ #endif // INCLUDE_LIBRARY #endif // INCLUDE_SUB #endif // INCLUDE_MAIN