結果

問題 No.665 Bernoulli Bernoulli
ユーザー smiken_61smiken_61
提出日時 2019-04-29 16:55:45
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,081 ms / 2,000 ms
コード長 8,749 bytes
コンパイル時間 1,931 ms
コンパイル使用メモリ 148,588 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-08-26 02:06:34
合計ジャッジ時間 19,127 ms
ジャッジサーバーID
(参考情報)
judge11 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 7 ms
4,376 KB
testcase_03 AC 1,081 ms
4,376 KB
testcase_04 AC 1,022 ms
4,376 KB
testcase_05 AC 955 ms
4,376 KB
testcase_06 AC 938 ms
4,376 KB
testcase_07 AC 908 ms
4,376 KB
testcase_08 AC 909 ms
4,376 KB
testcase_09 AC 1,033 ms
4,376 KB
testcase_10 AC 908 ms
4,376 KB
testcase_11 AC 1,060 ms
4,376 KB
testcase_12 AC 1,032 ms
4,376 KB
testcase_13 AC 1,071 ms
4,376 KB
testcase_14 AC 1,078 ms
4,376 KB
testcase_15 AC 932 ms
4,380 KB
testcase_16 AC 978 ms
4,380 KB
testcase_17 AC 946 ms
4,376 KB
testcase_18 AC 916 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

     #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;
  }
  
0