結果

問題 No.1102 Remnants
ユーザー motmot
提出日時 2020-07-03 22:52:09
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 1,577 ms / 2,000 ms
コード長 1,415 bytes
コンパイル時間 1,356 ms
コンパイル使用メモリ 83,408 KB
実行使用メモリ 8,000 KB
最終ジャッジ日時 2024-09-17 04:33:08
合計ジャッジ時間 37,285 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<iomanip>
#include<cmath>
#include<string>
#include<vector>
#include<list>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
using namespace std;
typedef long long ll;
#define fi first
#define se second
#define mp make_pair
#define rep(i, n) for(int i=0;i<n;++i)
#define rrep(i, n) for(int i=n;i>=0;--i)
const int inf=1e9+7;
const ll mod=1e9+7;
const ll big=1e18;
const double PI=2*asin(1);

ll fact1[200005];
ll fact2[200005];
ll N, K;

ll getcomb(ll a, ll b){
  ll ue = fact2[a-K];
  ll shita1=1, shita2=1, tmpshita1, tmpshita2;
  ll h = mod-2;
  ll two;
  while(h>0){
    tmpshita1 = fact1[b];
    tmpshita2 = fact2[a-b-K];
    two = 1;
    while(2*two<=h){
      two *= 2;
      tmpshita1 *= tmpshita1;
      tmpshita1 %= mod;
      tmpshita2 *= tmpshita2;
      tmpshita2 %= mod;
    }
    h -= two;
    shita1 *= tmpshita1;
    shita2 *= tmpshita2;
    shita1 %= mod;
    shita2 %= mod;
  }
  return ue*shita1%mod*shita2%mod;
}

int main() {
  fact1[0] = 1;
  ll tmp = 1;
  cin>>N>>K;
  for(ll i=0;i<200000000;++i){
    if(i>0) tmp *= i;
    tmp %= mod;
    if(i<200005){
      fact1[i] = tmp;
    }
    if(i>=K && i<K+200005){
      fact2[i-K] = tmp;
    }
  }
  ll A[N];
  for(int i=0;i<N;++i) cin>>A[i];
  ll ans = 0;
  for(int i=0;i<N;++i){
    ans += A[i]*getcomb(K+i, i)%mod*getcomb(K+N-i-1, N-i-1)%mod;
    ans %= mod;
  }
  cout<<ans<<endl;
}
0