#381673 (C++14) No.891 隣接3項間の漸化式

提出ソース

結果

問題	No.891 隣接3項間の漸化式
ユーザー	ganmodokix
提出日時	2019-09-20 22:36:20
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2 ms / 2,000 ms
コード長	6,694 bytes
コンパイル時間	2,887 ms
コンパイル使用メモリ	190,864 KB
実行使用メモリ	6,824 KB
最終ジャッジ日時	2024-11-14 21:40:17
合計ジャッジ時間	4,158 ms
ジャッジサーバーID （参考情報）	judge5 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,816 KB
testcase_02	AC	2 ms 6,820 KB
testcase_03	AC	2 ms 6,816 KB
testcase_04	AC	2 ms 6,820 KB
testcase_05	AC	2 ms 6,816 KB
testcase_06	AC	2 ms 6,816 KB
testcase_07	AC	2 ms 6,816 KB
testcase_08	AC	2 ms 6,820 KB
testcase_09	AC	2 ms 6,820 KB
testcase_10	AC	2 ms 6,816 KB
testcase_11	AC	2 ms 6,820 KB
testcase_12	AC	2 ms 6,816 KB
testcase_13	AC	2 ms 6,816 KB
testcase_14	AC	2 ms 6,816 KB
testcase_15	AC	2 ms 6,820 KB
testcase_16	AC	2 ms 6,820 KB
testcase_17	AC	2 ms 6,816 KB
testcase_18	AC	2 ms 6,816 KB
testcase_19	AC	2 ms 6,820 KB
testcase_20	AC	2 ms 6,816 KB
testcase_21	AC	2 ms 6,816 KB
testcase_22	AC	2 ms 6,820 KB
testcase_23	AC	2 ms 6,820 KB
testcase_24	AC	2 ms 6,816 KB
testcase_25	AC	2 ms 6,816 KB
testcase_26	AC	2 ms 6,816 KB
testcase_27	AC	2 ms 6,820 KB
testcase_28	AC	2 ms 6,820 KB
testcase_29	AC	2 ms 6,816 KB
testcase_30	AC	2 ms 6,816 KB
testcase_31	AC	2 ms 6,816 KB
testcase_32	AC	2 ms 6,820 KB
testcase_33	AC	2 ms 6,816 KB
testcase_34	AC	2 ms 6,824 KB
testcase_35	AC	2 ms 6,816 KB
testcase_36	AC	2 ms 6,820 KB
testcase_37	AC	2 ms 6,816 KB
testcase_38	AC	2 ms 6,816 KB
testcase_39	AC	2 ms 6,816 KB
testcase_40	AC	2 ms 6,816 KB
testcase_41	AC	2 ms 6,820 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

// May this submission get accepted!

#pragma GCC optimize ("O3")
#pragma GCC target ("tune=native")
#pragma GCC target ("avx")

#include <bits/stdc++.h>

// 汎用マクロ
#define ALL_OF(x) (x).begin(), (x).end()
#define REP(i,n) for (long long i=0, i##_len=(n); i<i##_len; i++)
#define RANGE(i,is,ie) for (long long i=(is), i##_end=(ie); i<=i##_end; i++)
#define DSRNG(i,is,ie) for (long long i=(is), i##_end=(ie); i>=i##_end; i--)
#define UNIQUE(v) do { sort((v).begin(), (v).end()); (v).erase(unique((v).begin(), (v).end()), (v).end()); } while (false)
template<class T> bool chmax(T &a, const T &b) {if (a < b) {a = b; return true;} return false; }
template<class T> bool chmin(T &a, const T &b) {if (a > b) {a = b; return true;} return false; }
#define INF 0x7FFFFFFF
#define LINF 0x7FFFFFFFFFFFFFFFLL
#define Yes(q) ((q) ? "Yes" : "No")
#define YES(q) ((q) ? "YES" : "NO")
#define Possible(q) ((q) ? "Possible" : "Impossible")
#define POSSIBLE(q) ((q) ? "POSSIBLE" : "IMPOSSIBLE")
#define DUMP(q) cerr << "[DEBUG] " #q ": " << (q) << " at " __FILE__ ":" << __LINE__ << endl
#define DUMPALL(q) do { cerr << "[DEBUG] " #q ": ["; REP(i, (q).size()) { cerr << (q)[i] << (i == i_len-1 ? "" : ", "); } cerr << "] at " __FILE__ ":" << __LINE__ << endl; } while (false)
template<class T> T gcd(const T &a, const T &b) { return a < b ? gcd(b, a) : b ? gcd(b, a % b) : a; }
template<class T> T lcm(const T &a, const T &b) { return a / gcd(a, b) * b; }

// gcc拡張マクロ
#define popcount __builtin_popcount
#define popcountll __builtin_popcountll

// エイリアス
#define DANCE_ long
#define ROBOT_ unsigned
#define HUMAN_ signed
#define CHOKUDAI_ const
using  ll = DANCE_ HUMAN_ DANCE_;
using ull = DANCE_ ROBOT_ DANCE_;
using cll = DANCE_ DANCE_ CHOKUDAI_;
using  ld = long double;
using namespace std;

// モジュール
// pdivの余りを取りたくない場合はmatrix<T>::operatorのコメントアウトをいじくれ
constexpr ll pdiv = 1000000007LL;
// vector<vector<T>> のラッパクラス (プァ～)
template<typename T = long long>
class matrix {
    vector<vector<T>> value;
public:
    T get(const size_t i) const { return this->rows() == 1 ? this->get(0, i) : this->get(i, 0); }
    T get(const size_t i, const size_t j) const { return this->value.at(i).at(j); }
    inline size_t rows() const { return this->value.size(); }
    inline size_t cols() const { return this->value.empty() ? 0 : this->value[0].size(); }
    matrix() : value() {}
    matrix(const matrix &a) : value(a.value) {}
    matrix(size_t n, size_t m = 0) : value(n, vector<T>(m ? m : n, 0)) {}
    matrix(const vector<vector<T>> &a) : value(a) {}
    matrix(initializer_list<initializer_list<T>> init) : value() {
        const size_t n = init.size();
        const size_t m = [&]() {
            size_t maxm = 0;
            for (const auto& initi : init) {
                const size_t mi = initi.size();
                if (mi > maxm) maxm = mi;
            }
            return maxm;
        }();
        this->value = vector<vector<T>>(n, vector<T>(m, 0));
        size_t i = 0, j = 0;
        for (const auto &initi : init) {
            for (const auto &v : initi) {
                this->value[i][j++] = v;
            }
            i++; j = 0;
        }
    }
    // ラムダ式の引数は(size_t, size_t)、返り値はTとすること (まあ整数型なら大丈夫でしょ)
    template <class LAMBDA>
    static matrix<T> map(const size_t n, const size_t m, LAMBDA f) {
        matrix<T> r(n, m);
        for (size_t i = 0; i < n; i++) {
            for (size_t j = 0; j < m; j++) {
                r.value[i][j] = f(i, j);
            }
        }
        return r;
    }
    static matrix eye(const size_t n) {
        matrix<T> r(n);
        for (size_t i = 0; i < n; i++) r.value[i][i] = 1;
        return r;
    }
    matrix<T> operator+ (const matrix<T> &a) {
        if (this->rows() != a.rows() || this->cols() != a.cols()) abort();
        const size_t n = this->rows();
        const size_t m = this->cols();
        return matrix<T>::map(n, m, [&](size_t i, size_t j) {
            // return this->value[i][j] + a.value[i][j];
            return (this->value[i][j] + a.value[i][j]) % pdiv;
        });
    }
    matrix<T> operator- (const matrix<T> &a) {
        if (this->rows() != a.rows() || this->cols() != a.cols()) abort();
        const size_t n = this->rows();
        const size_t m = this->cols();
        return matrix<T>::map(n, m, [&](size_t i, size_t j) {
            // return this->value[i][j] - a.value[i][j];
            return (this->value[i][j] + pdiv - a.value[i][j]) % pdiv;
        });
    }
    matrix<T> operator* (const matrix<T> &a) {
        if (this->cols() != a.rows()) abort();
        const size_t n = this->rows();
        const size_t m = this->cols();
        const size_t q = a.cols();
        return matrix<T>::map(n, q, [&](size_t i, size_t j) {
            T term = 0;
            for (size_t k = 0; k < m; k++)
                // term += this->value[i][k] * a.value[k][j];
                (term += this->value[i][k] * a.value[k][j]) %= pdiv;
            return term;
        });
    }
    matrix<T>& operator+= (const matrix<T> &a) {
        matrix<T> r = *this + a;
        this->value.swap(r.value);
        return *this;
    }
    matrix<T>& operator-= (const matrix<T> &a) {
        matrix<T> r = *this - a;
        this->value.swap(r.value);
        return *this;
    }
    matrix<T>& operator*= (const matrix<T> &a) {
        matrix<T> r = *this * a;
        this->value.swap(r.value);
        return *this;
    }
    matrix<T> transpose() {
        const size_t n = this->rows();
        const size_t m = this->cols();
        return matrix<T>::map(m, n, [&](size_t i, size_t j) {
            return this->value[j][i];
        });
    }
    matrix<T> pow(long long int n) {
        if (n == 0) {
            return matrix<T>::eye(this->rows());
        } else if (n == 1) {
            return *this;
        } else {
            matrix temp = this->pow(n / 2);
            return n % 2 == 0 ? temp * temp : temp * temp * *this;
        }
    }
};
template <typename T>
void DUMPMTR(const matrix<T> &m) {
    size_t r = m.rows(), c = m.cols();
    for (size_t i = 0; i < r; i++) {
        cerr << " ["[!i] << "[";
        for (size_t j = 0; j < c; j++) {
            cerr << setw(3) << m.get(i, j) << (j == c-1 ? "]" : ", ");
        }
        cerr << "]" << "\n]"[i == r-1];
    }
    cerr << "\n";
}

// 処理内容
int main() {
    
    ll a, b, n; cin >> a >> b >> n;
    matrix<ll> x0 = {{1}, {0}};
    matrix<ll> shazo = {
        {a, b},
        {1, 0}
    };

    cout << (shazo.pow(n) * x0).get(1) << endl;
    
}

yukicoder

結果

テストケース

ソースコード