結果

問題 No.840 ほむほむほむら
ユーザー torisasami4torisasami4
提出日時 2019-06-14 23:21:20
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,951 ms / 4,000 ms
コード長 3,274 bytes
コンパイル時間 934 ms
コンパイル使用メモリ 76,152 KB
実行使用メモリ 8,576 KB
最終ジャッジ日時 2024-04-26 22:07:36
合計ジャッジ時間 11,838 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
7,808 KB
testcase_01 AC 8 ms
7,808 KB
testcase_02 AC 38 ms
8,016 KB
testcase_03 AC 256 ms
8,576 KB
testcase_04 AC 5 ms
7,936 KB
testcase_05 AC 5 ms
7,808 KB
testcase_06 AC 12 ms
7,936 KB
testcase_07 AC 86 ms
8,064 KB
testcase_08 AC 472 ms
8,448 KB
testcase_09 AC 11 ms
7,808 KB
testcase_10 AC 4 ms
7,796 KB
testcase_11 AC 17 ms
7,748 KB
testcase_12 AC 131 ms
7,936 KB
testcase_13 AC 1,239 ms
8,448 KB
testcase_14 AC 159 ms
8,064 KB
testcase_15 AC 6 ms
7,808 KB
testcase_16 AC 23 ms
7,808 KB
testcase_17 AC 290 ms
7,936 KB
testcase_18 AC 1,576 ms
8,528 KB
testcase_19 AC 1,951 ms
8,448 KB
testcase_20 AC 6 ms
7,808 KB
testcase_21 AC 6 ms
7,936 KB
testcase_22 AC 28 ms
7,936 KB
testcase_23 AC 1,833 ms
8,576 KB
testcase_24 AC 27 ms
7,808 KB
testcase_25 AC 5 ms
7,808 KB
testcase_26 AC 39 ms
7,936 KB
testcase_27 AC 1,845 ms
8,448 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<list>
#include<set>
#include<string>
using namespace std;
int gcd(int a, int b) {
	int c = a % b;
	while (c != 0) {
		a = b;
		b = c;
		c = a % b;
	}
	return b;
}
struct UnionFind {
	vector<int> par; // par[i]:iの親の番号　(例) par[3] = 2 : 3の親が2

	UnionFind(int N) : par(N) { //最初は全てが根であるとして初期化
		for (int i = 0; i < N; i++) par[i] = i;
	}

	int root(int x) { // データxが属する木の根を再帰で得る：root(x) = {xの木の根}
		if (par[x] == x) return x;
		return par[x] = root(par[x]);
	}

	void unite(int x, int y) { // xとyの木を併合
		int rx = root(x); //xの根をrx
		int ry = root(y); //yの根をry
		if (rx == ry) return; //xとyの根が同じ(=同じ木にある)時はそのまま
		par[rx] = ry; //xとyの根が同じでない(=同じ木にない)時：xの根rxをyの根ryにつける
	}

	bool same(int x, int y) { // 2つのデータx, yが属する木が同じならtrueを返す
		int rx = root(x);
		int ry = root(y);
		return rx == ry;
	}
};
typedef long long ll;

ll M = 998244353;

vector<ll> fac(300001); //n!(mod M)
vector<ll> ifac(300001); //k!^{M-2} (mod M)

ll mpow(ll x, ll n) {
	ll ans = 1;
	while (n != 0) {
		if (n & 1) ans = ans * x % M;
		x = x * x % M;
		n = n >> 1;
	}
	return ans;
}
ll comb(ll a, ll b) {
	if (a == 0 && b == 0)return 1;
	if (a < b || a < 0)return 0;
	ll tmp = ifac[a - b] * ifac[b] % M;
	return tmp * fac[a] % M;
}
// mod. m での a の逆元 a^{-1} を計算する
long long modinv(long long a) {
	long long b = M, u = 1, v = 0;
	while (b) {
		long long t = a / b;
		a -= t * b; swap(a, b);
		u -= t * v; swap(u, v);
	}
	u %= M;
	if (u < 0) u += M;
	return u;
}
vector<vector<ll>> mul(vector<vector<ll>> a,vector<vector<ll>> b,int n){
	int i,j,k,t;
	vector<vector<ll>> c(n);
	for (i = 0; i < n; i++) {
		for (j = 0; j < n; j++) {
			t = 0;
			for (k = 0; k < n; k++)
				t = (t + a[i][k]*b[k][j] % M)%M;
			c[i].push_back(t);
		}
	}
	return c;
}
int main() {
	ll n, i,j,a,b,c,d,e,f, k,p,ans=0;
	cin >> n >> k;
	p = k * k*k;
	vector<vector<ll>> v(p);
	for (a = 0; a < k; a++) {
		for (b = 0; b < k; b++) {
			for (c = 0; c < k; c++) {
				for (d = 0; d < k; d++) {
					for (e = 0; e < k; e++) {
						for (f = 0; f < k; f++) {
							v[k*k*a + k * b + c].push_back(0);
						}
					}
				}
			}
		}
	}
	for (a = 0; a < k; a++) {
		for (b = 0; b < k; b++) {
			for (c = 0; c < k; c++) {
				for (d = 0; d < k; d++) {
					for (e = 0; e < k; e++) {
						for (f = 0; f < k; f++) {
							if (a == (d + 1) % k&&b == e && c == f)
								v[k*k*a + k * b + c][k*k*d + k * e + f]++;
							if (a == d&&b == (e+a)%k && c == f)
								v[k*k*a + k * b + c][k*k*d + k * e + f]++;
							if (a == d && b == e&& c == (f+b)%k)
								v[k*k*a + k * b + c][k*k*d + k * e + f]++;
						}
					}
				}
			}
		}
	}
	vector<vector<ll>> v2(p);
	for (i = 0; i < p; i++) {
		for (j = 0; j < p; j++) {
			if (i == j)
				v2[i].push_back(1);
			else
				v2[i].push_back(0);
		}
	}
	while (n > 0) {
		if (n % 2 == 1)
			v2 = mul(v, v2, p);
		v = mul(v, v, p);
		n /= 2;
	}
	for (a = 0; a < k; a++)
		for (b = 0; b < k; b++)
			ans = (ans + v2[k*k * a + k * b][0]) % M;
	cout << ans << endl;
}
0