結果

問題 No.665 Bernoulli Bernoulli
ユーザー kmjp
提出日時 2018-03-09 23:05:38
言語 C++11
(gcc 4.8.5)
結果
AC  
実行時間 23 ms
コード長 1,369 Byte
コンパイル時間 1,703 ms
使用メモリ 1,696 KB
最終ジャッジ日時 2019-11-21 17:39:41

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
sample1.txt AC 4 ms
1,544 KB
sample2.txt AC 3 ms
1,544 KB
sample3.txt AC 7 ms
1,540 KB
sample4.txt AC 21 ms
1,696 KB
test1.txt AC 22 ms
1,696 KB
test2.txt AC 21 ms
1,688 KB
test3.txt AC 20 ms
1,684 KB
test4.txt AC 20 ms
1,684 KB
test5.txt AC 21 ms
1,688 KB
test6.txt AC 23 ms
1,692 KB
test7.txt AC 20 ms
1,684 KB
test8.txt AC 22 ms
1,692 KB
test9.txt AC 21 ms
1,696 KB
test10.txt AC 22 ms
1,692 KB
test11.txt AC 22 ms
1,692 KB
test12.txt AC 21 ms
1,684 KB
test13.txt AC 21 ms
1,692 KB
test14.txt AC 21 ms
1,684 KB
test15.txt AC 21 ms
1,684 KB
テストケース一括ダウンロード

ソースコード

diff #
#include <bits/stdc++.h>
using namespace std;
typedef signed long long ll;

#undef _P
#define _P(...) (void)printf(__VA_ARGS__)
#define FOR(x,to) for(x=0;x<(to);x++)
#define FORR(x,arr) for(auto& x:arr)
#define ITR(x,c) for(__typeof(c.begin()) x=c.begin();x!=c.end();x++)
#define ALL(a) (a.begin()),(a.end())
#define ZERO(a) memset(a,0,sizeof(a))
#define MINUS(a) memset(a,0xff,sizeof(a))
//-------------------------------------------------------

ll N,K;
ll mo=1000000007;
ll P[101010];
ll fact[101010];

ll modpow(ll a, ll n = mo-2) {
	ll r=1;a%=mo;
	while(n) r=r*((n%2)?a:1)%mo,a=a*a%mo,n>>=1;
	return r;
}

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N>>K;
	
	ll ret=0;
	if(N<=K+2) {
		for(i=1;i<=N;i++) ret+=modpow(i,K);
		cout<<ret%mo<<endl;
		return;
	}
	
	for(i=1;i<=K+1;i++) (P[i]=modpow(i,K)+P[i-1])%=mo;
	fact[0]=1;
	for(i=1;i<=K+1;i++) fact[i]=fact[i-1]*i%mo;
	
	ll A=1,B=1;
	for(i=0;i<=K+1;i++) A=A*(N%mo-i+mo)%mo;
	
	for(i=0;i<=K+1;i++) {
		ll v=P[i]*A%mo*modpow(N%mo-i+mo)%mo;
		ll w=fact[K+1-i]*fact[i]%mo;
		v=v*modpow(w)%mo;
		
		if(i%2 != (K+1)%2) ret=(ret+mo-v)%mo;
		else ret=(ret+v)%mo;
	}
	
	cout<<ret<<endl;
	
}


int main(int argc,char** argv){
	string s;int i;
	if(argc==1) ios::sync_with_stdio(false), cin.tie(0);
	FOR(i,argc-1) s+=argv[i+1],s+='\n'; FOR(i,s.size()) ungetc(s[s.size()-1-i],stdin);
	cout.tie(0); solve(); return 0;
}
0