結果

問題 No.665 Bernoulli Bernoulli
コンテスト
ユーザー kriii
提出日時 2018-04-20 00:02:32
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 836 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 24,320 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-27 04:56:32
合計ジャッジ時間 1,002 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 15
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘long long int proc()’:
main.cpp:24:15: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   24 |         scanf ("%lld %d",&N,&K);
      |         ~~~~~~^~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <stdio.h>

const long long mod = 1000000007;
long long N,f[100100],fact[100100]; int K;

long long fpow(long long a, long long p)
{
	long long r = 1;
	while (p){
		if (p & 1) r = r * a % mod;
		a = a * a % mod;
		p /= 2;
	}
	return r;
}

long long inv(long long a)
{
	return fpow(a, mod-2);
}

long long proc()
{
	scanf ("%lld %d",&N,&K);
	N %= mod;
	for (int i=1;i<=K+1;i++) f[i] = (f[i-1] + fpow(i,K)) % mod;

	if (N <= K+1) return f[N];

	long long ans = 0;
	for (int i=1;i<=K+1;i++){
		long long r = f[i] * inv(fact[i] * fact[K+1-i] % mod * (N - i) % mod) % mod;
		if (i % 2 == K % 2) r = mod - r;
		ans = (ans + r) % mod;
	}
	for (int i=0;i<=K+1;i++) ans = ans * (N - i) % mod;

	return ans;
}

int main()
{
	fact[0] = 1;
	for (int i=1;i<100100;i++) fact[i] = fact[i-1] * i % mod;

	printf ("%lld\n",proc());

	return 0;
}
0