結果

問題 No.391 CODING WAR
ユーザー ooaiu
提出日時 2025-09-17 06:43:22
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 22 ms / 2,000 ms
コード長 4,322 bytes
コンパイル時間 3,082 ms
コンパイル使用メモリ 281,008 KB
実行使用メモリ 7,716 KB
最終ジャッジ日時 2025-09-17 06:43:27
合計ジャッジ時間 4,533 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
template <class F> class y_combinator {
    F f;

   public:
    y_combinator(F&& f) : f(std::forward<F>(f)) {}
    template <class... Args> auto operator()(Args&&... args) const { return f(*this, std::forward<Args>(args)...); }
};
using ll = long long;
using ld = long double;
using u64 = uint64_t;
using vi = vector<int>;
using vl = vector<ll>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
template <class T, class U = std::less<T>> using prique = std::priority_queue<T, std::vector<T>, U>;
inline constexpr int popcnt(u64 x) noexcept { return __builtin_popcountll(x); }
inline ll floor_div(ll a, ll b) noexcept { return a / b - (a % b && (a ^ b) < 0); }
inline ll ceil_div(ll a, ll b) noexcept { return floor_div(a + b - 1, b); }
template <class T> bool chmin(T& x, const T& y) noexcept { return (x > y ? x = y, true : false); }
template <class T> bool chmax(T& x, const T& y) noexcept { return (x < y ? x = y, true : false); }
template <class T> void dedup(std::vector<T>& v) { std::sort(std::begin(v), std::end(v)), v.erase(std::unique(std::begin(v), std::end(v)), std::end(v)); }
template <class F> ll bisect(ll ok, ll ng, const F& f) { while (abs(ok - ng) > 1) { ll mid = (ok + ng) / 2; (f(mid) ? ok : ng) = mid; } return ok; }
template <class F> ld bisect_real(ld ok, ld ng, const F& f, int iter = 80) { while (iter--) { ld mid = (ok + ng) / 2.; (f(mid) ? ok : ng) = mid; } return ok; }
inline void SCAN() {}
template <class H, class... T> inline void SCAN(H& h, T&... t) { std::cin >> h, SCAN(t...); }
#define INT(...) int __VA_ARGS__; SCAN(__VA_ARGS__)
#define LL(...) long long __VA_ARGS__; SCAN(__VA_ARGS__)
#define LD(...) long double __VA_ARGS__; SCAN(__VA_ARGS__)
#define STR(...) std::string __VA_ARGS__; SCAN(__VA_ARGS__)
#define CHAR(...) char __VA_ARGS__; SCAN(__VA_ARGS__)
#define VEC(type, name, size) std::vector<type> name(size); for (int i = 0; i < (int)size; i++) SCAN(name[i]);
#define VV(type, name, h, w) std::vector<std::vector<type>> name(h, std::vector<type>(w)); for (int i = 0; i < (int)h; i++) for (int j = 0; j < (int)w; j++) SCAN(name[i][j]);
#define overload4(a, b, c, d, e, ...) e
#define rep1(a) for (long long _i = 0; _i < (a); _i++)
#define rep2(i, a) for (long long i = 0; i < (a); i++)
#define rep3(i, a, b) for (long long i = (a); i < (b); i++)
#define rep4(i, a, b, c) for (long long i = (a); i < (b); i += (c))
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)
#define rrep(i, a, b, c) for (long long i = (a); i > (b); i += (c))
#define all(x) std::begin(x), std::end(x)
#define rall(x) std::rbegin(x), std::rend(x)
#define len(x) (long long)(size(x))
#define pb push_back
#ifndef LOCAL
#define debug(...)
#endif

template <class Mint>
struct combination {
   private:
    std::vector<Mint> _fact, _ifac;
    void expand(size_t N) {
        const size_t a = _fact.size();
        if(a > N) return;
        _fact.resize(N + 1);
        for(size_t i = a; i <= N; i++) _fact[i] = _fact[i - 1] * i;
        _ifac.resize(N + 1);
        _ifac[N] = Mint{1} / _fact[N];
        for(size_t i = N; i > a; i--) _ifac[i - 1] = _ifac[i] * i;
    }

   public:
    combination(size_t N = 0) : _fact(1, 1), _ifac(1, 1) {
        expand(N);
    }
    Mint fact(size_t x) {
        expand(x);
        return _fact[x];
    }
    Mint invfact(size_t x) {
        expand(x);
        return _ifac[x];
    }
    Mint perm(size_t N, size_t K) {
        if(N < K) return Mint{0};
        expand(N);
        return _fact[N] * _ifac[N - K];
    }
    Mint comb(size_t N, size_t K) {
        if(N < K) return Mint{0};
        expand(N);
        return _fact[N] * _ifac[K] * _ifac[N - K];
    }
    Mint homo(size_t N, size_t K) {
        if(N == 0) return K == 0 ? Mint{1} : Mint{0};
        return comb(N - 1 + K, K);
    }
};

#include <atcoder/modint>
using mint = atcoder::modint1000000007;
combination<mint> C;
void run_case() {
    LL(N, M);
    mint ans = 0;
    rep(i, M + 1) {
        int sgn = (M - i) & 1 ? -1 : 1;
        mint t = C.comb(M, i) * mint(i).pow(N);
        ans += t * sgn;
    }
    cout << ans.val() << "\n";
}
int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::fixed(std::cout).precision(16);
    int T = 1;
    while (T--) run_case();
    return 0;
}
0