結果

問題 No.1516 simple 門松列 problem Re:MASTER
コンテスト
ユーザー ayoiyouyoeyo
提出日時 2021-05-21 21:55:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 825 ms / 6,000 ms
コード長 2,908 bytes
コンパイル時間 3,407 ms
コンパイル使用メモリ 176,204 KB
実行使用メモリ 7,040 KB
最終ジャッジ日時 2024-10-10 08:30:30
合計ジャッジ時間 7,719 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <string>
#include <cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<iomanip>
#include<bitset>
#define _USE_MATH_DEFINES
#include <math.h>
#include <functional>
#include<complex>
#include<cassert>
#include<random>
using namespace std;
#include<atcoder/all>
using namespace atcoder;


#define rep(i,x) for(ll i=0;i<x;i++)
#define repn(i,x) for(ll i=1;i<=x;i++)

typedef long long ll;
typedef pair<ll, ll> P;
const ll INF = 1e17;
const ll MAX = 4000001;
const long double eps = 1E-14;

ll max(ll a, ll b) {
	if (a > b) { return a; }
	return b;
}

ll min(ll a, ll b) {
	if (a > b) { return b; }
	return a;
}

ll gcd(ll a, ll b) {
	if (b == 0) { return a; }
	if (a < b) { return gcd(b, a); }
	return gcd(b, a % b);
}

ll lcm(ll a, ll b) {
	return a / gcd(a, b) * b;
}

struct edge {
	ll id;
	ll fr;
	ll to;
	ll d;
};

using mint = modint;

typedef vector<ll> vll;
typedef vector <vector<ll>> vvll;
typedef vector<vector<vector<ll>>> vvvll;

typedef vector<mint> vmint;
typedef vector<vector<mint>> vvmint;
typedef vector<vector<vector<mint>>> vvvmint;

vmint f, finv, inv;

void cominit(ll N) {
	ll MOD = modint::mod();//デフォルトが998244353に注意

	f.assign(N + 1, 1);
	finv.assign(N + 1, 1);
	inv.assign(N + 1, 1);
	inv[1] = 1;

	repn(i, N) {
		f[i] = f[i - 1] * i;
		if (i > 1)inv[i] = -inv[MOD % i] * (MOD / i);
		finv[i] = finv[i - 1] * inv[i];
	}
}

mint com(ll a, ll b) {
	if (a < 0 || b < 0 || a < b) { return 0; }
	return f[a] * finv[b] * finv[a - b];
}


/////////////////////////////////////

vvmint prod(vvmint a, vvmint b) {
	ll N = a.size();
	vvmint ret(N, vmint(N, 0));
	rep(i, N)rep(j, N)rep(k, N) {
		ret[i][j] += a[i][k] * b[k][j];
	}
	return ret;
}

vvmint pw(vvmint a, ll K) {
	ll N = a.size();
	if(K==0){
		vvmint ret(N, vmint(N, 0));
		rep(i, N)ret[i][i] = 1;
		return ret;
	}

	vvmint ret = pw(a, K / 2);
	ret = prod(ret, ret);
	if (K % 2 == 1)ret = prod(ret, a);

	return ret;
}

bool cad(ll a, ll b, ll c) {
	if (a == b || a == c || b == c) { return false; }
	if (max(a, max(b, c)) == b || min(a, min(b, c)) == b) { return true; }
	return false;
}

int main() {
	ll N, K;
	cin >> N >> K;
	
	ll M = K * K;
	vvmint mat(2*M, vmint(2*M, 0));



	rep(i, K )rep(j, K)rep(k, K) {
		if (cad(i, j, k)) {
			//cout << i << j << k << endl;
			ll x = i * K + j;
			ll y = j * K + k;
			//cout << x << y << endl;
			mat[x + M][y + M] = 1;
			mat[x+M][y] = k;
			mat[x][y] = 1;
		}
	}

	mat = pw(mat, N - 2);
	
	//rep(i, 2*M) {
	//	rep(j, 2*M)cout << mat[i][j].val() << " ";
	//	cout << endl;
	//}

	vmint val(2 * M);
	rep(i, K)rep(j, K) {
		if (i != j) {
			ll x = i * K + j;
			val [x] = 1;
			val[x + M] = i + j;
		}
	}

	mint ans1 = 0, ans2 = 0;

	rep(i, 2*M)rep(j, 2*M) {
		if (i < M)ans1 += mat[i][j] * val[j];
		else { ans2 += mat[i][j] * val[j]; }
	}


	cout << ans1.val() << " " << ans2.val() << endl;


	
}
0