結果

問題 No.797 Noelちゃんとピラミッド
ユーザー shibh308shibh308
提出日時 2019-03-15 21:48:40
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 4,692 bytes
コンパイル時間 2,317 ms
コンパイル使用メモリ 204,336 KB
実行使用メモリ 5,576 KB
最終ジャッジ日時 2023-09-14 13:24:09
合計ジャッジ時間 5,980 ms
ジャッジサーバーID
(参考情報)
judge15 / judge11
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 36 ms
5,376 KB
testcase_04 AC 36 ms
5,392 KB
testcase_05 AC 35 ms
5,340 KB
testcase_06 AC 36 ms
5,544 KB
testcase_07 AC 36 ms
5,420 KB
testcase_08 AC 36 ms
5,572 KB
testcase_09 AC 35 ms
5,428 KB
testcase_10 AC 35 ms
5,420 KB
testcase_11 AC 36 ms
5,352 KB
testcase_12 AC 35 ms
5,376 KB
testcase_13 AC 36 ms
5,460 KB
testcase_14 AC 34 ms
5,376 KB
testcase_15 AC 35 ms
5,432 KB
testcase_16 AC 36 ms
5,464 KB
testcase_17 AC 36 ms
5,412 KB
testcase_18 AC 36 ms
5,576 KB
testcase_19 AC 36 ms
5,420 KB
testcase_20 AC 36 ms
5,396 KB
testcase_21 AC 36 ms
5,492 KB
testcase_22 AC 35 ms
5,336 KB
testcase_23 AC 26 ms
4,748 KB
testcase_24 AC 9 ms
4,380 KB
testcase_25 AC 23 ms
4,376 KB
testcase_26 AC 21 ms
4,520 KB
testcase_27 AC 35 ms
5,396 KB
testcase_28 AC 32 ms
5,084 KB
testcase_29 AC 6 ms
4,376 KB
testcase_30 AC 9 ms
4,380 KB
testcase_31 AC 26 ms
4,840 KB
testcase_32 AC 12 ms
4,380 KB
testcase_33 AC 22 ms
4,376 KB
testcase_34 AC 16 ms
4,376 KB
testcase_35 AC 36 ms
5,340 KB
testcase_36 AC 5 ms
4,376 KB
testcase_37 AC 32 ms
5,224 KB
testcase_38 AC 33 ms
5,156 KB
testcase_39 AC 21 ms
4,380 KB
testcase_40 AC 5 ms
4,380 KB
testcase_41 AC 7 ms
4,376 KB
testcase_42 AC 20 ms
4,376 KB
testcase_43 AC 2 ms
4,380 KB
testcase_44 AC 2 ms
4,376 KB
testcase_45 AC 2 ms
4,380 KB
testcase_46 AC 2 ms
4,376 KB
testcase_47 AC 1 ms
4,380 KB
testcase_48 AC 2 ms
4,380 KB
testcase_49 AC 2 ms
4,376 KB
testcase_50 AC 1 ms
4,380 KB
testcase_51 AC 2 ms
4,380 KB
testcase_52 AC 2 ms
4,376 KB
testcase_53 AC 2 ms
4,380 KB
testcase_54 AC 2 ms
4,376 KB
testcase_55 AC 2 ms
4,376 KB
testcase_56 AC 1 ms
4,376 KB
testcase_57 AC 2 ms
4,380 KB
testcase_58 AC 2 ms
4,380 KB
testcase_59 AC 2 ms
4,376 KB
testcase_60 AC 2 ms
4,380 KB
testcase_61 AC 1 ms
4,376 KB
testcase_62 AC 2 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;

using i64 = int64_t;

const i64 MOD = 1e9+7;

const i64 INF = 1e18+7;

// pythonのrangeのような範囲for文用のclass for(const auto& i : Range<>(10)) のように書く
template <typename T = i64>
struct Range{
    struct iterator{
        T value;
        const T step, last;
        const T& operator*(){return value;}
        iterator(T value, T step, T last) :
            value(value),
            step(step),
            last(last)
        {
        }
        iterator operator++(){value = step < static_cast<T>(0) ? max(value + step, last) : min(value + step, last); return *this;}
        bool operator!=(const iterator& x){return value != x.value;}
    };
    const T start, last, step;

    Range(const T start, const T last, const T step = static_cast<T>(1)) :
        start(start),
        last(last),
        step(step)
    {
    }

    Range(const T last) :
        start(0),
        last(last),
        step(1)
    {
    }

    iterator begin(){return iterator(start, step, last);}
    iterator end(){return iterator(last, step, last);}
};

// lambda式を用いた再帰
template <typename F>
struct FixPoint{
    const F _f;
    FixPoint(F&& f) : _f(forward<F>(f)){}

    template<typename... Types>
    decltype(auto) operator()(Types&&... args) const{
        return _f(*this, forward<Types>(args)...);
    }
};

template <typename F>
static decltype(auto) makeRec(F&& f){
    return FixPoint<F>(forward<F>(f));
}

// 多次元vectorの一斉初期化 makeVector<i64, 0>(a, b, ...)のように書く
template <typename T, T Value = T()>
vector<T> makeVector(size_t x){
    return vector<T>(x, T(Value));
}

template <typename T, T Value = T(), typename... Types>
auto makeVector(size_t x, Types... args){
    return vector<decltype(makeVector<T, Value>(args...))>(x, makeVector<T, Value>(args...));
}

// 最大値を更新し、更新できた時にはtrueを返す
template <typename T = i64>
bool chmax(T& a, T b){
    if(a < b){
        a = b;
        return true;
    }
    return false;
}

// 同様に最小値を更新する
template <typename T = i64>
bool chmin(T& a, T b){
    if(a > b){
        a = b;
        return true;
    }
    return false;
}

// 行数と変数名、値をclogに表示するデバッグ用print
#define dump(x) fprintf(stderr, "line =%4d, name =%7s , ", __LINE__, #x); clog << "value = " << x << endl;

// 同様の配列向けデバッグ用print
#define vecdump(x) fprintf(stderr, "line =%4d, name =%7s\n", __LINE__, #x); _dump_macro(x);

void _dump(int, string& x){
    clog << x << endl;
}

template <typename T>
void _dump(bool, T& x){
    clog << x << " ";
}

template <typename T, typename U = typename T::iterator>
void _dump(int, T& x){

    for(auto& elm : x)
        _dump(0, elm);

    clog << endl;
}

template <typename T>
void _dump_macro(T& x){
    _dump(0, x);
}

// input用の関数群
void _input(int, string& x){
    cin >> x;
}

template <typename T>
void _input(bool, T& x){
    cin >> x;
}

template <typename T, typename U = typename T::iterator>
void _input(int, T& x){

    for(auto& elm : x)
        _input(0, elm);
}

template <typename T>
void input_single(T& x){
    _input(0, x);
}

auto input(){}

template <typename T, typename... Types>
void input(T& value, Types&&... args){
    input_single(value);
    input(forward<Types>(args)...);
};

void _pararell_input(size_t){}

template <typename T, typename... Types>
void _pararell_input(size_t index, T& value, Types&&... args){
    input(value[index]);
    _pararell_input(index, forward<Types>(args)...);
}

template <typename... Types>
void pararell_input(size_t count, Types&&... args){
    for(const auto& i : Range<>(count))
        _pararell_input(i, forward<Types>(args)...);
}


signed main(){

    cin.tie(0);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(20);

    int n;
    input(n);
    auto a = makeVector<i64>(n);
    input(a);

    // 二項係数なので殴れば終わり

    auto mpow = makeRec([&](auto&& f, i64 x, i64 y) -> i64{
        if(y <= 1)
            return y ? x : 1L;
        i64 tmp = f(x, y >> 1);
        tmp = tmp * tmp % MOD;
        return (tmp * f(x, y & 1)) % MOD;
    });

    auto fac = makeVector<i64, 1>(n + 1);
    auto inv = makeVector<i64, 1>(n + 1);
    for(const auto& i : Range<>(n)){
        fac[i + 1] = (fac[i] * (i + 1)) % MOD;
        inv[i + 1] = mpow(fac[i + 1], 1e9 + 5);
    }

    auto ncr = [&](int x, int y){
        return fac[x] * inv[y] % MOD * inv[x - y] % MOD;
    };

    i64 ans = 0;
    for(const auto& i : Range<>(n)){
        ans = (ans + a[i] * ncr(n - 1, i)) % MOD;
    }
    cout << ans << endl;

}
0