結果

問題 No.1050 Zero (Maximum)
コンテスト
ユーザー i_mrhj
提出日時 2020-05-11 10:13:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 130 ms / 2,000 ms
コード長 1,862 bytes
コンパイル時間 818 ms
コンパイル使用メモリ 96,264 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-18 05:43:55
合計ジャッジ時間 2,344 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <cstdio>
#include <climits>
#include <cmath>
#include <iostream>
#include <iomanip>
#include <string>
#include <cstdio>
#include <climits>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <utility>
#include <queue>
#include <cstring>
#include <set>
#include <map>
#include <complex>

#define rep(i, n) for (int i = 0; i < int(n); i++)
using namespace std;
long long MOD = 1000000007;
long long INF = 1000000000000000; //10^15
typedef long long ll;
typedef unsigned long long ull;


ll powMod(ll x, ll n, ll mod) {
  if (n == 0) return 1;
  ll t = powMod(x, n/2, mod);
  t = t * t % mod;
  if (n & 1) return t * x % mod;
  return t;
}

ll gcd(ll a, ll b) {
  if (a == 0 || b == 0) return a + b;
  if (b > a) return gcd(b, a);
  return gcd(b, a % b);
}

void MatrixPro(ll *g, ll *h, ll *m, int size) {
  //g * h = m
  rep(i, size) rep(j, size) {
    m[size * i + j] = 0;
    rep(k, size) {
      m[size * i + j] += g[size * i + k] * h[size * k + j] % MOD;
      m[size * i + j] %= MOD;
    }
  }
  return;
}

//g^n = h, size <= 100
void MatrixExpo(ll *g, ll *h, int size, ll n) {
  if (n == 0) {
    rep(i, size) rep(j, size) {
      if (i == j) h[size * i + j] = 1;
      else h[size * i + j] = 0;
    }
    return;
  }
  
  ll a[10010];
  MatrixExpo(g, a, size, n / 2);
  ll b[10010];
  MatrixPro(a, a, b, size);
  if (!(n & 1)) {
    rep(i, size) rep(j, size) {
      h[size * i + j] = b[size * i + j];
    }
    return;
  } else {
    MatrixPro(b, g, h, size);
    return;
  }
}


int main(void) {

  ll m, k;
  cin >> m >> k;

  ll x[3000] = {};
  for (int i = 0; i < m; i++) {
    for (int j = 0; j < m; j++) {
      int t = i*j%m;
      x[t*m+i]++;
      t = (i+j)%m;
      x[t*m+i]++;
    }
  }
  //rep(i, m*m) cout << x[i] << endl;

  ll y[3000] = {};
  MatrixExpo(x, y, m, k);

  cout << y[0] << endl;

}
0