結果
問題 | No.2136 Dice Calendar? |
ユーザー | CuriousFairy315 |
提出日時 | 2022-10-21 15:37:37 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 913 ms / 5,000 ms |
コード長 | 2,743 bytes |
コンパイル時間 | 1,070 ms |
コンパイル使用メモリ | 85,256 KB |
実行使用メモリ | 94,140 KB |
最終ジャッジ日時 | 2023-09-13 16:37:56 |
合計ジャッジ時間 | 8,049 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 37 ms
93,860 KB |
testcase_01 | AC | 38 ms
93,912 KB |
testcase_02 | AC | 51 ms
93,892 KB |
testcase_03 | AC | 38 ms
93,936 KB |
testcase_04 | AC | 39 ms
93,896 KB |
testcase_05 | AC | 43 ms
93,856 KB |
testcase_06 | AC | 43 ms
93,852 KB |
testcase_07 | AC | 46 ms
94,084 KB |
testcase_08 | AC | 49 ms
93,904 KB |
testcase_09 | AC | 53 ms
93,852 KB |
testcase_10 | AC | 56 ms
93,848 KB |
testcase_11 | AC | 66 ms
93,896 KB |
testcase_12 | AC | 76 ms
93,852 KB |
testcase_13 | AC | 67 ms
93,860 KB |
testcase_14 | AC | 79 ms
93,900 KB |
testcase_15 | AC | 153 ms
93,932 KB |
testcase_16 | AC | 186 ms
93,856 KB |
testcase_17 | AC | 150 ms
93,912 KB |
testcase_18 | AC | 356 ms
93,884 KB |
testcase_19 | AC | 433 ms
93,880 KB |
testcase_20 | AC | 510 ms
94,088 KB |
testcase_21 | AC | 632 ms
94,140 KB |
testcase_22 | AC | 764 ms
93,848 KB |
testcase_23 | AC | 913 ms
93,856 KB |
testcase_24 | AC | 59 ms
93,916 KB |
testcase_25 | AC | 77 ms
93,916 KB |
testcase_26 | AC | 822 ms
93,944 KB |
ソースコード
#pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include<iostream> #include<vector> #include<bitset> using namespace std; using ll = long long int; typedef bitset<1 << 29> bs; // 64MB程度 int deBrujin32[32] = { 0, 1, 28, 2, 29, 14, 24, 3, 30, 22, 20, 15, 25, 17, 4, 8, 31, 27, 13, 23, 21, 19, 16, 7, 26, 12, 18, 6, 11, 5, 10, 9 }; int lsb(unsigned int i) { // 丁度1bit立っている値に対してその立っている位置を返す return deBrujin32[i * 0x077CB531 >> 27]; } ll calc_partition(int multiset) { // 与えられた多重集合に対して、立っているbitの位置を保持する数列Pを返す ll partition = 0; for (int i = 5; i <= 45; i += 5) { int lob = multiset & -multiset; partition += (1LL + lsb(lob)) << i; multiset -= lob; } return partition; } int get_partition(ll partition, int index) { // multiSetでindex番目に立っているbitの位置を求める return (int) (partition >> 5 * index & 0b11111); } ll multichoose(ll *factorial, int multiset) { // multiSetで与えられた多重集合を並べてできる組合せ ll partition = calc_partition(multiset); ll multichoose = factorial[get_partition(partition, 9) - 9]; for (int i = 0; i < 9; ++i) multichoose /= factorial[get_partition(partition, i + 1) - get_partition(partition, i) - 1]; return multichoose; } int next(int multiset, vector<int> &dice, bs &check_unique, int *stack, int length) {// diceを追加したときにできる新たな多重集合のうち、新しく発見したものをstackに入れる ll partition = calc_partition(multiset); for (int result : dice) { int mask = (1 << get_partition(partition, result)) - 1; int next = (multiset & ~mask) << 1 | (multiset & mask); if (check_unique[next]) continue; check_unique.set(next); stack[length++] = next; } return length; } int main() { int N; cin >> N; vector<vector<int>> S(N, vector<int>(6)); for (int i = 0;i < N;++ i) for (int j = 0;j < 6;++ j) { cin >> S[i][j]; -- S[i][j]; } ll factorial[21] = {1}; for (int i = 1;i <= 20;++ i) factorial[i] = factorial[i - 1] * i; int now_stack[3200000] = {0b111111111}, next_stack[3200000]; // 初項M_0を求める int now_index = 1, next_index = 0; bs unique_check(0); for (vector<int> dice : S) { for (int i = 0;i < now_index;++ i) next_index = next(now_stack[i], dice, unique_check, next_stack, next_index); // M_iからM_{i+1}を求める swap(now_stack, next_stack); now_index = next_index; next_index = 0; } int ans = 0; for (int i = 0;i < now_index;++ i) ans = (int)((ans + multichoose(factorial, now_stack[i])) % 998'244'353); cout << ans << endl; }