結果
問題 | No.391 CODING WAR |
ユーザー | outline |
提出日時 | 2021-02-03 19:16:17 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 24 ms / 2,000 ms |
コード長 | 4,166 bytes |
コンパイル時間 | 1,395 ms |
コンパイル使用メモリ | 139,804 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-06-30 03:31:54 |
合計ジャッジ時間 | 2,343 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 1 ms
5,376 KB |
testcase_03 | AC | 2 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 2 ms
5,376 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 2 ms
5,376 KB |
testcase_08 | AC | 2 ms
5,376 KB |
testcase_09 | AC | 24 ms
5,376 KB |
testcase_10 | AC | 22 ms
5,376 KB |
testcase_11 | AC | 2 ms
5,376 KB |
testcase_12 | AC | 2 ms
5,376 KB |
testcase_13 | AC | 19 ms
5,376 KB |
testcase_14 | AC | 18 ms
5,376 KB |
testcase_15 | AC | 20 ms
5,376 KB |
testcase_16 | AC | 13 ms
5,376 KB |
testcase_17 | AC | 15 ms
5,376 KB |
testcase_18 | AC | 10 ms
5,376 KB |
testcase_19 | AC | 11 ms
5,376 KB |
ソースコード
#include <iostream> #include <vector> #include <algorithm> #include <cmath> #include <queue> #include <string> #include <map> #include <set> #include <stack> #include <tuple> #include <deque> #include <array> #include <numeric> #include <bitset> #include <iomanip> #include <cassert> #include <chrono> #include <random> #include <limits> #include <iterator> #include <functional> #include <sstream> #include <fstream> #include <complex> #include <cstring> #include <unordered_map> #include <unordered_set> using namespace std; using ll = long long; constexpr int INF = 1001001001; constexpr int mod = 1000000007; // constexpr int mod = 998244353; template<class T> inline bool chmax(T& x, T y){ if(x < y){ x = y; return true; } return false; } template<class T> inline bool chmin(T& x, T y){ if(x > y){ x = y; return true; } return false; } struct mint { int x; mint() : x(0) {} mint(int64_t y) : x(y >= 0 ? y % mod : (mod - (-y) % mod) % mod) {} mint& operator+=(const mint& p){ if((x += p.x) >= mod) x -= mod; return *this; } mint& operator-=(const mint& p){ if((x -= p.x) < 0) x += mod; return *this; } mint& operator*=(const mint& p){ x = (int)(1LL * x * p.x % mod); return *this; } mint& operator/=(const mint& p){ *this *= p.inverse(); return *this; } mint operator-() const { return mint(-x); } mint operator+(const mint& p) const { return mint(*this) += p; } mint operator-(const mint& p) const { return mint(*this) -= p; } mint operator*(const mint& p) const { return mint(*this) *= p; } mint operator/(const mint& p) const { return mint(*this) /= p; } bool operator==(const mint& p) const { return x == p.x; } bool operator!=(const mint& p) const { return x != p.x; } mint pow(int64_t n) const { mint res = 1, mul = x; while(n > 0){ if(n & 1) res *= mul; mul *= mul; n >>= 1; } return res; } mint inverse() const { return pow(mod - 2); } friend ostream& operator<<(ostream& os, const mint& p){ return os << p.x; } friend istream& operator>>(istream& is, mint& p){ int64_t val; is >> val; p = mint(val); return is; } }; template<typename T> struct Combination{ int sz; vector<T> fact_; vector<T> ifact_; vector<T> inv_; Combination(int n = 1e+6) : sz(n) { fact_.resize(sz + 1); ifact_.resize(sz + 1); inv_.resize(sz + 1); fact_[0] = ifact_[sz] = inv_[0] = 1; for(int i = 1; i <= sz; ++i) fact_[i] = fact_[i - 1] * i; ifact_[sz] /= fact_[sz]; for(int i = sz; i > 0; --i) ifact_[i - 1] = ifact_[i] * i; for(int i = 1; i <= sz; ++i) inv_[i] = ifact_[i] * fact_[i - 1]; } inline T fact(int k) const { if(k < 0 || k > sz) return 0; return fact_[k]; } inline T ifact(int k) const { if(k < 0 || k > sz) return 0; return ifact_[k]; } inline T inv(int k) const { if(k < 0 || k > sz) return 0; return inv_[k]; } // nPk T get_permutation(int n, int k){ if(n < 0 || k < 0 || n < k) return 0; return fact(n) * ifact(n - k); } // nCk T get_combination(int n, int k){ if(n < 0 || k < 0 || n < k) return 0; return fact(n) * ifact(k) * ifact(n - k); } // nHk T get_repetition(int n, int k){ if(n == 0 && k == 0) return 1; else return get_combination(n + k - 1, k); } }; int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); ll N, M; cin >> N >> M; if(N < M){ cout << 0 << endl; return 0; } Combination<mint> comb(M); mint ans = 0; for(int i = 0; i < M; ++i){ if(i % 2 == 0){ ans += comb.get_combination(M, i) * mint(M - i).pow(N); } else{ ans -= comb.get_combination(M, i) * mint(M - i).pow(N); } } cout << ans << endl; return 0; }