#ifdef DEBUG #define _GLIBCXX_DEBUG #define UNTIE ios_base::sync_with_stdio( false ); cin.tie( nullptr ); signal( SIGABRT , &AlertAbort ) #define DEXPR( LL , BOUND , VALUE , DEBUG_VALUE ) CEXPR( LL , BOUND , DEBUG_VALUE ) #define CERR( MESSAGE ) cerr << MESSAGE << endl; #define COUT( ANSWER ) cout << ANSWER << endl #define ASSERT( A , MIN , MAX ) CERR( "ASSERTチェック: " << ( MIN ) << ( ( MIN ) <= A ? "<=" : ">" ) << A << ( A <= ( MAX ) ? "<=" : ">" ) << ( MAX ) ); assert( ( MIN ) <= A && A <= ( MAX ) ) #define LIBRARY_SEARCH if( LibrarySearch() != 0 ){ QUIT; }; #define START_WATCH( PROCESS_NAME ) StartWatch( PROCESS_NAME ) #define STOP_WATCH( HOW_MANY_TIMES ) StopWatch( HOW_MANY_TIMES ) #else #pragma GCC optimize ( "O3" ) #pragma GCC optimize( "unroll-loops" ) #pragma GCC target ( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" ) #define UNTIE ios_base::sync_with_stdio( false ); cin.tie( nullptr ) #define DEXPR( LL , BOUND , VALUE , DEBUG_VALUE ) CEXPR( LL , BOUND , VALUE ) #define CERR( MESSAGE ) #define COUT( ANSWER ) cout << ANSWER << "\n" #define ASSERT( A , MIN , MAX ) assert( ( MIN ) <= A && A <= ( MAX ) ) #define LIBRARY_SEARCH #define START_WATCH( PROCESS_NAME ) #define STOP_WATCH( HOW_MANY_TIMES ) #endif // #define RANDOM_TEST #include using namespace std; using uint = unsigned int; using ll = long long; using ull = unsigned long long; #define ATT __attribute__( ( target( "sse4.2,fma,avx2,popcnt,lzcnt,bmi2" ) ) ) #define TYPE_OF( VAR ) decay_t #define CEXPR( LL , BOUND , VALUE ) constexpr LL BOUND = VALUE #define CIN( LL , A ) LL A; cin >> A #define CIN_ASSERT( A , MIN , MAX ) TYPE_OF( MAX ) A; SET_ASSERT( A , MIN , MAX ) #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( 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 AUTO_ITR( ARRAY ) auto itr_ ## ARRAY = ARRAY .begin() , end_ ## ARRAY = ARRAY .end() #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 QUIT return 0 #define SET_PRECISION( DECIMAL_DIGITS ) cout << fixed << setprecision( DECIMAL_DIGITS_ ) #ifdef DEBUG inline void AlertAbort( int n ) { CERR( "abort関数が呼ばれました。assertマクロのメッセージが出力されていない場合はオーバーフローの有無を確認をしてください。" ); } void StartWatch( const string& process_name = "nothing" ); void StopWatch( const int& how_many_times = 1 ); #endif #if defined( DEBUG ) && defined( RANDOM_TEST ) inline CEXPR( int , bound_random_test_num , 1000 ); #define START_MAIN FOR( random_test_num , 0 , bound_random_test_num ){ CERR( "(" << random_test_num << ")" ); ll GetRand( const ll& Rand_min , const ll& Rand_max ); #define SET_ASSERT( A , MIN , MAX ) CERR( #A << " = " << ( A = GetRand( MIN , MAX ) ) ) #define RETURN( ANSWER ) if( ( ANSWER ) == guchoku ){ CERR( ( ANSWER ) << " == " << guchoku ); continue; } else { CERR( ( ANSWER ) << " != " << guchoku ); QUIT; } #define FINISH_MAIN CERR( "" ); } #else #define START_MAIN #define SET_ASSERT( A , MIN , MAX ) cin >> A; ASSERT( A , MIN , MAX ) #define RETURN( ANSWER ) COUT( ( ANSWER ) ); QUIT #define FINISH_MAIN #endif template inline T Absolute( const T& a ){ return a > 0 ? a : -a; } template inline T Residue( const T& a , const T& p ){ return a >= 0 ? a % p : p - 1 - ( ( - ( a + 1 ) ) % p ); } #define POWER( ANSWER , ARGUMENT , EXPONENT ) \ static_assert( ! is_same::value && ! is_same::value ); \ TYPE_OF( ARGUMENT ) ANSWER{ 1 }; \ { \ TYPE_OF( ARGUMENT ) ARGUMENT_FOR_SQUARE_FOR_POWER = ( ARGUMENT ); \ 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 *= ARGUMENT_FOR_SQUARE_FOR_POWER; \ } \ ARGUMENT_FOR_SQUARE_FOR_POWER *= ARGUMENT_FOR_SQUARE_FOR_POWER; \ EXPONENT_FOR_SQUARE_FOR_POWER /= 2; \ } \ } \ #define POWER_MOD( ANSWER , ARGUMENT , EXPONENT , MODULO ) \ ll ANSWER{ 1 }; \ { \ ll 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 FACTORIAL_MOD( ANSWER , ANSWER_INV , INVERSE , MAX_INDEX , CONSTEXPR_LENGTH , MODULO ) \ static ll ANSWER[CONSTEXPR_LENGTH]; \ static ll ANSWER_INV[CONSTEXPR_LENGTH]; \ static ll INVERSE[CONSTEXPR_LENGTH]; \ { \ ll VARIABLE_FOR_PRODUCT_FOR_FACTORIAL = 1; \ ANSWER[0] = VARIABLE_FOR_PRODUCT_FOR_FACTORIAL; \ FOREQ( i , 1 , MAX_INDEX ){ \ ANSWER[i] = ( VARIABLE_FOR_PRODUCT_FOR_FACTORIAL *= i ) %= ( MODULO ); \ } \ ANSWER_INV[0] = ANSWER_INV[1] = INVERSE[1] = VARIABLE_FOR_PRODUCT_FOR_FACTORIAL = 1; \ FOREQ( i , 2 , MAX_INDEX ){ \ ANSWER_INV[i] = ( VARIABLE_FOR_PRODUCT_FOR_FACTORIAL *= INVERSE[i] = ( MODULO ) - ( ( ( ( MODULO ) / i ) * INVERSE[ ( MODULO ) % i ] ) % ( MODULO ) ) ) %= ( MODULO ); \ } \ } \ // 二分探索テンプレート // EXPRESSIONがANSWERの広義単調関数の時、EXPRESSION >= TARGETの整数解を格納。 #define BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , DESIRED_INEQUALITY , TARGET , INEQUALITY_FOR_CHECK , UPDATE_U , UPDATE_L , UPDATE_ANSWER ) \ static_assert( ! is_same::value && ! is_same::value ); \ ll ANSWER = MINIMUM; \ if( MINIMUM <= MAXIMUM ){ \ ll VARIABLE_FOR_BINARY_SEARCH_L = MINIMUM; \ ll VARIABLE_FOR_BINARY_SEARCH_U = MAXIMUM; \ ANSWER = ( VARIABLE_FOR_BINARY_SEARCH_L + VARIABLE_FOR_BINARY_SEARCH_U ) / 2; \ ll VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH; \ while( VARIABLE_FOR_BINARY_SEARCH_L != VARIABLE_FOR_BINARY_SEARCH_U ){ \ VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH = ( EXPRESSION ) - ( TARGET ); \ CERR( "二分探索中: " << VARIABLE_FOR_BINARY_SEARCH_L << "<=" << ANSWER << "<=" << VARIABLE_FOR_BINARY_SEARCH_U << ":" << EXPRESSION << "-" << TARGET << "=" << VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH ); \ if( VARIABLE_FOR_DIFFERENCE_FOR_BINARY_SEARCH INEQUALITY_FOR_CHECK 0 ){ \ VARIABLE_FOR_BINARY_SEARCH_U = UPDATE_U; \ } else { \ VARIABLE_FOR_BINARY_SEARCH_L = UPDATE_L; \ } \ ANSWER = UPDATE_ANSWER; \ } \ CERR( "二分探索終了: " << VARIABLE_FOR_BINARY_SEARCH_L << "<=" << ANSWER << "<=" << VARIABLE_FOR_BINARY_SEARCH_U << ":" << EXPRESSION << ( EXPRESSION > TARGET ? ">" : EXPRESSION < TARGET ? "<" : "=" ) << TARGET ); \ CERR( ( EXPRESSION DESIRED_INEQUALITY TARGET ? "二分探索成功" : "二分探索失敗" ) ); \ assert( EXPRESSION DESIRED_INEQUALITY TARGET ); \ } else { \ CERR( "二分探索失敗: " << MINIMUM << ">" << MAXIMUM ); \ assert( MINIMUM <= MAXIMUM ); \ } \ // 単調増加の時にEXPRESSION >= TARGETの最小解を格納。 #define BS1( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , TARGET ) \ BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , >= , TARGET , >= , ANSWER , ANSWER + 1 , ( VARIABLE_FOR_BINARY_SEARCH_L + VARIABLE_FOR_BINARY_SEARCH_U ) / 2 ) \ // 単調増加の時にEXPRESSION <= TARGETの最大解を格納。 #define BS2( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , TARGET ) \ BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , <= , TARGET , > , ANSWER - 1 , ANSWER , ( VARIABLE_FOR_BINARY_SEARCH_L + 1 + VARIABLE_FOR_BINARY_SEARCH_U ) / 2 ) \ // 単調減少の時にEXPRESSION >= TARGETの最大解を格納。 #define BS3( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , TARGET ) \ BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , >= , TARGET , < , ANSWER - 1 , ANSWER , ( VARIABLE_FOR_BINARY_SEARCH_L + 1 + VARIABLE_FOR_BINARY_SEARCH_U ) / 2 ) \ // 単調減少の時にEXPRESSION <= TARGETの最小解を格納。 #define BS4( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , TARGET ) \ BS( ANSWER , MINIMUM , MAXIMUM , EXPRESSION , <= , TARGET , <= , ANSWER , ANSWER + 1 , ( VARIABLE_FOR_BINARY_SEARCH_L + VARIABLE_FOR_BINARY_SEARCH_U ) / 2 ) \ // 圧縮用 #define TE template #define TY typename #define US using #define ST static #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 MO move #define TH this #define CRI CO int& #define CRUI CO uint& #define CRL CO ll& #define ASK_NUMBER( ... ) \ CERR( "問題の区分は以下の中で何番に該当しますか?" ); \ problems = { __VA_ARGS__ }; \ problems_size = problems.size(); \ FOR( i , 0 , problems_size ){ \ CERR( i << ": " << problems[i] ); \ } \ cin >> num; \ CERR( "" ); \ num_temp = 0; \ if( num < 0 || num >= problems_size ){ \ CERR( "返答は" << problems_size - 1 << "以下の非負整数にしてください。" ); \ CERR( "終了します。" ); \ CERR( "" ); \ return -1; \ } \ #define ASK_YES_NO( QUESTION ) \ CERR( QUESTION << "[y/n]" ); \ cin >> reply; \ if( reply != "y" && reply != "n" ){ \ CERR( "y/nのいずれかで答えてください。" ); \ CERR( "終了します。" ); \ CERR( "" ); \ return -1; \ } \ CERR( "" ); \ int LibrarySearch( int num = -1 ) { vector problems{}; int problems_size = 13; int num_temp = 0; string reply{}; if( num == -1 ){ ASK_YES_NO( "ライブラリーを探索しますか?" ); if( reply == "y" ){ CERR( "ライブラリーを探索します。" ); } else { CERR( "ライブラリーを探索せずに続行します。" ); CERR( "" ); return 0; } ASK_NUMBER( "数や関数に関する問題。" , "配列に関する問題。" , "文字列に関する問題。" , "順列に関する問題。" , "矩形領域に関する問題。" , "グラフに関する問題。" , "確率/期待値に関する問題。" , "ゲームに関する問題。" , "論理に関する問題。" , "半順序集合に関する問題。" , "関数適用に関する問題。" , "構築問題。" ); } else { CERR( "" ); } CEXPR( int , num_graph , 5 ); CEXPR( int , num_game , 7 ); if( num == num_temp++ ){ ASK_YES_NO( "入力は1つの数か、1つの数と法を表す数ですか?" ); if( reply == "y" ){ CERR( "まずは小さい入力の場合を愚直に計算し、OEISで検索しましょう。" ); CERR( "https://oeis.org/?language=japanese" ); CERR( "" ); CERR( "次に出力の定義と等価な式を考察しましょう。" ); CERR( "- 単調ならば、冪乗や階乗" ); CERR( "- 定義にp進法が使われていれば、各種探索アルゴリズム" ); CERR( "- 入力が素数に近い場合に規則性があれば、p進付値、p進法、" ); CERR( " オイラー関数、約数の個数など" ); CERR( "を検討しましょう。" ); } else { ASK_NUMBER( "求解問題。" , "最大化問題。" , "数え上げ問題。" , "ソート問題。" ); if( num == num_temp++ ){ CERR( "まず" ); CERR( "- 単調関数は二分探索" ); CERR( "- 可微分関数はニュートン法" ); CERR( "- 一次関数は掃き出し法" ); CERR( "を検討しましょう。" ); CERR( "" ); CERR( "それらが間に合わない時は" ); CERR( "- 探索範囲を絞れそうな場合は全探策" ); CERR( "- 多変数の合成関数で表せる場合は半分全列挙" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 凸関数は三分探索" ); CERR( "- 可微分関数は微分のニュートン法" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 変数の対称性があれば大小関係を制限した全探策" ); CERR( "- 何らかの約数となるなど動く範囲が狭い変数があればそれらを決め打った全探策" ); CERR( "- 多変数の合成関数で表せる場合は半分全列挙" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "集合Sを何らかの順序でソートした配列aに関する問題で、" ); CERR( "- 与えられた要素sが下から何番目かを答える場合は、非負整数iごとに" ); CERR( " 不等式a[i]<=sを判定する方法を考察し、iに関する二分探索" ); CERR( "- 与えられた非負整数iに対するa[i]を答える場合は、Sの要素sごとに" ); CERR( " 不等式a[i]<=sを判定する方法を考察し、sに関する二分探索" ); CERR( "を検討しましょう。" ); } } CERR( "" ); ASK_YES_NO( "マルチテストケースですか?" ); if( reply == "y" ){ CERR( "テストケース全体でのNの総和に直接上限が与えられている問題では、" ); CERR( "ライブラリーの使用時は、配列の初期化が各テストケースに必要となる場合に" ); CERR( "TLEとなる可能性が高いです。適宜動的配列に置き換えましょう。" ); CERR( "" ); CERR( "配列を手元の環境でデバッグしやすくするためにstaticをつけている場合は" ); CERR( "テストケースを跨いで値が残ってしまわないように注意しましょう。" ); CERR( "" ); CERR( "前計算の候補としては" ); CERR( "- 素数列挙" ); CERR( "- 1つまたは複数の整数の約数列挙" ); CERR( "- オイラー関数の値の列挙" ); CERR( "- サブゴールとなる関係式を満たす解の列挙" ); CERR( "を検討しましょう。" ); } else { CERR( "入力が大きい場合と小さい場合で解法を変える考察を忘れないようにしましょう。" ); } CERR( "" ); } else if( num == num_temp++ ){ ASK_YES_NO( "入力に配列が与えられますか?" ); if( reply == "y" ){ ASK_NUMBER( "区間等の領域処理問題。" , "成分を受け取る関数の総和問題" , "部分列を受け取る関数の総和問題" , "その他の関数に関する問題。" , "隣接成分間関係式に関する問題。" ); if( num == num_temp++ ){ ASK_NUMBER( "可換群構造+を使う問題。" , "可換羃等モノイド構造∨を使う問題。" , "モノイド構造*を使う問題。" , "非結合的マグマ構造*を使う問題。" , "集合へのマグマ作用(*,\\cdot)を使う問題。" , "モノイドへのマグマ作用(+,\\cdot)を使う問題。" , "定数とのmaxを取った値の区間和取得を使う問題。" ); if( num == num_temp++ ){ CERR( "- 区間加算/区間取得が必要ならば可換群BIT" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\BIT\\Template" ); CERR( "- 一点代入/一点加算/区間取得が必要ならば可換群平方分割" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\SqrtDecomposition\\Template" ); CERR( "- 区間以外の領域で加算/全更新後の一点取得が必要ならば階差数列" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\Tree\\DifferenceSeqeuence" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 一点代入/区間加算/一点取得/区間取得が必要ならば可換羃等モノイドBIT" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\BIT\\IntervalMax\\Template" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 一点代入/区間取得が必要ならばモノイドBIT" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\BIT\\Template\\Monoid" ); CERR( "- 一点加算/区間加算/一点取得/区間取得が必要ならばモノイド平方分割" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\SqrtDecomposition\\Template\\Monoid" ); CERR( "- 一点代入/一点取得/区間取得が必要ならばモノイドセグメント木" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\SegmentTree" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 写像のコード化" ); CERR( " \\Mathematics\\Function\\Encoder" ); CERR( "によりモノイドに帰着させることを検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 一点作用/区間作用/一点取得が必要ならば双対平方分割" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\SqrtDecomposition\\Template\\Dual" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "- 区間代入/区間作用/区間加算/一点取得/区間取得が必要な場合は遅延評価平方分割" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\SqrtDecomposition\\Template\\LazyEvaluation" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "maxで全体更新でなく区間更新をする場合の汎用的な解法は分かりません。" ); CERR( "例えば区間が包含について単調でmaxを取る値も単調であれば全体更新と" ); CERR( "同様の処理ができます。状況に応じた考察をしましょう。" ); CERR( "" ); CERR( "maxで全体更新をする場合、maxを取る値は単調である場合に帰着できます。" ); CERR( "maxで全体更新をしない場合、つまりただmaxの区間和を処理するだけの場合、" ); CERR( "クエリの順番を入れ替えることができるので、単調な全体更新に帰着できます。" ); CERR( "従って以下では単調な全体更新の問題を考えます。" ); CERR( "" ); CERR( "maxを取る定数を変数化し、元の値との大小を表す{0,1}値の係数を考えます。" ); CERR( "すると区間作用前後の値は統一的にその係数と変数を使って表せます。" ); CERR( "配列の各成分の係数の値が変化するイベントとクエリをソートして管理し、" ); CERR( "クエリがイベントを跨ぐたびに係数を更新することを検討しましょう。" ); CERR( "" ); CERR( "例えばクエリB_qに対するmax(A_i,B_q)の区間和は、" ); CERR( "- 優先度つきキューA'={(A_i,i)|i}(構築O(N log N))" ); CERR( "- (B_q,q)_qをソートしたB'(構築O(Q log Q))" ); CERR( "- 長さNの数列C=(0,...,0)(構築O(N))" ); CERR( "を用意し、B'を前から探索して順に各クエリ(B_q,q)を処理します。" ); CERR( "具体的にはA'を前から探索して順にA_i=3かつ(n,L)=(2,2)で" ); CERR( "- Q(i)=「i=3」" ); CERR( "- R_0(b_0,b_1)=「b_0b_1」" ); CERR( "と表されます。" ); CERR( "" ); if( num == num_temp++ ){ CERR( "P(B)を満たすAの連続部分列Bの数え上げは、" ); CERR( "- R_lたちがlに依存しないならば尺取り法O(N)" ); CERR( "- R_lたちがlに依存する場合、" ); CERR( " - O(N^2)が通りそうなら左端を固定した愚直探索" ); CERR( " - O(N^2)が通らなさそうならR_lたちの読み替え" ); } else if( num == num_temp++ ){ CERR( "P(B)を満たすAの連続とは限らない部分列Bの数え上げは、" ); CERR( "- n-1<=i<=max{j<=N|Q(j)}を満たす各i" ); CERR( "- (0,1,...,N-1)の長さn-1の各部分列s" ); CERR( "に対する" ); CERR( "「長さiで、任意の0<=l<=i-nに対しR_l(B)を満たし、" ); CERR( " 末尾n-1項がsに対応するAの部分列Bの個数dp[i][s]」" ); CERR( "を管理するi,sに関する動的計画法" ); CERR( "\\Mathematics\\Combinatorial\\Counting\\IncreasingSubsequence" ); CERR( "\\Mathematics\\Combinatorial\\Counting\\IncreasingSubsequence\\Subwalk" ); } else if( num == num_temp++ ){ CERR( "P(B)を満たすAの部分順列Bの数え上げは、" ); CERR( "- n-1<=|S|<=max{j<=N|Q(j)}を満たす(0,1,...,N-1)の部分集合S" ); CERR( "- Sの長さn-1の各部分順列s" ); CERR( "に対する" ); CERR( "「任意の0<=l<=|S|-nに対しR_l(B)を満たし、末尾n-1項がsに対応し、」" ); CERR( " 全体がSに対応するAの部分順列Bの個数dp[S][s]」" ); CERR( "を管理するS,sに関する動的計画法を検討しましょう。" ); CERR( "" ); CERR( "sの網羅は[0,N)^{n-1}の全探策でもS内の順列探索と定数倍しか変わらないので" ); CERR( "実装の速さを優先しましょう。" ); CERR( "" ); CERR( "Nが小さい場合は概算値" ); CERR( "7! ≒ 5×10^3" ); CERR( "8! ≒ 4×10^4" ); CERR( "9! ≒ 4×10^5" ); CERR( "10! ≒ 4×10^6" ); CERR( "11! ≒ 4×10^7" ); CERR( "12! ≒ 5×10^8" ); CERR( "を参考に順列の全列挙" ); CERR( "\\Mathematics\\Combinatorial\\Permutation" ); } CERR( "を検討しましょう。" ); CERR( "" ); CERR( "特にR_l(B)たちがgcdやmaxなどの羃等演算に関する等式で与えられる場合は、" ); CERR( "不等式の方が扱いやすいのでゼータ変換/メビウス変換を検討しましょう。" ); } } } else { ASK_NUMBER( "配列を受け取る関数に関する問題。" , "隣接成分間関係式に関する問題。" ); if( num == num_temp++ ){ ASK_NUMBER( "最大化問題。" , "数え上げ問題。" ); if( num == num_temp++ ){ CERR( "- 取り得る値が少なく関数が長さに関して再帰的構造を持つ場合は、" ); CERR( " 「長さiの時に可能な値全体または一部の集合dp[i]」" ); CERR( " を管理するiに関する動的計画法" ); CERR( "- 「v以上の値を取り得るか否か」が判定可能である時は" ); CERR( " vに関する二分探索" ); } else { CERR( "- マルチテストケースで配列の種類が少ない場合は、" ); CERR( " 全ての配列に対する関数の値の前計算" ); CERR( "- 取り得る値が少なく関数が長さに関して再帰的構造を持つ場合は、" ); CERR( " 「長さiの時に値vである配列の総数dp[i][v]」" ); CERR( " を管理するi,vに関する動的計画法" ); } } else { CERR( "- いくつかの条件の重ね合わせの時は包除原理" ); CERR( "- 全順序の場合は数の分割方法などへの翻訳" ); CERR( "- 疎な半順序の場合はグラフの前計算" ); } CERR( "を検討しましょう。" ); } } else if( num == num_temp++ ){ ASK_NUMBER( "部分列マッチング問題。" , "回文探索問題。" ); if( num == num_temp++ ){ CERR( "基本的には丁寧にループを回して解きましょう。" ); CERR( "- 比較対象が少ない場合、前または後ろから順に探索(貪欲法/動的計画法)" ); CERR( "- ワイルドカードを含む場合、" ); CERR( " - 前または後ろから順に場合分けをしてO(N)で処理できるか" ); CERR( " - 可能な代入方法を絞り込んでO(N)種類に落せるか" ); CERR( "- 比較回数が多い場合、ローリングハッシュ" ); CERR( " \\Utility\\String\\RollingHash" ); CERR( "- マッチングする文字列の最長化をする場合、Zアルゴリズム" ); CERR( " https://qiita.com/Pro_ktmr/items/16904c9570aa0953bf05" ); CERR( "- マッチングする文字数の最大化をする場合、文字の種類分の{0,1}値配列に" ); CERR( " 分けて内積の最大化(添え字を反転させて適当な法での畳み込み)" ); CERR( " \\Mathematics\\Arithmetic\\Mod" ); CERR( " \\Mathematics\\Polynomial" ); CERR( "を検討しましょう。" ); } else if( num == num_temp++ ){ CERR( "回文判定は長さに関して再帰的に計算できます。" ); CERR( "- O(N^2)が通る場合、愚直な再帰により前計算で全ての部分列の回文判定" ); CERR( "- O(N^2)が通らない場合、Manacherのアルゴリズムやローリングハッシュで前計算" ); CERR( " https://snuke.hatenablog.com/entry/2014/12/02/235837" ); CERR( "を検討しましょう。" ); } } else if( num == num_temp++ ){ CERR( "符号の計算は転倒数の計算に帰着させましょう。" ); CERR( "符号と何かの積の和は行列式に帰着させましょう。" ); CERR( "- 行列式そのものなら行基本変形でO(N^3)" ); CERR( "- 余因子展開の途中の値はメモ化再帰でO(N 2^N)" ); CERR( "で計算できます。" ); CERR( "" ); CERR( "1つの順列の転倒数は、" ); CERR( "- O(N^2)が通りそうならば愚直な二重ループ" ); CERR( "- O(N log_2 N)が通りそうならば可換群BIT" ); CERR( " \\Mathematics\\Combinatorial\\Permutation" ); CERR( " \\Mathematics\\SetTheory\\DirectProduct\\AffineSpace\\BIT" ); CERR( "で計算しましょう。" ); CERR( "" ); CERR( "条件を満たす順列全体をわたる転倒数の総和は、" ); CERR( "各i \ class BREADTH ## FirstSearch_Body \ { \ \ protected: \ int m_V; \ int m_init; \ list m_next; \ bool m_found[V_max]; \ int m_prev[V_max]; \ \ public: \ inline BREADTH ## FirstSearch_Body( const int& V ); \ inline BREADTH ## FirstSearch_Body( const int& V , const int& init ); \ \ inline void Reset( const int& init ); \ inline void Shift( const int& init ); \ \ inline const int& size() const; \ inline const int& init() const; \ inline bool& found( const int& i ); \ inline const int& prev( const int& i ) const; \ \ int Next(); \ \ private: \ virtual list e( const int& t ) = 0; \ \ }; \ \ template E(const int&)> \ class BREADTH ## FirstSearch : \ public BREADTH ## FirstSearch_Body \ { \ \ public: \ \ template inline BREADTH ## FirstSearch( const Args&... args ); \ \ private: \ inline list e( const int& t ); \ \ }; \ \ template E(const int&)> void BREADTH ## FirstConnectedComponent( const int& V , int ( &vertex )[V_max] , int& count ); \ #define DEFINITION_OF_FIRST_SEARCH( BREADTH , PUSH ) \ template inline BREADTH ## FirstSearch_Body::BREADTH ## FirstSearch_Body( const int& V ) : m_V( V ) , m_init() , m_next() , m_found() , m_prev() { assert( m_V <= V_max ); for( int i = 0 ; i < m_V ; i++ ){ m_prev[i] = -1; } } \ template inline BREADTH ## FirstSearch_Body::BREADTH ## FirstSearch_Body( const int& V , const int& init ) : BREADTH ## FirstSearch_Body( V ) { m_init = init; m_next.push_back( m_init ); m_found[m_init] = true; } \ template E(const int&)> template inline BREADTH ## FirstSearch::BREADTH ## FirstSearch( const Args&... args ) : BREADTH ## FirstSearch_Body( args... ) {} \ \ template inline void BREADTH ## FirstSearch_Body::Reset( const int& init ) { m_init = init; assert( m_init < m_V ); m_next.clear(); m_next.push_back( m_init ); for( int i = 0 ; i < m_V ; i++ ){ m_found[i] = i == m_init; m_prev[i] = -1; } } \ template inline void BREADTH ## FirstSearch_Body::Shift( const int& init ) { m_init = init; assert( m_init < m_V ); m_next.clear(); if( ! m_found[m_init] ){ m_next.push_back( m_init ); m_found[m_init] = true; } } \ \ template inline const int& BREADTH ## FirstSearch_Body::size() const { return m_V; } \ template inline const int& BREADTH ## FirstSearch_Body::init() const { return m_init; } \ template inline bool& BREADTH ## FirstSearch_Body::found( const int& i ) { assert( i < m_V ); return m_found[i]; } \ template inline const int& BREADTH ## FirstSearch_Body::prev( const int& i ) const { assert( i < m_V ); return m_prev[i]; } \ \ template \ int BREADTH ## FirstSearch_Body::Next() \ { \ \ if( m_next.empty() ){ \ \ return -1; \ \ } \ \ const int i_curr = m_next.front(); \ m_next.pop_front(); \ list edge = e( i_curr ); \ \ while( ! edge.empty() ){ \ \ const int& i = edge.front(); \ bool& found_i = found( i ); \ \ if( ! found_i ){ \ \ m_next.PUSH( i ); \ m_prev[i] = i_curr; \ found_i = true; \ \ } \ \ edge.pop_front(); \ \ } \ \ return i_curr; \ \ } \ \ template E(const int&)> inline list BREADTH ## FirstSearch::e( const int& t ) { return E( t ); } \ \ template E(const int&)> void BREADTH ## FirstConnectedComponentSearch( const int& V , int ( &vertex )[V_max] , int& count ) \ { \ \ BREADTH ## FirstSearch bfs{ V }; \ count = 0; \ \ for( int i = 0 ; i < V ; i++ ){ \ \ vertex[i] = -1; \ \ } \ \ for( int i = 0 ; i < V ; i++ ){ \ \ if( vertex[i] == -1 ){ \ \ bfs.Shift( i ); \ int j = bfs.Next(); \ \ while( j != -1 ? vertex[j] == 0 : false ){ \ \ vertex[j] = count; \ j = bfs.Next(); \ \ } \ \ count++; \ \ } \ \ } \ \ return; \ \ } \ DECLARATION_OF_FIRST_SEARCH( Depth ); DEFINITION_OF_FIRST_SEARCH( Depth , push_front ); // digitはAncestorとLCAにのみ使用。普段は0で良い。 // 2^16 = 65536 // 2^17 = 131072 // 2^18 = 262144 template E(const int&),int digit = 0> class DepthFirstSearchOnTree : public DepthFirstSearch { private: // メモリが厳しい場合、以下で不要なものを消す。 int m_reversed[V_max]; vector m_children[V_max]; int m_children_num[V_max]; bool m_set_children; int m_depth[V_max]; int m_height[V_max]; bool m_set_height; int m_weight[V_max]; bool m_set_weight; int m_doubling[digit][V_max]; bool m_set_doubling; public: inline DepthFirstSearchOnTree( const int& V , const int& root ); inline void Reset( const int& init ) = delete; inline void Shift( const int& init ) = delete; inline const int& Root() const; inline const int& Parent( const int& i ) const; inline const vector& Children( const int& i ); inline const int& Depth( const int& i ) const; inline const int& Height( const int& i ); inline const int& Weight( const int& i ); // 探索順にノードを番号づける。 inline const int& NodeNumber( const int& i , const bool& reversed = false ) const; // 共通の親を持つノード間で昇順に番号づける。 inline const int& ChildrenNumber( const int& i ); // 各ノードの高さ < 2^digitの場合のみサポート。 int Ancestor( int i , int n ); int LCA( int i , int j ); int LCA( int i , int j , int& i_prev , int& j_prev ); // (T,m_T:T^2->T,e_T:1->T)が可換モノイドの場合のみサポート。 // dp[i][j] = f(iを根とみなした時のjの子ノードkを渡るdp[i][k]のm_Tに関する添字昇順の積,j) // を満たす二重配列dpの対角成分dp[i][i]をO(m_V)で求めてdに格納する。 template void RerootingDP( T ( &d )[V_max] ); private: void SetChildren(); void SetHeight(); void SetWeight(); // 各ノードの高さ < 2^digitの場合のみサポート。 // LCA()を呼ぶ前にAncestor()経由で完全にダブリングを設定するため、 // 遅延評価する../../../../Mathematics/Function/Iteration/Doubling/のダブリングで代用しない。 void SetDoubling(); }; template E(const int&),int digit> inline DepthFirstSearchOnTree::DepthFirstSearchOnTree( const int& V , const int& root ) : DepthFirstSearch( V , root ) , m_reversed() , m_children() , m_set_children() , m_depth() , m_height() , m_set_height() , m_weight() , m_set_weight() , m_doubling() , m_set_doubling() { int n = DepthFirstSearch::size(); while( --n >= 0 ){ const int& i = m_reversed[n] = DepthFirstSearch::Next(); const int& j = Parent( i ); if( j != -1 ){ m_depth[i] = m_depth[j] + 1; } } } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::Root() const { return DepthFirstSearch::init(); } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::Parent( const int& i ) const { return DepthFirstSearch::prev( i ); } template E(const int&),int digit> inline const vector& DepthFirstSearchOnTree::Children( const int& i ) { if( ! m_set_children ){ SetChildren(); } return m_children[i]; } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::Depth( const int& i ) const { return m_depth[i]; } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::Height( const int& i ) { if( ! m_set_height ){ SetHeight(); } return m_height[i]; } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::Weight( const int& i ) { if( ! m_set_weight ){ SetWeight(); } return m_weight[i]; } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::NodeNumber( const int& i , const bool& reversed ) const { return m_reversed[reversed ? i : DepthFirstSearch::size() - 1 - i]; } template E(const int&),int digit> inline const int& DepthFirstSearchOnTree::ChildrenNumber( const int& i ) { if( ! m_set_children ){ SetChildren(); } return m_children_num[i]; } template E(const int&),int digit> int DepthFirstSearchOnTree::Ancestor( int i , int n ) { if( ! m_set_doubling ){ SetDoubling(); } assert( ( n >> digit ) == 0 ); int d = 0; while( n != 0 ){ if( ( n & 1 ) == 1 ){ assert( ( i = m_doubling[d][i] ) != -1 ); } d++; n >>= 1; } return i; } template E(const int&),int digit> int DepthFirstSearchOnTree::LCA( int i , int j ) { int diff = Depth( i ) - Depth( j ); if( diff < 0 ){ swap( i , j ); diff *= -1; } i = Ancestor( i , diff ); if( i == j ){ return i; } int d = digit; while( --d >= 0 ){ const int ( &doubling_d )[V_max] = m_doubling[d]; const int& doubling_d_i = doubling_d[i]; const int& doubling_d_j = doubling_d[j]; if( doubling_d_i != doubling_d_j ){ i = doubling_d_i; j = doubling_d_j; assert( i != -1 ); assert( j != -1 ); } } return Parent( i ); } template E(const int&),int digit> int DepthFirstSearchOnTree::LCA( int i , int j , int& i_prev , int& j_prev ) { if( i == j ){ i_prev = j_prev = -1; return i; } int diff = Depth( i ) - Depth( j ); if( diff < 0 ){ return LCA( j , i , j_prev , i_prev ); } if( diff > 0 ){ i_prev = Ancestor( i , diff - 1 ); i = Parent( i_prev ); assert( i != -1 ); if( i == j ){ j_prev = -1; return i; } } else if( ! m_set_doubling ){ SetDoubling(); } int d = digit; while( --d >= 0 ){ const int ( &doubling_d )[V_max] = m_doubling[d]; const int& doubling_d_i = doubling_d[i]; const int& doubling_d_j = doubling_d[j]; if( doubling_d_i != doubling_d_j ){ i = doubling_d_i; j = doubling_d_j; assert( i != -1 ); assert( j != -1 ); } } i_prev = i; j_prev = j; return Parent( i_prev ); } template E(const int&),int digit> void DepthFirstSearchOnTree::SetChildren() { assert( !m_set_children ); m_set_children = true; const int& V = DepthFirstSearch::size(); for( int i = 0 ; i < V ; i++ ){ const int& j = Parent( i ); if( j == -1 ){ m_children_num[i] = -1; } else { vector& m_children_j = m_children[j]; m_children_num[i] = m_children_j.size(); m_children_j.push_back( i ); } } return; } template E(const int&),int digit> void DepthFirstSearchOnTree::SetHeight() { assert( !m_set_height ); m_set_height = true; const int& V = DepthFirstSearch::size(); for( int i = 0 ; i < V ; i++ ){ const int& reversed_i = m_reversed[i]; const int& parent_i = Parent( reversed_i ); if( parent_i != -1 ){ int& height_parent_i = m_height[parent_i]; const int& height_i = m_height[reversed_i]; height_parent_i > height_i ? height_parent_i : height_parent_i = height_i + 1; } } return; } template E(const int&),int digit> void DepthFirstSearchOnTree::SetWeight() { assert( !m_set_weight ); m_set_weight = true; const int& V = DepthFirstSearch::size(); for( int i = 0 ; i < V ; i++ ){ const int& reversed_i = m_reversed[i]; const int& parent_i = Parent( reversed_i ); if( parent_i != -1 ){ m_weight[parent_i] += m_weight[reversed_i] + 1; } } return; } template E(const int&),int digit> void DepthFirstSearchOnTree::SetDoubling() { assert( !m_set_doubling ); m_set_doubling = true; const int& V = DepthFirstSearch::size(); { int ( &doubling_0 )[V_max] = m_doubling[0]; const int& r = Root(); for( int i = 0 ; i < V ; i++ ){ doubling_0[i] = Parent( i ); } } for( int d = 1 ; d < digit ; d++ ){ int ( &doubling_d )[V_max] = m_doubling[d]; int ( &doubling_d_minus )[V_max] = m_doubling[d-1]; for( int i = 0 ; i < V ; i++ ){ const int& doubling_d_minus_i = doubling_d_minus[i]; doubling_d[i] = doubling_d_minus_i == -1 ? -1 : doubling_d_minus[doubling_d_minus_i]; } } return; } template E(const int&),int digit> template void DepthFirstSearchOnTree::RerootingDP( T ( &d )[V_max] ) { if( ! m_set_children ){ SetChildren(); } const int& V = DepthFirstSearch::size(); const T& e = e_T(); vector children_value[V_max] = {}; vector left_sum[V_max] = {}; vector right_sum[V_max] = {}; for( int i = 0 ; i < V ; i++ ){ children_value[i].resize( m_children[i].size() ); } for( int n = 0 ; n < V ; n++ ){ const int& i = NodeNumber( n , true ); const vector& children_value_i = children_value[i]; const int size_i = children_value_i.size(); T temp = e; vector& left_sum_i = left_sum[i]; left_sum_i.reserve( size_i + 1 ); left_sum_i.push_back( temp ); for( int m = 0 ; m < size_i ; m++ ){ left_sum_i.push_back( temp = m_T( temp , children_value_i[m] ) ); } const int& j = Parent( i ); if( j != -1 ){ children_value[j][m_children_num[i]] = f( temp , i ); } temp = e; vector& right_sum_i = right_sum[i]; right_sum_i.resize( size_i ); for( int m = 1 ; m <= size_i ; m++ ){ right_sum_i[ size_i - m ] = temp; temp = m_T( children_value_i[size_i - m] , temp ); } } for( int n = 1 ; n < V ; n++ ){ const int& i = NodeNumber( n ); const int& j = Parent( i ); const int& k = ChildrenNumber( i ); vector& left_sum_i = left_sum[i]; vector& right_sum_i = right_sum[i]; const int size_i = right_sum_i.size(); const T rest_i = f( m_T( left_sum[j][k] , right_sum[j][k] ) , j ); for( int m = 0 ; m <= size_i ; m++ ){ T& left_sum_im = left_sum_i[m]; left_sum_im = m_T( rest_i , left_sum_im ); } } for( int i = 0 ; i < V ; i++ ){ d[i] = f( left_sum[i].back() , i ); } return; } inline DEXPR( int , bound_N , 100000 , 100 ); // 0が5個 list e[bound_N] = {}; inline list E( const int& i ) { return e[i]; } inline CEXPR( ll , P , 998244353 ); inline pair add( const pair& t0 , const pair& t1 ) { pair answer{ t0.first + t1.first , t0.second + t1.second }; answer.first < P ? answer.first : answer.first -= P; return answer; } inline const pair& zero() { static const pair z{ 0 , 0 }; return z; } ll A[bound_N]; int length[bound_N]; ll power_10[11] = { 1 }; inline pair f( const pair& t , const int& i ) { return { ( t.first * power_10[length[i]] + A[i] * ( t.second + 1 ) ) % P , t.second + 1 }; } int main() { UNTIE; LIBRARY_SEARCH; START_MAIN; CIN_ASSERT( N , 0 , bound_N ); FOR( i , 0 , N ){ CIN( string , Ai ); assert( ( length[i] = Ai.size() ) <= 10 ); assert( ( A[i] = stoll( Ai ) ) > 0 ); } FOR( i , 1 , N ){ CIN_ASSERT( ui , 1 , N ); CIN_ASSERT( vi , 1 , N ); assert( --ui != --vi ); e[ui].push_back( vi ); e[vi].push_back( ui ); } FOREQ( i , 1 , 10 ){ power_10[i] = power_10[i-1] * 10 % P; } DepthFirstSearchOnTree dfst{ N , 0 }; pair d[bound_N]; dfst.RerootingDP,add,zero,f>( d ); ll answer = 0; FOR( i , 0 , N ){ ( answer += d[i].first ) < P ? answer : answer -= P; } COUT( answer ); FINISH_MAIN; QUIT; }