結果

問題 No.1102 Remnants
ユーザー SSlime
提出日時 2020-07-04 00:24:48
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 685 ms / 2,000 ms
コード長 1,041 bytes
コンパイル時間 801 ms
コンパイル使用メモリ 73,040 KB
最終ジャッジ日時 2025-01-11 15:14:51
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

const long long mod = 1000000007;
using ll = long long;

long long modpow(long long n, long long k, long long m)
{
    long long res = 1;
    while (k > 0)
    {
        if (k & 1)
            res = res * n % m;
        k = k >> 1;
        n = n * n % m;
    }
    return res;
}

int main()
{
    ll n, k;
    std::cin >> n >> k;
    std::vector<ll> A(n);
    for (int i = 0; i < n; i++)
        std::cin >> A.at(i);
    ll ans = 0, c = 1, f = 1;
    for (ll i = 1; i < n + k; i++)
    {
        f = (f * i) % mod;
        if (i == k)
            c = (c * modpow(f, mod - 2, mod)) % mod;
        if (i == n - 1)
            c = (c * modpow(f, mod - 2, mod)) % mod;
        if (i == n - 1 + k)
            c = c * f % mod;
    }
    for (ll i = 0; i < n; i++)
    {
        ll a = A.at(i);
        ans = (ans + a * c % mod)%mod;
        c = c * (k + i + 1) % mod * (n - i - 1) % mod;
        c = c * modpow((i + 1) * (n - 1 - i + k) % mod, mod - 2, mod) % mod;
    }
    std::cout << ans << std::endl;
}
0