結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2020-04-24 21:42:09 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 47 ms / 2,000 ms |
コード長 | 1,341 bytes |
コンパイル時間 | 578 ms |
コンパイル使用メモリ | 65,024 KB |
実行使用メモリ | 26,976 KB |
最終ジャッジ日時 | 2024-10-15 02:37:31 |
合計ジャッジ時間 | 3,055 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
// ボール: 区別あり, 箱:区別あり, 入れ方: 1 つ以上#include <iostream>using namespace std;const int MOD = 1000000007;const long long MAXN = 1001024;// 繰り返し 2 乗法long long mod_pow(long long a, long long n) {long long res = 1;while (n > 0) {if (n & 1) res = res * a % MOD;a = a * a % MOD;n >>= 1;}return res;}long long fac[MAXN], finv[MAXN], inv[MAXN];// 前処理 O(n)void comb_init(){fac[0] = fac[1] = 1;finv[0] = finv[1] = 1;inv[1] = 1;for(long long i = 2; i < MAXN; i++){fac[i] = fac[i - 1] * i % MOD;inv[i] = MOD - inv[MOD%i] * (MOD / i) % MOD;finv[i] = finv[i - 1] * inv[i] % MOD;}}// 二項係数計算 O(1)long long mod_comb(long long n, long long k){if (n < k) return 0;if (n < 0 || k < 0) return 0;return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;}int main() {long long n, k; cin >> n >> k;long long ans = 0;comb_init();for(int i = 0; i <= k; i++) {// i 個の箱を選んで, 絶対ボールを入れないことにする// 残りの箱に自由に入れるint rest = k - i;long long add = mod_comb(k, i) * mod_pow(rest, n) % MOD;if(i % 2 == 0) ans = (ans + add ) % MOD;else ans = (ans - add + MOD) % MOD;}cout << ans << endl;return 0;}