結果

問題 No.2136 Dice Calendar?
コンテスト
ユーザー CuriousFairy315
提出日時 2022-10-21 07:56:55
言語 Java
(openjdk 23)
結果
AC  
実行時間 1,332 ms / 5,000 ms
コード長 3,233 bytes
コンパイル時間 2,617 ms
コンパイル使用メモリ 92,528 KB
実行使用メモリ 144,360 KB
最終ジャッジ日時 2024-06-30 19:57:06
合計ジャッジ時間 14,580 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.BitSet;
import java.util.Scanner;
import java.util.function.IntConsumer;

public class Main {
	public static void main(String[] args) {
		try (Scanner sc = new Scanner(System.in)) {
			int N = sc.nextInt();
			int[][] S = new int[N][6];
			for (int i = 0; i < N; ++i) for (int j = 0; j < 6; ++j) S[i][j] = sc.nextInt() - 1;

			final int MOD = 998_244_353;
			long[] factorial = new long[N + 1];
			factorial[0] = 1;
			for (int i = 1; i < factorial.length; ++i) factorial[i] = factorial[i - 1] * i;

			BitSet set = new BitSet(1 << N + 9); // 64MB程度
			IntStack now = new IntStack(3200000), next = new IntStack(3200000);
			now.add(0b111111111); // 初項M_0を求める
			for (int[] dice : S) {
				IntStack put = next;
				now.forEach(multiSet -> next(multiSet, dice, set, put)); // M_iからM_{i+1}を求める
				IntStack swap = now;
				now = next;
				next = swap;
				next.clear();
			}

			int ans = 0;
			for (int i = 0; i < now.size; ++i) ans = (int) ((ans + multichoose(factorial, now.stack[i])) % MOD);
			System.out.println(ans);
		}
	}

	private static int[] 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 };

	public static int numberOfTrailingZeros(int i) { // 丁度1bit立っている値に対してその立っている位置を返す
		if (i == 0) return 32;
		return deBrujin32[i * 0x077CB531 >>> 27];
	}

	public static long calcPartition(int multiSet) { // 与えられた多重集合に対して、立っているbitの位置を保持する数列Pを返す
		long partition = 0;
		for (int i = 5; i <= 45; i += 5) {
			int lob = multiSet & -multiSet;
			partition += 1L + numberOfTrailingZeros(lob) << i;
			multiSet -= lob;
		}
		return partition;
	}

	public static int getPartition(long partition, int index) { // multiSetでindex番目に立っているbitの位置を求める
		return (int) (partition >> 5 * index & 0b11111);
	}

	public static long multichoose(long[] factorial, int multiSet) { // multiSetで与えられた多重集合を並べてできる組合せ
		long partition = calcPartition(multiSet);
		long multichoose = factorial[getPartition(partition, 9) - 9];
		for (int i = 0; i < 9; ++i)
			multichoose /= factorial[getPartition(partition, i + 1) - getPartition(partition, i) - 1];
		return multichoose;
	}

	public static void next(int multiSet, int[] dice, BitSet set, IntStack stack) {// diceを追加したときにできる新たな多重集合のうち、新しく発見したものをstackに入れる
		long partition = calcPartition(multiSet);
		for (int result : dice) {
			int mask = (1 << getPartition(partition, result)) - 1;
			int next = (multiSet & ~mask) << 1 | multiSet & mask;
			if (!set.get(next)) {
				set.set(next);
				stack.add(next);
			}
		}
	}

	public static class IntStack {
		int[] stack;
		int size = 0;

		IntStack(int length) {
			stack = new int[length];
		}

		void add(int value) {
			stack[size++] = value;
		}

		int peek() {
			return stack[size - 1];
		}

		int poll() {
			return stack[--size];
		}

		void forEach(IntConsumer f) {
			for (int i = 0; i < size; ++i) f.accept(stack[i]);
		}

		void clear() {
			size = 0;
		}
	}
}
0