結果
| 問題 | No.665 Bernoulli Bernoulli |
| ユーザー |
 smiken_61
|
| 提出日時 | 2019-04-29 16:55:45 |
| 言語 | C++11 (gcc 11.4.0) |
| 結果 |
AC
|
| 実行時間 | 935 ms / 2,000 ms |
| コード長 | 8,749 bytes |
| コンパイル時間 | 1,769 ms |
| コンパイル使用メモリ | 163,632 KB |
| 実行使用メモリ | 6,944 KB |
| 最終ジャッジ日時 | 2024-06-06 21:01:50 |
| 合計ジャッジ時間 | 16,189 ms |
|
ジャッジサーバーID （参考情報） |
judge5 / judge2 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 2 ms
6,812 KB |
| testcase_01 | AC | 1 ms
6,944 KB |
| testcase_02 | AC | 7 ms
6,940 KB |
| testcase_03 | AC | 927 ms
6,944 KB |
| testcase_04 | AC | 881 ms
6,940 KB |
| testcase_05 | AC | 825 ms
6,940 KB |
| testcase_06 | AC | 806 ms
6,940 KB |
| testcase_07 | AC | 781 ms
6,940 KB |
| testcase_08 | AC | 809 ms
6,940 KB |
| testcase_09 | AC | 894 ms
6,944 KB |
| testcase_10 | AC | 782 ms
6,944 KB |
| testcase_11 | AC | 911 ms
6,944 KB |
| testcase_12 | AC | 881 ms
6,940 KB |
| testcase_13 | AC | 935 ms
6,940 KB |
| testcase_14 | AC | 923 ms
6,944 KB |
| testcase_15 | AC | 788 ms
6,944 KB |
| testcase_16 | AC | 837 ms
6,940 KB |
| testcase_17 | AC | 801 ms
6,944 KB |
| testcase_18 | AC | 780 ms
6,940 KB |
ソースコード
#include <bits/stdc++.h>
// #include <boost/multiprecision/cpp_int.hpp>
#define int long long
#define inf 1000000007
#define pa pair<int,int>
#define ll long long
#define pal pair<double,double>
#define ppap pair<pa,int>
#define PI 3.14159265358979323846
#define paa pair<int,char>
#define mp make_pair
#define pb push_back
#define EPS (1e-10)
int dx[8]={0,1,0,-1,1,1,-1,-1};
int dy[8]={1,0,-1,0,-1,1,1,-1};
using namespace std;
class pa3{
public:
int x;
int y,z;
pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
bool operator < (const pa3 &p) const{
if(x!=p.x) return x<p.x;
if(y!=p.y) return y<p.y;
return z<p.z;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator > (const pa3 &p) const{
if(x!=p.x) return x>p.x;
if(y!=p.y) return y>p.y;
return z>p.z;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa3 &p) const{
return x==p.x && y==p.y && z==p.z;
}
bool operator != (const pa3 &p) const{
return !( x==p.x && y==p.y && z==p.z);
}
};
class pa4{
public:
int x;
int y,z,w;
pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
bool operator < (const pa4 &p) const{
if(x!=p.x) return x<p.x;
if(y!=p.y) return y<p.y;
if(z!=p.z)return z<p.z;
return w<p.w;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator > (const pa4 &p) const{
if(x!=p.x) return x>p.x;
if(y!=p.y) return y>p.y;
if(z!=p.z)return z>p.z;
return w>p.w;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa4 &p) const{
return x==p.x && y==p.y && z==p.z &&w==p.w;
}
};
class pa2{
public:
int x,y;
pa2(int x=0,int y=0):x(x),y(y) {}
pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
bool operator < (const pa2 &p) const{
return y != p.y ? y<p.y: x<p.x;
}
bool operator > (const pa2 &p) const{
return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa2 &p) const{
return abs(x-p.x)==0 && abs(y-p.y)==0;
}
bool operator != (const pa2 &p) const{
return !(abs(x-p.x)==0 && abs(y-p.y)==0);
}
};
string itos( int i ) {
ostringstream s ;
s << i ;
return s.str() ;
}
int gcd(int v,int b){
if(v>b) return gcd(b,v);
if(v==b) return b;
if(b%v==0) return v;
return gcd(v,b%v);
}
int mod;
int extgcd(int a, int b, int &x, int &y) {
if (b == 0) {
x = 1;
y = 0;
return a;
}
int d = extgcd(b, a%b, y, x);
y -= a/b * x;
return d;
}
pa operator+(const pa & l,const pa & r) {
return {l.first+r.first,l.second+r.second};
}
pa operator-(const pa & l,const pa & r) {
return {l.first-r.first,l.second-r.second};
}
int pr[1200010];
int inv[1200010];
int beki(int wa,int rr,int warukazu){
if(rr==0) return 1%warukazu;
if(rr==1) return wa%warukazu;
wa%=warukazu;
if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
ll zx=beki(wa,rr/2,warukazu);
return (zx*zx)%warukazu;
}
int comb(int nn,int rr){
int r=pr[nn]*inv[rr];
r%=mod;
r*=inv[nn-rr];
r%=mod;
return r;
}
void gya(int ert){
pr[0]=1;
for(int i=1;i<=ert;i++){
pr[i]=(pr[i-1]*i)%mod;
}
inv[ert]=beki(pr[ert],mod-2,mod);
for(int i=ert-1;i>=0;i--){
inv[i]=inv[i+1]*(i+1)%mod;
}
}
// cin.tie(0);
// ios::sync_with_stdio(false);
//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;
//sort(ve.begin(),ve.end(),greater<int>());
// mt19937(clock_per_sec);
// mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()) ;
//----------------kokomade tenpure------------
// 0 0-0
// 1 1-3
// 2 3-5
// 3 6-6
signed main(){
cin.tie(0);
ios::sync_with_stdio(false);
int n,k;
cin>>n>>k;
if(n<=k+1){
int ans=0;
for(int i=1;i<=n;i++){
ans+=beki(i,k,inf);
if(ans>=inf)ans-=inf;
}
cout<<ans<<endl;
return 0;
}
else{
int ans=0;
int ima=0;
for(int i=0;i<=k+1;i++){
int bunbo=1;
for(int j=0;j<=k+1;j++)if(j!=i){
int kake=i-j;
if(kake<0)kake+=inf;
bunbo*=kake;
bunbo%=inf;
}
int bunsi=1;
for(int j=0;j<=k+1;j++)if(j!=i){
int kake=n-j;
kake%=inf;
bunsi*=kake;
bunsi%=inf;
}
ima+=beki(i,k,inf);
// cout<<ima<<" "<<bunbo<<" "<<bunsi<<endl;
if(ima>=inf)ima-=inf;
ans+=bunsi*ima%inf*beki(bunbo,inf-2,inf)%inf;
if(ans>=inf) ans-=inf;
}
cout<<ans<<endl;
}
return 0;
}