結果

問題 No.665 Bernoulli Bernoulli
ユーザー bal4ubal4u
提出日時 2019-07-20 11:18:22
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 1,087 bytes
コンパイル時間 761 ms
コンパイル使用メモリ 31,744 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-09 10:15:50
合計ジャッジ時間 1,263 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 4 ms
6,944 KB
testcase_04 AC 4 ms
6,940 KB
testcase_05 AC 4 ms
6,940 KB
testcase_06 AC 4 ms
6,940 KB
testcase_07 AC 4 ms
6,944 KB
testcase_08 AC 4 ms
6,944 KB
testcase_09 AC 4 ms
6,940 KB
testcase_10 AC 4 ms
6,940 KB
testcase_11 AC 4 ms
6,940 KB
testcase_12 AC 4 ms
6,944 KB
testcase_13 AC 5 ms
6,944 KB
testcase_14 AC 4 ms
6,940 KB
testcase_15 AC 3 ms
6,940 KB
testcase_16 AC 3 ms
6,944 KB
testcase_17 AC 3 ms
6,944 KB
testcase_18 AC 4 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// yukicoder: No.665 Bernoulli Bernoulli
// 2019.7.20 bal4u

#include <stdio.h>

typedef long long ll;
const int M = 1000000007;

int powmod(int b, int p) {
   int r = 1;
   while (p) {
      if (p & 1) r = (ll)r * b % M;
      b = (ll)b * b % M;
	  p >>= 1;
   }
   return r;
}

int egcd(int a, int b, int *x, int *y) {
	int d;
	if (b == 0) { *x = 1; *y = 0; return a; }
	d = egcd(b, a % b, y, x);
	*y -= a / b * (*x);
	return d;
}

int inv(int a) {
    int x, y;
    egcd(a, M, &x, &y);
	return x % M;
}

int fact[10005] = {1};
int xpow[10005];

int main()
{
	int i, N, K, K1, ans;
	
	{ ll _; scanf("%lld%d", &_, &K), N = (int)(_ % M), K1 = K+1; }
	
	for (i = 1; i <= K1; i++) {
		fact[i] = (ll)fact[i-1]*i % M;
		xpow[i] = (xpow[i-1] + powmod(i, K)) % M;
	}
	if (N <= K1) { printf("%d\n", xpow[N]); return 0; }

	ans = 0; for (i = 1; i <= K1; i++) {
		ll t = (ll)xpow[i] * inv((ll)fact[i]*fact[K1-i] % M * (N-i) % M);
		if (!((i&1) ^ (K&1))) t = -t;
		ans = (t + ans) % M;
	}
	for (i = N-K1; i <= N; i++) ans = (ll)ans*i % M;
	if (ans < 0) ans += M;
	printf("%d\n", ans);
	return 0;
}
0