結果

問題 No.1489 Repeat Cumulative Sum
コンテスト
ユーザー ぷら
提出日時 2021-04-23 22:57:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 108 ms / 2,000 ms
コード長 955 bytes
コンパイル時間 1,953 ms
コンパイル使用メモリ 197,008 KB
最終ジャッジ日時 2025-01-21 00:20:45
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int mod = 1000000007;

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

int main() {
    int N;
    long long M;
    cin >> N >> M;
    vector<int>A(N-1);
    for(int i = 0; i < N-1; i++) {
        cin >> A[i];
    }
    vector<long long>tmp(N);
    long long tmp2 = 1,tmp3 = 1;
    for(int i = 0; i < N; i++) {
        tmp2 *= i+1;
        tmp3 *= (M+i-1)%mod;
        tmp2 %= mod;
        tmp3 %= mod;
        tmp[i] = tmp3*modpow(tmp2,mod-2)%mod;
    }
    vector<long long>sum(N+1);
    for(int i = 0; i < N; i++) {
        sum[i+1] = sum[i]+tmp[i];
        sum[i+1] %= mod;
    }
    long long ans = 0;
    for(int i = 0; i < N-1; i++) {
        ans += A[i];
        ans += A[i]*sum[N-i-1];
        ans %= mod;
    }
    cout << ans << endl;
}
0