結果
問題 | No.1035 Color Box |
ユーザー |
|
提出日時 | 2020-09-22 00:32:44 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 58 ms / 2,000 ms |
コード長 | 1,361 bytes |
コンパイル時間 | 1,102 ms |
コンパイル使用メモリ | 78,152 KB |
実行使用メモリ | 6,400 KB |
最終ジャッジ日時 | 2024-06-25 00:00:23 |
合計ジャッジ時間 | 2,784 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
#include <algorithm>#include <cmath>#include <cstdio>#include <deque>#include <iostream>#include <map>#include <numeric>#include <queue>#include <set>#include <stack>#include <string>#include <vector>using namespace std;typedef long long ll;#define MOD 1000000007// x^nll mod_pow(ll x, ll n) {if (n == 0) return 1;ll res = mod_pow(x * x % MOD, n / 2);if (n & 1) res = res * x % MOD;return res;}// x^{-1}ll mod_inv(ll x) { return mod_pow(x, MOD - 2); }vector<ll> fact;vector<ll> fact_inv;void init_fact(int n) {fact.resize(n + 1);fact_inv.resize(n + 1);fact[0] = fact_inv[0] = 1;for (int i = 0; i < n; ++i) {fact[i + 1] = fact[i] * (i + 1) % MOD;fact_inv[i + 1] = fact_inv[i] * mod_inv(i + 1) % MOD;}}ll nCr(ll n, ll r) {if (n < r || n < 0 || r < 0) return 0;return fact[n] * fact_inv[r] % MOD * fact_inv[n - r] % MOD;}ll nPr(ll n, ll r) { return nCr(n, r) * fact[r] % MOD; }int main() {int n, m;cin >> n >> m;init_fact(n + m);ll ans = mod_pow(m, n);ll sub = 0;bool flag = true;for (int i = 1; i <= m - 1; ++i) {ll res = mod_pow(m - i, n);res = (res * nCr(m, i)) % MOD;if (flag) {sub = (sub + res) % MOD;} else {sub = (sub + MOD - res) % MOD;}flag = !flag;}cout << (ans + MOD - sub) % MOD << endl;return 0;}