結果

問題 No.1050 Zero (Maximum)
ユーザー akun0716akun0716
提出日時 2020-05-09 08:59:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,865 bytes
コンパイル時間 872 ms
コンパイル使用メモリ 89,996 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-05 04:54:10
合計ジャッジ時間 1,745 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include<vector>
#include<algorithm>
#include<string>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<math.h>
using namespace std;
typedef long long ll;
#define int long long
typedef vector<int> VI;
typedef pair<int, int> pii;
typedef vector<pii> VP;
typedef vector<string> VS;
typedef priority_queue<int> PQ;
template<class T>bool chmax(T &a, const T &b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b < a) { a = b; return 1; } return 0; }
#define fore(i,a) for(auto &i:a)
#define REP(i,n) for(int i=0;i<n;i++)
#define eREP(i,n) for(int i=0;i<=n;i++)
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define eFOR(i,a,b) for(int i=(a);i<=(b);++i)
#define SORT(c) sort((c).begin(),(c).end())
#define rSORT(c) sort((c).rbegin(),(c).rend())
#define LB(x,a) lower_bound((x).begin(),(x).end(),(a))
#define UB(x,a) upper_bound((x).begin(),(x).end(),(a))
#define INF 1000000000
#define LLINF 9223372036854775807
#define mod 1000000007
//priority_queue<int,vector<int>, greater<int> > q2;

typedef vector<VI> mat;

mat mul(mat &A, mat &B) {
	mat C(A.size(), VI(B[0].size()));
	REP(i, A.size()) {
		REP(k, B.size()) {
			REP(j, B[0].size()) {
				C[i][j] = (C[i][j] + (A[i][k] * B[k][j]));
				C[i][j] %= mod;
			}
		}
	}
	return C;
}

mat pow(mat A, int N) {
	mat B(A.size(), VI(A.size()));
	REP(i, A.size())B[i][i] = 1;
	while (N > 0) {
		if (N & 1)B = mul(A, B);
		A = mul(A, A);
		N /= 2;
	}
	return B;
}


signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);

	int M, K; cin >> M >> K;

	mat C(M, VI(M));
	REP(i, M) {
		REP(j, M) {
			int cnt = 0;
			REP(k, M) {
				if ((i + k) % M == j)cnt++;
				if ((i * k) % M == j)cnt++;
			}

			C[i][j] = cnt;
		}
	}
	C = pow(C, K);
	mat A(1, VI(M, 0));
	A[0][0] = 1;
	mat ans = mul(C, A);
	cout << ans[0][0] << endl;

	return 0;
}

0