結果
問題 | No.665 Bernoulli Bernoulli |
ユーザー |
|
提出日時 | 2018-03-09 23:23:25 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 471 ms / 2,000 ms |
コード長 | 1,875 bytes |
コンパイル時間 | 663 ms |
コンパイル使用メモリ | 75,228 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-10 19:36:18 |
合計ジャッジ時間 | 8,854 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 15 |
ソースコード
#include <iostream>#include <algorithm>#include <vector>using namespace std;long power(long base, int exponent){if(exponent % 2){return power(base, exponent - 1) * base % long(1e9 + 7);}else if(exponent){long root_ans = power(base, exponent / 2);return root_ans * root_ans % long(1e9 + 7);}else{return 1;}}vector<long> factorial{1};vector<long> factorial_inverse_element{1};long combination(int n, int r){return factorial[n] * factorial_inverse_element[r] % long(1e9 + 7) * factorial_inverse_element[n - r] % long(1e9 + 7);}int main(){long n;int k;cin >> n >> k;for(long i = 1; i < 1e5; i++){factorial.push_back(factorial.back() * i % long(1e9 + 7));factorial_inverse_element.push_back(power(factorial.back(), 1e9 + 5));}long bernoulli_numbers[k + 1];bernoulli_numbers[0] = 1;for(int i = 1; i <= k; i++){bernoulli_numbers[i] = 0;for(int j = 0; j < i; j++){if(j % 2){bernoulli_numbers[i] += bernoulli_numbers[j] * combination(i + 1, j) % long(1e9 + 7);}else{bernoulli_numbers[i] += long(1e9 + 7) - bernoulli_numbers[j] * combination(i + 1, j) % long(1e9 + 7);}}bernoulli_numbers[i] %= long(1e9 + 7);if(i % 2){bernoulli_numbers[i] *= long(1e9 + 7) - power(combination(i + 1, i), 1e9 + 5);}else{bernoulli_numbers[i] *= power(combination(i + 1, i), 1e9 + 5);}bernoulli_numbers[i] %= long(1e9 + 7);}long ans = 0;for(int i = 0; i <= k; i++){ans += combination(k + 1, i) * bernoulli_numbers[i] % long(1e9 + 7) * power(n % long(1e9 + 7), k + 1 - i) % long(1e9 + 7);}cout << ans % long(1e9 + 7) * power(k + 1, 1e9 + 5) % long(1e9 + 7) << endl;}