結果

問題 No.1102 Remnants
ユーザー Shibuyap
提出日時 2020-07-03 21:41:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 175 ms / 2,000 ms
コード長 1,597 bytes
コンパイル時間 2,017 ms
コンパイル使用メモリ 193,940 KB
最終ジャッジ日時 2025-01-11 14:21:34
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define srep(i,s,t) for (int i = s; i < t; ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
using namespace std;
typedef long long int ll;
typedef pair<int,int> P;
#define yn {puts("YES");}else{puts("NO");}
#define MAX_N 200005

// ax + by = gcd(a, b) となるような (x, y) を求める
// a と b は互いに素として ax + by = 1 となる (x, y) を求める
long long extGCD(long long a, long long b, long long &x, long long &y) {
    if (b == 0) {
        x = 1;
        y = 0;
        return a;
    }
    long long d = extGCD(b, a%b, y, x); // 再帰
    y -= a / b * x;
    return d;
}


// 負の数に対応した mod
inline long long mod(long long a, long long m) {
    return (a % m + m) % m;
}


// 逆元計算 (a と m が互いに素であることが必要)
long long modinv(long long a, long long m) {
    long long x, y;
    extGCD(a, m, x, y);
    return mod(x, m); // x % m だが、x が負かもしれないので
}

int main() {
    ll n, k;
    cin >> n >> k;
    ll a[n];
    rep(i,n)cin >> a[i];

    ll ans[n] = {};
    rep(i,n)ans[i] = 1;

    ll c = 1;
    ll MOD = 1000000007;

    rep(i,n){
        ans[i] = ans[i] * c % MOD;
        c = c * (k+i+1) % MOD * modinv(i+1,MOD) % MOD;
    }

    c = 1;
    rep(i,n){
        ans[n-1-i] = ans[n-1-i] * c % MOD;
        c = c * (k+i+1) % MOD * modinv(i+1,MOD) % MOD;
    }

    ll sum = 0;
    rep(i,n){
        ans[i] = ans[i] * a[i] % MOD;
        sum += ans[i];
        sum %= MOD;
    }

    cout << sum << endl;
    return 0;
}
 
 

0