結果

問題 No.444 旨味の相乗効果
コンテスト
ユーザー はまやんはまやん
提出日時 2017-04-20 14:34:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,977 bytes
コンパイル時間 1,937 ms
コンパイル使用メモリ 176,744 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-19 20:44:15
合計ジャッジ時間 4,655 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4 WA * 1
other AC * 14 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;i++)



int mod = 1000000007;
int add(int x, int y) { return (x += y) >= mod ? x - mod : x; }
template<class... T> int add(int x, T... y) { return add(x, add(y...)); }
int mul(int x, int y) { return 1LL * x * y % mod; }
template<class... T> int mul(int x, T... y) { return mul(x, mul(y...)); }
int sub(int x, int y) { return add(x, mod - y); }
int modpow(int a, long long b) { int ret = 1; while (b > 0) { if (b & 1) ret = 1LL * ret * a % mod; a = 1LL * a * a % mod; b >>= 1; } return ret; }
int modinv(int a) { return modpow(a, mod - 2); }
//-----------------------------------------------------------------------------------
typedef vector<int> vi;
typedef vector<vi> vvi;
vi mul_mat_vec(vvi a, vi b)
{
    int n = b.size();
    vi ret(n, 0);
    rep(i, 0, n)
    {
        ret[i] = 0;
        rep(j, 0, n) ret[i] = add(ret[i], mul(a[i][j],b[j]));
    }
    return ret;
}

vvi mul_mat_mat(vvi a, vvi b)
{
    int n = a.size();
    vvi ret(n, vi(n, 0));
    rep(i, 0, n) rep(j, 0, n)
    {
        ret[i][j] = 0;
        rep(k, 0, n) ret[i][j] = add(ret[i][j], mul(a[i][k], b[k][j]));
    }
    return ret;
}
vvi fastpow(vvi x, int n)
{
    vvi ret(x.size(), vi(x.size(), 0));
    rep(i, 0, x.size()) ret[i][i] = 1;

    while (0 < n)
    {
        if ((n % 2) == 0)
        {
            x = mul_mat_mat(x, x);
            n >>= 1;
        }
        else
        {
            ret = mul_mat_mat(ret, x);
            --n;
        }
    }
    return ret;
}
//-----------------------------------------------------------------------------------
int N, C, A[80];
int main() {
    cin >> N >> C;
    rep(i, 0, N) cin >> A[i];


    vvi ma(N, vi(N, 0));
    rep(i, 0, N) rep(j, i, N) ma[j][i] = A[i];
    ma = fastpow(ma, C);

    vi ve(N);
    rep(i, 0, N) ve[i] = 1;

    ve = mul_mat_vec(ma, ve);

    int ans = ve[N - 1];
    rep(i, 0, N) ans = sub(ans, modpow(A[i], C));
    cout << ans << endl;
}
0