結果

問題 No.2136 Dice Calendar?
ユーザー CuriousFairy315CuriousFairy315
提出日時 2022-10-21 15:16:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 4,306 ms / 5,000 ms
コード長 1,975 bytes
コンパイル時間 278 ms
コンパイル使用メモリ 86,888 KB
実行使用メモリ 327,944 KB
最終ジャッジ日時 2023-09-13 16:19:54
合計ジャッジ時間 28,049 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 69 ms
71,328 KB
testcase_01 AC 70 ms
70,900 KB
testcase_02 AC 130 ms
78,348 KB
testcase_03 AC 69 ms
71,304 KB
testcase_04 AC 70 ms
71,084 KB
testcase_05 AC 104 ms
77,460 KB
testcase_06 AC 95 ms
77,100 KB
testcase_07 AC 105 ms
77,424 KB
testcase_08 AC 111 ms
77,368 KB
testcase_09 AC 119 ms
78,480 KB
testcase_10 AC 126 ms
78,524 KB
testcase_11 AC 155 ms
81,284 KB
testcase_12 AC 180 ms
83,072 KB
testcase_13 AC 158 ms
81,320 KB
testcase_14 AC 213 ms
85,976 KB
testcase_15 AC 531 ms
135,888 KB
testcase_16 AC 756 ms
166,636 KB
testcase_17 AC 647 ms
144,196 KB
testcase_18 AC 1,652 ms
269,000 KB
testcase_19 AC 2,366 ms
301,528 KB
testcase_20 AC 2,316 ms
268,104 KB
testcase_21 AC 3,927 ms
313,520 KB
testcase_22 AC 4,204 ms
327,944 KB
testcase_23 AC 4,306 ms
272,688 KB
testcase_24 AC 80 ms
103,960 KB
testcase_25 AC 191 ms
114,392 KB
testcase_26 AC 3,633 ms
286,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
S = [list(map(lambda x: int(x) - 1, input().split())) for _ in range(N)]
factorial = [1]
for i in range(N): factorial.append(factorial[i] * (i + 1))

partition = [0] * 10
partition[9] = N + 9

deBrujin32 = [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]

def numberOfTrailingZeros(i): return deBrujin32[i * 0x077CB531 >> 27 & 0b11111] # 丁度1bit立っている値に対してその立っている位置を返す

def calcPartition(multiSet): # 与えられた多重集合に対して、立っているbitの位置を保持する数列Pを返す
	for i in range(1, 9):
		lob = multiSet & -multiSet
		partition[i] = numberOfTrailingZeros(lob) + 1
		multiSet -= lob

def multichoose(multiSet): # multiSetで与えられた多重集合を並べてできる組合せ
	calcPartition(multiSet)
	ret = factorial[N]
	for i in range(9): ret //= factorial[partition[i + 1] - partition[i] - 1]
	return ret

def nextSet(multiSet, dice, uniqueCheck, nextList): # diceを追加したときの多重集合をnextListに入れる
	calcPartition(multiSet)
	for result in dice: # 出目がresultだった時
		mask = (1 << partition[result]) - 1
		nextSet = (multiSet & 0x1FFFFFFF - mask) << 1 | (multiSet & mask)
		if (uniqueCheck[nextSet >> 6] & 1 << (nextSet & 0x3F)) == 0: # まだこの多重集合を計算対象にしていないなら
			uniqueCheck[nextSet >> 6] |= 1 << (nextSet & 0x3F)
			nextList.append(nextSet)

nowList, nextList = [0b11111111], [] # 初項M_0を求める
uniqueCheck = [0] * (1 << N + 2) # 既に調べた多重集合を管理するためのBitSet、32MB程度
for dice in S:
	nextList.clear()
	for i in nowList: uniqueCheck[i >> 6] = 0
	for multiSet in nowList: nextSet(multiSet, dice, uniqueCheck, nextList) # M_iからM_{i+1}を求める
	nowList, nextList = nextList, nowList

ans = 0
for multiSet in nowList: ans += multichoose(multiSet)
ans %= 998_244_353
print(ans)
0