結果

問題 No.1050 Zero (Maximum)
ユーザー kusaf_kusaf_
提出日時 2024-11-02 02:43:05
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 19 ms / 2,000 ms
コード長 2,533 bytes
コンパイル時間 3,848 ms
コンパイル使用メモリ 259,540 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-02 02:43:11
合計ジャッジ時間 5,169 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 5 ms
6,816 KB
testcase_03 AC 4 ms
6,816 KB
testcase_04 AC 12 ms
6,820 KB
testcase_05 AC 13 ms
6,816 KB
testcase_06 AC 7 ms
6,820 KB
testcase_07 AC 8 ms
6,816 KB
testcase_08 AC 2 ms
6,820 KB
testcase_09 AC 5 ms
6,816 KB
testcase_10 AC 17 ms
6,820 KB
testcase_11 AC 12 ms
6,820 KB
testcase_12 AC 2 ms
6,816 KB
testcase_13 AC 2 ms
6,820 KB
testcase_14 AC 3 ms
6,816 KB
testcase_15 AC 2 ms
6,816 KB
testcase_16 AC 17 ms
6,816 KB
testcase_17 AC 19 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using namespace atcoder;
using ll = long long;
using mint = modint1000000007;

template<typename T> struct Matrix : vector<vector<T>> {
  using vector<vector<T>>::vector;
  using vector<vector<T>>::operator=;
  Matrix() {}
  Matrix(ll n) {
    this->assign(n, vector<T>(n, 0));
    for(ll i = 0; i < n; i++) { (*this)[i][i] = 1; }
  }
  Matrix(ll n, ll m, T x = 0) {
    for(ll i = 0; i < n; i++) { this->push_back(vector<T>(m, x)); }
  }
  Matrix(vector<vector<T>> v) { *this = v; }
  Matrix operator+(const Matrix &m) const { return Matrix(*this) += m; }
  Matrix operator-(const Matrix &m) const { return Matrix(*this) -= m; }
  Matrix operator*(const Matrix &m) const { return Matrix(*this) *= m; }
  Matrix operator*(const T &x) const { return Matrix(*this) *= x; }
  Matrix operator^(ll n) const { return Matrix(*this) ^= n; }
  Matrix operator+=(const Matrix &m) const {
    ll h = this->size(), w = (*this)[0].size();
    assert(h == m.size() && w == m[0].size());
    for(ll i = 0; i < h; i++) {
      for(ll j = 0; j < w; j++) { *this[i][j] += m[i][j]; }
    }
    return *this;
  }
  Matrix operator-=(const Matrix &m) {
    ll h = this->size(), w = (*this)[0].size();
    assert(h == m.size() && w == m[0].size());
    for(ll i = 0; i < h; i++) {
      for(ll j = 0; j < w; j++) { *this[i][j] -= m[i][j]; }
    }
    return *this;
  }
  Matrix operator*=(const Matrix &m) {
    ll h = this->size(), w = (*this)[0].size();
    assert(w == (ll)m.size());
    vector<vector<T>> r(h, vector<T>(m[0].size(), T(0)));
    for(ll i = 0; i < h; i++) {
      for(ll j = 0; j < (ll)m[0].size(); j++) {
        for(ll k = 0; k < w; k++) { r[i][j] += (*this)[i][k] * m[k][j]; }
      }
    }
    this->swap(r);
    return *this;
  }
  Matrix operator*=(const T &x) {
    ll h = this->size(), w = (*this)[0].size();
    for(ll i = 0; i < h; i++) {
      for(ll j = 0; j < w; j++) { *this[i][j] *= x; }
    }
    return *this;
  }
  Matrix operator^=(ll n) {
    ll h = this->size();
    Matrix m(h);
    while(n) {
      if(n & 1) { m *= *this; }
      *this *= *this;
      n >>= 1LL;
    }
    this->swap(m);
    return *this;
  }
};

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  ll N, K;
  cin >> N >> K;

  Matrix<mint> dp(N, 1), m(N, N);
  dp[0][0] = 1;

  for(ll i = 0; i < N; i++) {
    for(ll j = 0; j < N; j++) {
      m[(i + j) % N][i]++;
      m[i * j % N][i]++;
    }
  }

  dp = (m ^ K) * dp;
  cout << dp[0][0].val() << "\n";
}
0