#480198 (C++17) No.1050 Zero (Maximum)

• 問題: No.1050 Zero (Maximum)
• ユーザー: wotsushi
• 提出日時: 2020-05-10 17:04:55
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 54 ms / 2,000 ms
• コード長: 7,882 bytes
• コンパイル時間: 2,133 ms
• コンパイル使用メモリ: 216,420 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-07-07 22:13:09
• 合計ジャッジ時間: 3,084 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,816 KB
testcase_01	AC	2 ms; 6,944 KB
testcase_02	AC	13 ms; 6,940 KB
testcase_03	AC	9 ms; 6,940 KB
testcase_04	AC	32 ms; 6,940 KB
testcase_05	AC	35 ms; 6,944 KB
testcase_06	AC	16 ms; 6,944 KB
testcase_07	AC	19 ms; 6,940 KB
testcase_08	AC	3 ms; 6,940 KB
testcase_09	AC	10 ms; 6,940 KB
testcase_10	AC	44 ms; 6,944 KB
testcase_11	AC	31 ms; 6,944 KB
testcase_12	AC	3 ms; 6,944 KB
testcase_13	AC	1 ms; 6,944 KB
testcase_14	AC	2 ms; 6,940 KB
testcase_15	AC	1 ms; 6,944 KB
testcase_16	AC	50 ms; 6,940 KB
testcase_17	AC	54 ms; 6,940 KB

ソースコード

#pragma region template 2.4
#include <bits/stdc++.h>
using namespace std;
template <typename T>
using pq_asc = priority_queue<T, vector<T>, greater<T>>;
typedef long long ll;
typedef vector<ll> vi;
typedef vector<vi> vvi;
typedef pair<ll, ll> ii;
typedef vector<ii> vii;
typedef vector<string> vs;
#define REP(i, n) for (ll i = 0; i < (n); ++i)
#define REP1(i, n) for (ll i = 1; i <= (n); ++i)
#define FOR(i, a) for (auto &i : a)
#define CH(f, x, y) x = f(x, y)
#define IN(T, x) \
    T x;         \
    cin >> x;
#define AIN(T, a, n) \
    vector<T> a(n);  \
    FOR(i, a)        \
    cin >> i;
#define A2IN(T1, a, T2, b, n) \
    vector<T1> a(n);          \
    vector<T2> b(n);          \
    REP(i, n)                 \
    cin >> a[i] >> b[i];
#define OUT(x) cout << (x) << endl;
#define FOUT(x) cout << fixed << setprecision(15) << (x) << endl;
#define ALL(a) (a).begin(), (a).end()
#define SORT(a) sort(ALL(a))
#define RSORT(a) \
    SORT(a);     \
    reverse(ALL(a))
#define DUMP(x) cout << #x << " = " << (x) << endl;
#define DUMPA(a)          \
    cout << #a << " = {"; \
    JOUT(ALL(a), ", ", cout) << "}" << endl;

template <typename T>
ostream &JOUT(T s, T e, string sep = " ", ostream &os = cout)
{
    if (s != e)
    {
        os << *s;
        ++s;
    }
    while (s != e)
    {
        os << sep << *s;
        ++s;
    }
    return os;
}

ostream &YES(bool cond, string yes = "Yes", string no = "No", ostream &os = cout)
{
    if (cond)
    {
        os << yes << endl;
    }
    else
    {
        os << no << endl;
    }
    return os;
}

template <typename T1, typename T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p)
{
    os << '(' << p.first << ", " << p.second << ')';
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, const vector<T> &v)
{
    os << '[';
    JOUT(ALL(v), ", ", os) << ']';
    return os;
}

const ll INF = 1e18;
const ll MOD = 1e9 + 7;
#pragma endregion template
#pragma region mint 1.1
struct mint
{
    ll a;

    mint(ll a = 0) : a((a % MOD + MOD) % MOD) {}
    mint pow(ll y)
    {
        ll exp = y;
        mint res = 1;
        mint p = a;
        while (exp)
        {
            if (exp & 1)
            {
                res *= p;
            }
            exp >>= 1;
            p *= p;
        }
        return res;
    }
    mint pow(mint y)
    {
        return pow(y.a);
    }
    mint &operator+=(ll y)
    {
        *this = a + y;
        return *this;
    }
    mint &operator+=(mint y)
    {
        *this += y.a;
        return *this;
    }
    mint &operator-=(ll y)
    {
        *this = a - y;
        return *this;
    }
    mint &operator-=(mint y)
    {
        *this -= y.a;
        return *this;
    }
    mint &operator*=(ll y)
    {
        *this = a * y;
        return *this;
    }
    mint &operator*=(mint y)
    {
        *this *= y.a;
        return *this;
    }
    mint &operator/=(ll y)
    {
        *this *= mint(y).pow(MOD - 2);
        return *this;
    }
    mint &operator/=(const mint y)
    {
        *this /= y.a;
        return *this;
    }
    mint &operator++()
    {
        *this += 1;
        return *this;
    }
    mint &operator++(int)
    {
        *this += 1;
        return *this;
    }
    mint &operator--()
    {
        *this -= 1;
        return *this;
    }
    mint &operator--(int)
    {
        *this -= 1;
        return *this;
    }
};

mint operator+(const mint &x, const mint &y)
{
    return mint(x.a) += y;
}
mint operator+(const mint &x, const ll &y)
{
    return mint(x.a) += y;
}
mint operator+(const ll &x, mint &y)
{
    return mint(x) += y;
}
mint operator-(const mint &x, const mint &y)
{
    return mint(x.a) -= y;
}
mint operator-(const mint &x, const ll &y)
{
    return mint(x.a) -= y;
}
mint operator-(const ll &x, const mint &y)
{
    return mint(x) -= y;
}
mint operator*(const mint &x, const mint &y)
{
    return mint(x.a) *= y;
}
mint operator*(const mint &x, const ll &y)
{
    return mint(x.a) *= y;
}
mint operator*(const ll &x, const mint &y)
{
    return mint(x) *= y;
}
mint operator/(const mint &x, const mint &y)
{
    return mint(x.a) /= y;
}
mint operator/(const mint &x, const ll &y)
{
    return mint(x.a) /= y;
}
mint operator/(const ll &x, const mint &y)
{
    return mint(x) /= y;
}
bool operator==(const mint &x, const mint &y)
{
    return x.a == y.a;
}
bool operator==(const mint &x, const ll &y)
{
    return x.a == y;
}
bool operator==(const ll &x, mint &y)
{
    return x == y.a;
}
bool operator!=(const mint &x, const mint &y)
{
    return !(x == y);
}
bool operator!=(const mint &x, const ll &y)
{
    return !(x == y);
}
bool operator!=(const ll &x, mint &y)
{
    return !(x == y);
}
istream &operator>>(istream &is, mint &i)
{
    is >> i.a;
    return is;
}
ostream &operator<<(ostream &os, const mint &i)
{
    os << i.a;
    return os;
}

typedef vector<mint> vm;
#pragma endregion mint
#pragma region matrix 1.0
template <typename T>
struct matrix
{
    ll n, m;
    vector<vector<T>> A;

    matrix() {}
    matrix(ll n) : n(n), m(n)
    {
        A.assign(n, vector<T>(n));
    }
    matrix(ll n, ll m) : n(n), m(m)
    {
        A.assign(n, vector<T>(m));
    }
    matrix(vector<vector<T>> X) : A(X)
    {
        n = X.size();
        m = X[0].size();
    }
    matrix pow(ll k)
    {
        matrix<T> res = matrix(n, n);
        REP(i, n)
        {
            res[i][i] = 1;
        }
        matrix P = A;
        while (k)
        {
            if (k & 1)
            {
                res *= P;
            }
            k >>= 1;
            P *= P;
        }
        return res;
    }
    matrix &operator+=(matrix<T> B)
    {
        REP(i, n)
        {
            REP(j, m)
            {
                A[i][j] += B[i][j];
            }
        }
        return *this;
    }
    matrix &operator-=(matrix<T> B)
    {
        REP(i, n)
        {
            REP(j, m)
            {
                A[i][j] -= B[i][j];
            }
        }
        return *this;
    }
    matrix &operator*=(T x)
    {
        REP(i, n)
        {
            REP(j, m)
            {
                A[i][j] *= x;
            }
        }
        return *this;
    }
    matrix &operator*=(matrix<T> B)
    {
        vector<vector<T>> X(n, vector<T>(m));
        REP(i, n)
        {
            REP(j, B.m)
            {
                REP(k, m)
                {
                    X[i][j] += A[i][k] * B[k][j];
                }
            }
        }
        A = X;
        return *this;
    }
    matrix &operator/=(T x)
    {
        REP(i, n)
        {
            REP(j, m)
            {
                A[i][j] /= x;
            }
        }
        return *this;
    }

    vector<T> &operator[](int i)
    {
        return A[i];
    }
};

template <typename T>
matrix<T> operator+(const matrix<T> &A, const matrix<T> &B)
{
    return matrix(A.A) += B;
}
template <typename T>
matrix<T> operator-(const matrix<T> &A, const matrix<T> &B)
{
    return matrix(A.A) -= B;
}
template <typename T>
matrix<T> operator*(const matrix<T> &A, const T &x)
{
    return matrix(A.A) *= x;
}
template <typename T>
matrix<T> operator*(const T &x, const matrix<T> &A)
{
    return matrix(A.A) *= x;
}
template <typename T>
matrix<T> operator*(const matrix<T> &A, const matrix<T> &B)
{
    return matrix(A.A) *= B;
}
template <typename T>
matrix<T> operator/(const matrix<T> &A, const T &x)
{
    return matrix(A.A) /= x;
}
template <typename T>
bool operator==(const matrix<T> &x, const matrix<T> &y)
{
    return x.A == y.A;
}
template <typename T>
bool operator!=(const matrix<T> &x, const matrix<T> &y)
{
    return x.A != y.A;
}

#pragma endregion matrix

int main()
{
    IN(ll, M);
    IN(ll, K);
    matrix<mint> A(M);
    REP(i, M)
    {
        REP(j, M)
        {
            ++A[(i + j) % M][j];
            ++A[(i * j) % M][j];
        }
    }
    OUT(A.pow(K)[0][0]);
}