結果
問題 | No.823 Many Shifts Easy |
ユーザー | kyort0n |
提出日時 | 2019-04-26 21:46:53 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 689 ms / 2,000 ms |
コード長 | 1,549 bytes |
コンパイル時間 | 1,728 ms |
コンパイル使用メモリ | 168,504 KB |
実行使用メモリ | 26,952 KB |
最終ジャッジ日時 | 2024-11-25 03:04:08 |
合計ジャッジ時間 | 6,043 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 208 ms
26,952 KB |
testcase_01 | AC | 219 ms
26,872 KB |
testcase_02 | AC | 222 ms
26,812 KB |
testcase_03 | AC | 689 ms
26,792 KB |
testcase_04 | AC | 212 ms
26,880 KB |
testcase_05 | AC | 494 ms
26,776 KB |
testcase_06 | AC | 687 ms
26,736 KB |
testcase_07 | AC | 208 ms
26,864 KB |
testcase_08 | AC | 654 ms
26,892 KB |
testcase_09 | AC | 268 ms
26,756 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<ll, ll> l_l; typedef pair<int, int> i_i; #define EPS (1e-7) #define INF (1e9) #define PI (acos(-1)) const ll mod = 1000000007; ll N, K; ll inv[1000000]; ll FactorialInv[1000000]; ll Factorial[1000000]; ll beki(ll a, ll b){ if(b == 0){ return 1; } ll ans = beki(a, b / 2); ans = ans * ans % mod; if(b % 2 == 1){ ans = ans * a % mod; } return ans; } void init_combination(){ inv[1] = 1; FactorialInv[1] = 1; Factorial[1] = 1; Factorial[0] = 1; FactorialInv[0] = 1; inv[1] = 0; for(int i = 2; i < 1000000; i++){ inv[i] = beki(i, mod - 2); Factorial[i] = Factorial[i - 1] * i % mod; FactorialInv[i] = FactorialInv[i - 1] * inv[i] % mod; } } ll combination(ll a, ll b){ if(a < 0) return 0; if(b < 0) return 0; if(b > a) return 0; if((a == b) || (b == 0)){ return 1; } ll ans = Factorial[a] * FactorialInv[b] % mod; ans = ans * FactorialInv[a - b] % mod; return ans; } int main() { //cout.precision(10); cin.tie(0); ios::sync_with_stdio(false); cin >> N >> K; init_combination(); ll ans = 0; for(ll i = 1; i <= N; i++) { ans += (i * combination(N-1, K) % mod) * Factorial[K]; ans %= mod; if(i < N) ans += (i * combination(N-2, K-2) % mod) * (inv[2] * Factorial[K] % mod); ans %= mod; cerr << i << " " << ans << endl; } cout << ans << endl; return 0; }