結果

問題 No.2136 Dice Calendar?
コンテスト
ユーザー CuriousFairy315
提出日時 2022-11-23 17:05:51
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 1,769 ms / 5,000 ms
コード長 2,798 bytes
コンパイル時間 1,789 ms
コンパイル使用メモリ 116,440 KB
実行使用メモリ 110,760 KB
最終ジャッジ日時 2024-09-24 20:52:24
合計ジャッジ時間 12,886 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
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