結果

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

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
sample1.txt AC 3 ms
6,868 KB
sample2.txt AC 3 ms
6,872 KB
sample3.txt AC 7 ms
6,868 KB
sample4.txt AC 22 ms
6,868 KB
test1.txt AC 21 ms
6,872 KB
test2.txt AC 20 ms
6,868 KB
test3.txt AC 21 ms
6,868 KB
test4.txt AC 20 ms
8,916 KB
test5.txt AC 21 ms
6,872 KB
test6.txt AC 22 ms
6,868 KB
test7.txt AC 20 ms
8,912 KB
test8.txt AC 22 ms
6,872 KB
test9.txt AC 21 ms
6,868 KB
test10.txt AC 22 ms
6,868 KB
test11.txt AC 22 ms
6,872 KB
test12.txt AC 21 ms
6,868 KB
test13.txt AC 21 ms
6,872 KB
test14.txt AC 21 ms
6,868 KB
test15.txt AC 20 ms
6,868 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