結果
問題 | No.391 CODING WAR |
ユーザー | Grenache |
提出日時 | 2016-07-10 15:34:20 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 197 ms / 2,000 ms |
コード長 | 4,164 bytes |
コンパイル時間 | 3,784 ms |
コンパイル使用メモリ | 78,564 KB |
実行使用メモリ | 54,700 KB |
最終ジャッジ日時 | 2024-10-13 10:15:41 |
合計ジャッジ時間 | 7,119 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 106 ms
54,416 KB |
testcase_01 | AC | 105 ms
54,492 KB |
testcase_02 | AC | 107 ms
54,260 KB |
testcase_03 | AC | 109 ms
54,384 KB |
testcase_04 | AC | 109 ms
54,368 KB |
testcase_05 | AC | 52 ms
36,924 KB |
testcase_06 | AC | 110 ms
54,408 KB |
testcase_07 | AC | 109 ms
54,404 KB |
testcase_08 | AC | 108 ms
54,208 KB |
testcase_09 | AC | 197 ms
54,620 KB |
testcase_10 | AC | 188 ms
54,520 KB |
testcase_11 | AC | 50 ms
37,112 KB |
testcase_12 | AC | 111 ms
54,372 KB |
testcase_13 | AC | 189 ms
54,372 KB |
testcase_14 | AC | 176 ms
54,496 KB |
testcase_15 | AC | 187 ms
54,520 KB |
testcase_16 | AC | 160 ms
54,376 KB |
testcase_17 | AC | 165 ms
54,700 KB |
testcase_18 | AC | 149 ms
54,504 KB |
testcase_19 | AC | 151 ms
54,444 KB |
ソースコード
import java.io.*; import java.util.*; public class Main_yukicoder391 { public static void main(String[] args) { Scanner sc = new Scanner(System.in); Printer pr = new Printer(System.out); long n = sc.nextLong(); int m = sc.nextInt(); if (m > n) { pr.println(0); } else { PC pc = new PC(1_000_000, 1_000_000_007); pr.println(pc.S(n, m) * pc.fact[m] % pc.MOD); } sc.close(); pr.close(); } @SuppressWarnings("unused") private static class PC { // MOD must be prime number. int MOD; // fact[i] : i! % MOD long[] fact; // ifact[i] : 1/i! % MOD long[] ifact; PC(int size, int MOD) { // O(size) // n=sizeまでのnCrを求める。 // nHrはn+r-1Crになってしまうので注意 this.MOD = MOD; fact = new long[size + 1]; fact[0] = 1; for (int i = 1; i <= size; i++) { fact[i] = fact[i - 1] * i % MOD; } ifact = new long[size + 1]; int loop = MOD - 2; long x = fact[size]; ifact[size] = 1; while (loop > 0) { if (loop % 2 == 1) { ifact[size] = ifact[size] * x % MOD; } x = x * x % MOD; loop /= 2; } for (int i = size - 1; i >= 0; i--) { ifact[i] = ifact[i + 1] * (i + 1) % MOD; } } // 組合せの数 int C(int n, int r) { if (r > n) { return 0; } return (int)(((fact[n] * ifact[n - r]) % MOD) * ifact[r] % MOD); } // 順列 int P(int n, int r) { if (r > n) { return 0; } return (int)((fact[n] * ifact[n -r]) % MOD); } // 重複組み合わせ // 異なるn種のものから重複を許してr個を選ぶ場合の数 // 0個の種類もあり得る int H(int n, int r) { if (n == 0 && r == 0) { return 1; } return C(n + r - 1, r); } // 組合せの数(nが大きいとき) // O(r)で求めることができる。rはsizeの大きさまで int C2(long n, int r) { long ret = ifact[r]; for (int i = 1; i <= r; i++) { long tmp = (n - r + i) % MOD; ret = (ret * tmp) % MOD; } return (int)ret; } // 第2種スターリング数 // n人をちょうどr個のグループに分ける(グループの区別はなし) // グループの区別をする場合はr!S(n,r)。全射の場合の数と同義 // O(r log n) int S(long n, int r) { //全射の場合の数を包除原理を使って求めて、1/r!をかける。 long ret = 0; for (int i = 1; i <= r; i++) { long tmp = (r - i) % 2 == 0 ? 1 : -1; tmp *= pow(i, n) * C(r, i) % MOD; ret = (ret + tmp + MOD) % MOD; } ret = ret * ifact[r] % MOD; return (int)ret; } long pow(int a, long n) { long loop = n; long ret = 1; long x = a; while (loop > 0) { if (loop % 2 == 1) { ret = ret * x % MOD; } x = x * x % MOD; loop /= 2; } return ret; } } @SuppressWarnings("unused") private static class Scanner { BufferedReader br; Iterator<String> it; Scanner (InputStream in) { br = new BufferedReader(new InputStreamReader(in)); } String next() throws RuntimeException { try { if (it == null || !it.hasNext()) { it = Arrays.asList(br.readLine().split(" ")).iterator(); } return it.next(); } catch (IOException e) { throw new IllegalStateException(); } } int nextInt() throws RuntimeException { return Integer.parseInt(next()); } long nextLong() throws RuntimeException { return Long.parseLong(next()); } float nextFloat() throws RuntimeException { return Float.parseFloat(next()); } double nextDouble() throws RuntimeException { return Double.parseDouble(next()); } void close() { try { br.close(); } catch (IOException e) { // throw new IllegalStateException(); } } } private static class Printer extends PrintWriter { Printer(PrintStream out) { super(out); } } }