結果

問題 No.2136 Dice Calendar?
ユーザー CuriousFairy315CuriousFairy315
提出日時 2022-11-23 17:05:51
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 1,807 ms / 5,000 ms
コード長 2,798 bytes
コンパイル時間 1,805 ms
コンパイル使用メモリ 112,520 KB
実行使用メモリ 110,772 KB
最終ジャッジ日時 2023-10-25 01:55:22
合計ジャッジ時間 13,597 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 29 ms
25,524 KB
testcase_01 AC 30 ms
25,524 KB
testcase_02 AC 37 ms
25,524 KB
testcase_03 AC 28 ms
25,524 KB
testcase_04 AC 29 ms
25,524 KB
testcase_05 AC 29 ms
25,524 KB
testcase_06 AC 29 ms
25,524 KB
testcase_07 AC 30 ms
25,524 KB
testcase_08 AC 31 ms
25,524 KB
testcase_09 AC 36 ms
25,524 KB
testcase_10 AC 38 ms
25,524 KB
testcase_11 AC 58 ms
25,524 KB
testcase_12 AC 67 ms
25,524 KB
testcase_13 AC 58 ms
25,544 KB
testcase_14 AC 82 ms
25,848 KB
testcase_15 AC 248 ms
35,204 KB
testcase_16 AC 309 ms
37,296 KB
testcase_17 AC 235 ms
47,792 KB
testcase_18 AC 699 ms
53,936 KB
testcase_19 AC 846 ms
70,024 KB
testcase_20 AC 1,020 ms
74,568 KB
testcase_21 AC 1,244 ms
104,552 KB
testcase_22 AC 1,587 ms
107,984 KB
testcase_23 AC 1,807 ms
110,772 KB
testcase_24 AC 79 ms
90,328 KB
testcase_25 AC 115 ms
90,700 KB
testcase_26 AC 1,608 ms
109,000 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

class Program
{
	static void Main(string[] args)
	{
		int N = int.Parse(Console.ReadLine());
		int[][] S = new int[N][];
		for (int i = 0;i < N;++ i) {
			S[i] = Console.ReadLine().Split(' ').Select(x => int.Parse(x) - 1).ToArray();
		}
		
		const 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;
		
		BitArray set = new BitArray(1 << N + 9); // 64MB程度
		Stack<int> now = new Stack<int>(3200000), next = new Stack<int>(3200000);
		now.Push(0b111111111); // 初項M_0を求める
		foreach (int[] dice in S) {
			foreach (int multiSet in now) nextSet(multiSet, dice, set, next); // M_iからM_{i+1}を求める
			Stack<int> swap = now;
			now = next;
			next = swap;
			next.Clear();
		}
		
		int ans = 0;
		foreach (int multiSet in now) ans = (int) ((ans + multichoose(factorial, multiSet)) % MOD);
		Console.WriteLine(ans);
	}
	
	static readonly 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 };
	
	static int numberOfTrailingZeros(int i) { // 丁度1bit立っている値に対してその立っている位置を返す
		if (i == 0) return 32;
		return deBrujin32[(uint)(i * 0x077CB531) >> 27];
	}
	
	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;
	}
	
	static int getPartition(long partition, int index) { // multiSetでindex番目に立っているbitの位置を求める
		return (int) (partition >> 5 * index & 0b11111);
	}
	
	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;
	}
	
	static void nextSet(int multiSet, int[] dice, BitArray set, Stack<int> stack) {// diceを追加したときにできる新たな多重集合のうち、新しく発見したものをstackに入れる
		long partition = calcPartition(multiSet);
		foreach (int result in dice) {
			int mask = (1 << getPartition(partition, result)) - 1;
			int next = (multiSet & ~mask) << 1 | multiSet & mask;
			if (!set.Get(next)) {
				set.Set(next, true);
				stack.Push(next);
			}
		}
	}
}
0