結果

問題 No.1050 Zero (Maximum)
コンテスト
ユーザー betit0919
提出日時 2020-05-08 22:07:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 1,770 bytes
コンパイル時間 870 ms
コンパイル使用メモリ 103,528 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-04 00:45:38
合計ジャッジ時間 1,791 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <map>
#include <numeric>
#include <random>
#include <queue>
#include <deque>
#include <tuple>
#include <iomanip>
#include <iterator>
#include <functional>

using namespace std;
typedef long long ll;

const int INF = (1 << 30) - 1;
const ll INFLL= (1LL << 61) - 1;
const int MOD = 1000000007;
#define ALL(a) (a).begin(),(a).end()
#define rALL(a) (a).rbegin(),(a).rend()
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n) FOR(i,0,n)

struct matrix{
  ll space[50][50];

  matrix operator *(const matrix &another) const{
    matrix ans;
    for(int i = 0; i < 50; i++){
      for(int j = 0; j < 50; j++){
        ans.space[i][j] = 0;
        for(int k = 0; k < 50; k++){
          ans.space[i][j] += space[i][k] * another.space[k][j];
          ans.space[i][j] %= MOD;
        }
      }
    }
    return ans;
  }
};

const matrix A={{{1, 1, 1, 1}, {1, 0, 0, 0}, {0, 1, 0, 0}, {0, 0, 1, 0}}};
const matrix E={{{1, 0, 0, 0}, {0, 1, 0, 0}, {0, 0, 1, 0}, {0, 0, 0, 1}}};

matrix power(long exponent, matrix &A, matrix &E){
  if(exponent % 2){
    return power(exponent - 1, A, E) * A;
  }else if(exponent){
    matrix root_ans = power(exponent / 2, A, E);
    return root_ans * root_ans;
  }else{
    return E;
  }
}


int main(){
  cin.tie(nullptr);
  ios::sync_with_stdio(false);
  ll M,K;
  cin>>M>>K;
  matrix A,E;
  REP(i,M){
    REP(j,M){
      E.space[i][j]=(i==j);
    }
  }

  REP(i,M){
    REP(j,M){
      ll tmp=0;
      // 足し算
      REP(k,M){
        tmp+=((i+k)%M==j);
      }
      // 掛け算
      REP(k,M){
        tmp+=((i*k)%M==j);
      }
      A.space[i][j]=tmp;
    }
  }

  cout<<power(K, A, E).space[0][0]<<endl;
}
0