結果

問題 No.1102 Remnants
ユーザー DriceDrice
提出日時 2020-07-03 23:05:37
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 60 ms / 2,000 ms
コード長 1,438 bytes
コンパイル時間 198 ms
コンパイル使用メモリ 33,792 KB
実行使用メモリ 9,512 KB
最終ジャッジ日時 2024-09-17 04:45:30
合計ジャッジ時間 1,711 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<cstdio>
const long long mod = 1e9+7;
long long f[200005];
long long cc[200005];
long long a[200005];
long long sum[200005];

long long power(long long a,long long b){
    long long res = 1;
    while(b){
        if(b&1) res = res*a%mod;
        a = a*a%mod;
        b /= 2;
    }
    return res;
}

long long inv(long long u){ return power(u,mod-2); }

long long getComb(int n,int m){
    if(m>n || m<0) return 0;
    else{
        long long res = f[n]*inv(f[n-m])%mod*inv(f[m])%mod;
        return res;
    }
}

void preWork(long long k,long long n){
    f[0] = 1;
    for(int i = 1; i <= n; i++){
        f[i] = f[i-1]*(long long)i%mod;
    }
    long long fac = 1;
    for(long long i = k-1; i >= k; i--) fac = fac*i%mod;
    cc[0] = fac; 
    for(long long i = 1; i <= n; i++){
        fac = fac*(long long)(k+i-1)%mod;
        cc[i] = fac*inv(f[i])%mod;
    }
    sum[0] = cc[0];
    for(int i = 1; i <= n; i++){
        sum[i] = (sum[i-1]+cc[i])%mod;
    }
}
/*
5 100000000
cc[0] = 1
cc[1] = 100000000
cc[2] = 15000000
cc[3] = 6500000
cc[4] = 3737500
cc[5] = 2466750
*/

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    preWork(k,n);
    long long ans = 0;
    for(int i = 1; i <= n; i++){ 
        long long u;
        scanf("%lld",&u);
        long long cnt = sum[i-1]*sum[n-i]%mod;
        if(k==0) cnt = 1;
        ans += u*cnt%mod;
        if(ans>=mod) ans -= mod;
    }
    printf("%lld\n",ans);
    return 0;
}
0