結果

問題 No.1035 Color Box
ユーザー potoooooooopotoooooooo
提出日時 2020-04-24 21:44:35
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 61 ms / 2,000 ms
コード長 2,019 bytes
コンパイル時間 1,754 ms
コンパイル使用メモリ 166,916 KB
実行使用メモリ 10,368 KB
最終ジャッジ日時 2024-04-23 03:07:12
合計ジャッジ時間 3,785 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 60 ms
10,368 KB
testcase_01 AC 22 ms
10,236 KB
testcase_02 AC 23 ms
10,240 KB
testcase_03 AC 23 ms
10,164 KB
testcase_04 AC 24 ms
10,240 KB
testcase_05 AC 22 ms
10,240 KB
testcase_06 AC 21 ms
10,240 KB
testcase_07 AC 22 ms
10,240 KB
testcase_08 AC 56 ms
10,240 KB
testcase_09 AC 35 ms
10,240 KB
testcase_10 AC 22 ms
10,112 KB
testcase_11 AC 23 ms
10,240 KB
testcase_12 AC 21 ms
10,240 KB
testcase_13 AC 49 ms
10,112 KB
testcase_14 AC 52 ms
10,240 KB
testcase_15 AC 61 ms
10,368 KB
testcase_16 AC 22 ms
10,368 KB
testcase_17 AC 22 ms
10,240 KB
testcase_18 AC 21 ms
10,240 KB
testcase_19 AC 60 ms
10,240 KB
testcase_20 AC 22 ms
10,240 KB
testcase_21 AC 21 ms
10,112 KB
testcase_22 AC 21 ms
10,100 KB
testcase_23 AC 22 ms
10,240 KB
testcase_24 AC 22 ms
10,240 KB
testcase_25 AC 21 ms
10,348 KB
testcase_26 AC 22 ms
10,240 KB
testcase_27 AC 21 ms
10,368 KB
testcase_28 AC 23 ms
10,112 KB
testcase_29 AC 22 ms
10,240 KB
testcase_30 AC 21 ms
10,224 KB
testcase_31 AC 20 ms
10,128 KB
testcase_32 AC 21 ms
10,280 KB
testcase_33 AC 22 ms
10,240 KB
testcase_34 AC 21 ms
10,240 KB
testcase_35 AC 23 ms
10,240 KB
testcase_36 AC 21 ms
10,348 KB
testcase_37 AC 21 ms
10,240 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

template<int MAX>
struct Combination {
    vector<long long> _inv, _fac, _ifac;
    const long long MOD;
    Combination(long long mod) : _inv(MAX+1), _fac(MAX+1), _ifac(MAX+1), MOD(mod) { init(); }
    inline void init(long long mod = 0) { init_inv(), init_fac(); }

    inline void init_inv() {
        _inv[0] = 0;
        for (int i = 1; i <= MAX; i++) {
            if (i == 1) _inv[i] = 1;
            else {
                _inv[i] = (MOD - (MOD / i) * _inv[MOD % i]) % MOD;
                if (_inv[i] < 0) _inv[i] += MOD;
            }
        }
    }
    inline void init_fac() {
        _fac[0] = _ifac[0] = 1;
        for (int i = 1; i <= MAX; i++) {
            _fac[i] = _fac[i-1] * i % MOD;
            _ifac[i] = _ifac[i-1] * _inv[i] % MOD;
        }
    }

    inline long long modpow(long long n, long long r) {
        n %= MOD, r %= (MOD-1);
        if (!n) return 0;
        long long ret = 1; long long tmp = n;
        while (r != 0) {
            if (r % 2) ret *= tmp;
            tmp *= tmp; tmp %= MOD; ret %= MOD;
            r /= 2;
        }
        return ret;
    }
    inline const long long invb(int i) { return (i <= MAX) ? _inv[i] : this->modpow(i, MOD-2); }
    inline const long long inv(int i) { return _inv[i]; }
    inline const long long fac(int i) { return _fac[i]; }
    inline const long long ifac(int i) { return _ifac[i]; }

    const long long operator()(int n, int r) {
        if (n < r) return 0;
        return fac(n) * ifac(r) % MOD * ifac(n-r) % MOD;
    }
};

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    constexpr int MAX = 300000;
    constexpr int mod = 1000000007;
    Combination<MAX> C(mod);
    long long m, n; cin >> n >> m;
    long long ans = C.modpow(m, n);
    for (int i = 1; i <= m-1; i++) {
        if (i & 1) ans -= C.modpow(m-i, n) * C(m, m-i) % mod;
        else ans += C.modpow(m-i, n) * C(m, m-i) % mod;
    }
    cout << (ans % mod + mod) % mod << "\n";
    return 0;
}
0