結果
問題 |
No.462 6日知らずのコンピュータ
|
ユーザー |
|
提出日時 | 2016-12-13 17:51:47 |
言語 | Java (openjdk 23) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,686 bytes |
コンパイル時間 | 5,240 ms |
コンパイル使用メモリ | 84,852 KB |
実行使用メモリ | 41,892 KB |
最終ジャッジ日時 | 2024-11-30 01:11:34 |
合計ジャッジ時間 | 19,081 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 54 WA * 30 |
ソースコード
package yukicoder; import java.util.*; public class Q462 { final static long MODULO = 1_000_000_000 + 7; public static void main(String[] args) { long[] fac = new long[100]; long[] invfac = new long[100]; fac[0] = 1; for (int i = 1; i < 100; ++i) { fac[i] = fac[i - 1] * i % MODULO; } invfac[invfac.length - 1] = pow(fac[fac.length - 1], MODULO - 2); for (int i = invfac.length - 2; i >= 0; --i) { invfac[i] = invfac[i + 1] * (i + 1) % MODULO; } long[][] dp = new long[100][]; // dp[i][j]:i個からj個数を作る方法 for (int i = 0; i < 100; ++i) { dp[i] = new long[i + 1]; dp[i][0] = 1; for (int j = 1; j <= i; ++j) { dp[i][j] = dp[i - 1][j - 1] * j % MODULO; if (j <= i - 1) dp[i][j] += dp[i - 1][j] * (j + 1) % MODULO; dp[i][j] %= MODULO; } } long[][] dp3 = new long[100][100]; dp3[1][0] = 1; for (int i = 2; i < 100; ++i) { for (int j = 0; j <= i; ++j) { if (j > 0) dp3[i][j] = dp3[i - 1][j - 1] * j % MODULO; if (j <= i - 1){ dp3[i][j] += (dp3[i - 1][j] * j + dp[i - 1][j]) % MODULO; } dp3[i][j] %= MODULO; } } dp3[0][0] = 1; Scanner sc = new Scanner(System.in); int n = sc.nextInt(); int k = sc.nextInt(); if (k == 0) { System.out.println(dp[n][n] % MODULO); return; } long[] a = new long[k]; long[] diff = new long[k + 1]; for (int i = 0; i < k; ++i) { a[i] = sc.nextLong(); } for (int i = 0; i + 1 < k; ++i) { diff[i + 1] = count(a[i], a[i + 1]); if (diff[i + 1] < 0) { System.out.println(0); return; } } diff[0] = count(0, a[0]); diff[k] = count(a[k - 1], (1L << n) - 1); long[] dp1 = new long[n + 1]; dp1[0] = 1; if (a[0] != 0) dp1[1] = 1; for (int i = 0; i < k + 1; ++i) { for (int j = n; j >= 1; --j) { boolean f = false; if ((a[k - 1] != (1L << n) - 1 && i == k)) f = true; for (int l = 1; l <= diff[i] + (f ? 0 : -1) && j - l >= 0; ++l) { if (f && l == diff[i]) { dp1[j] += dp1[j - l] * dp[(int) diff[i]][l] % MODULO; } else dp1[j] += dp1[j - l] * dp3[(int) diff[i]][l] % MODULO; dp1[j] %= MODULO; } } } System.out.println(dp1[n + 1 - k]); } static long count(long min, long max) { for (int i = 0; i < 60; ++i) { if ((min & (1L << i)) > 0 && (max & (1L << i)) == 0) { return -1; } } int diff = Long.bitCount(max) - Long.bitCount(min); return diff; } static void tr(Object... objects) { System.out.println(Arrays.deepToString(objects)); } static long pow(long a, long n) { long ret = 1; for (; n > 0; n >>= 1, a = (a * a) % MODULO) { if (n % 2 == 1) ret = (ret * a) % MODULO; } return ret; } }