結果
問題 | No.391 CODING WAR |
ユーザー | hamray |
提出日時 | 2020-04-20 21:23:18 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 25 ms / 2,000 ms |
コード長 | 6,380 bytes |
コンパイル時間 | 1,569 ms |
コンパイル使用メモリ | 168,776 KB |
実行使用メモリ | 9,916 KB |
最終ジャッジ日時 | 2024-10-06 09:14:36 |
合計ジャッジ時間 | 2,430 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 7 ms
8,284 KB |
testcase_01 | AC | 8 ms
7,900 KB |
testcase_02 | AC | 8 ms
8,416 KB |
testcase_03 | AC | 8 ms
9,040 KB |
testcase_04 | AC | 8 ms
9,916 KB |
testcase_05 | AC | 7 ms
9,116 KB |
testcase_06 | AC | 8 ms
8,156 KB |
testcase_07 | AC | 10 ms
8,764 KB |
testcase_08 | AC | 8 ms
8,892 KB |
testcase_09 | AC | 25 ms
8,988 KB |
testcase_10 | AC | 24 ms
8,724 KB |
testcase_11 | AC | 15 ms
8,996 KB |
testcase_12 | AC | 7 ms
8,284 KB |
testcase_13 | AC | 22 ms
9,264 KB |
testcase_14 | AC | 20 ms
8,608 KB |
testcase_15 | AC | 22 ms
8,592 KB |
testcase_16 | AC | 17 ms
8,680 KB |
testcase_17 | AC | 18 ms
8,152 KB |
testcase_18 | AC | 15 ms
9,688 KB |
testcase_19 | AC | 16 ms
8,416 KB |
ソースコード
#include <bits/stdc++.h> //typedef //-------------------------#include <bits/stdc++.h> const double pi = 3.141592653589793238462643383279; using namespace std; //conversion //------------------------------------------ inline int toInt(string s) { int v; istringstream sin(s); sin >> v; return v; } template <class T> inline string toString(T x) { ostringstream sout; sout << x; return sout.str(); } inline int readInt() { int x; scanf("%d", &x); return x; } //typedef //------------------------------------------ typedef vector<int> VI; typedef vector<VI> VVI; typedef vector<string> VS; typedef pair<int, int> PII; typedef pair<long long, long long> PLL; typedef pair<int, PII> TIII; typedef long long LL; typedef unsigned long long ULL; typedef vector<LL> VLL; typedef vector<VLL> VVLL; //container util //------------------------------------------ #define ALL(a) (a).begin(), (a).end() #define RALL(a) (a).rbegin(), (a).rend() #define PB push_back #define MP make_pair #define SZ(a) int((a).size()) #define SQ(a) ((a) * (a)) #define EACH(i, c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); ++i) #define EXIST(s, e) ((s).find(e) != (s).end()) #define SORT(c) sort((c).begin(), (c).end()) //repetition //------------------------------------------ #define FOR(i, s, n) for (int i = s; i < (int)n; ++i) #define REP(i, n) FOR(i, 0, n) #define MOD 1000000007 #define rep(i, a, b) for (int i = a; i < (b); ++i) #define trav(a, x) for (auto &a : x) #define all(x) x.begin(), x.end() #define sz(x) (int)(x).size() typedef long long ll; typedef pair<int, int> P; typedef pair<int, int> pii; typedef vector<int> vi; const double EPS = 1E-8; #define chmin(x, y) x = min(x, y) #define chmax(x, y) x = max(x, y) class UnionFind { public: vector<int> par; vector<int> siz; UnionFind(int sz_) : par(sz_), siz(sz_, 1) { for (ll i = 0; i < sz_; ++i) par[i] = i; } void init(int sz_) { par.resize(sz_); siz.assign(sz_, 1LL); for (ll i = 0; i < sz_; ++i) par[i] = i; } int root(int x) { if (par[x] == x) { return x; } else { return par[x] = root(par[x]); } } bool merge(int x, int y) { x = root(x); y = root(y); if (x == y) return false; if (siz[x] < siz[y]) swap(x, y); siz[x] += siz[y]; par[y] = x; return true; } bool issame(int x, int y) { return root(x) == root(y); } int size(int x) { return siz[root(x)]; } }; ll modPow(ll x, ll n, ll mod = MOD) { if (n == 0) return 1; ll res = 1; while (n) { if (n & 1) res = (res * x) % mod; res %= mod; x = x * x % mod; n >>= 1; } return res; } #define SIEVE_SIZE 5000000 + 10 bool sieve[SIEVE_SIZE]; void makeSieve() { for (int i = 0; i < SIEVE_SIZE; ++i) sieve[i] = true; sieve[0] = sieve[1] = false; for (int i = 2; i * i < SIEVE_SIZE; ++i) if (sieve[i]) for (int j = 2; i * j < SIEVE_SIZE; ++j) sieve[i * j] = false; } bool isprime(ll n) { if (n == 0 || n == 1) return false; for (ll i = 2; i * i <= n; ++i) if (n % i == 0) return false; return true; } const int MAX = 200010; long long fac[MAX], finv[MAX], inv[MAX]; const ll mod2 = 998244353; const ll mod3 = 1777777777; // テーブルを作る前処理 void COMinit() { fac[0] = fac[1] = 1; finv[0] = finv[1] = 1; inv[1] = 1; for (ll i = 2; i < MAX; 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; //cout << fac[i] << " " << inv[i] << " " << finv[i] << endl; } } // 二項係数計算 long long COM(int n, int k) { if (n < k) return 0; if (n < 0 || k < 0) return 0; return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD; } long long extGCD(long long a, long long b, long long &x, long long &y) { if (b == 0) { x = 1; y = 0; return a; } long long d = extGCD(b, a % b, y, x); y -= a / b * x; return d; } // 負の数にも対応した mod (a = -11 とかでも OK) inline long long mod(long long a, long long m) { return (a % m + m) % m; } // 逆元計算 (ここでは a と m が互いに素であることが必要) long long modinv(long long a, long long m) { long long x, y; extGCD(a, m, x, y); return mod(x, m); // 気持ち的には x % m だが、x が負かもしれないので } ll GCD(ll a, ll b) { if (b == 0) return a; return GCD(b, a % b); } typedef vector<ll> vec; typedef vector<vec> mat; mat mul(mat &A, mat &B) { mat C(A.size(), vec((int)B[0].size())); for (int i = 0; i < A.size(); ++i) { for (int k = 0; k < B.size(); ++k) { for (int j = 0; j < B[0].size(); ++j) { C[i][j] = (C[i][j] + A[i][k] % MOD * B[k][j] % MOD) % MOD; } } } return C; } mat matPow(mat A, ll n) { mat B(A.size(), vec((int)A.size())); for (int i = 0; i < A.size(); ++i) { B[i][i] = 1; } while (n > 0) { if (n & 1) B = mul(B, A); A = mul(A, A); n >>= 1; } return B; } template <class TYPE> struct Fenwick { int N; vector<ll> bit; Fenwick() : N(0) {} Fenwick(int n) : N(n), bit(n + 1, 0) {} void add(int i, ll x) { //0-indexedを避けるために加算 while (i <= N) { bit[i] += x; i += (i & -i); } } ll sum(int i) { ll res = 0; while (i > 0) { res += bit[i]; i -= (i & -i); } return res; } ll range(int i, int j) { return sum(j) - sum(i); } ll get(int i) { return sum(i) - sum(i - 1); } ll getKth(ll k) { ++k; int res = 0; int n = 1; while (n < bit.size()) n *= 2; for (int i = n / 2; i > 0; i /= 2) { if (res + i < bit.size() && bit[res + i] < k) { k = k - bit[res + i]; res = res + i; } } return res + 1; } // bitは1-indexedで実装しているのでバグった時は初めに確認すること }; int main() { cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(4); ll N, K; cin >> N >> K; ll ans = 0; COMinit(); for (int i = 0; i <= K; i++) { ll c = 1; if ((K - i) % 2 == 1) { c *= -1; } ans += c * COM(K, i) * modPow(i, N, MOD); ans = (ans + MOD) % MOD; } cout << ans << endl; return 0; }