結果

問題 No.840 ほむほむほむら
コンテスト
ユーザー ふっぴー
提出日時 2019-06-14 23:05:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 249 ms / 4,000 ms
コード長 3,459 bytes
コンパイル時間 1,693 ms
コンパイル使用メモリ 175,492 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-14 13:04:39
合計ジャッジ時間 3,949 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <unistd.h>

using namespace std;

#define DEBUG(x) cerr<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cerr<<#v<<":";for(int i=0;i<v.size();i++) cerr<<" "<<v[i]; cerr<<endl
#define DEBUG_MAT(v) cerr<<#v<<endl;for(int i=0;i<v.size();i++){for(int j=0;j<v[i].size();j++) {cerr<<v[i][j]<<" ";}cerr<<endl;}


typedef long long ll;
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
#define pll pair<ll,ll>
template<class S, class T> pair<S, T> operator+(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first + t.first, s.second + t.second); }
template<class S, class T> pair<S, T> operator-(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first - t.first, s.second - t.second); }
template<class S, class T> ostream& operator<<(ostream& os, pair<S, T> p) { os << "(" << p.first << ", " << p.second << ")"; return os; }
#define X first
#define Y second
#define rep(i,n) for(int i=0;i<(n);i++)
#define rep1(i,n) for(int i=1;i<=(n);i++)
#define rrep(i,n) for(int i=(n)-1;i>=0;i--)
#define rrep1(i,n) for(int i=(n);i>0;i--)
#define REP(i,a,b) for(int i=a;i<b;i++)
#define in(x, a, b) (a <= x && x < b)
#define all(c) c.begin(),c.end()
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 (a>b) { a = b; return 1; } return 0; }
#define UNIQUE(v) v.erase(std::unique(v.begin(), v.end()), v.end());
const ll inf = 1000000001;
const ll INF = (ll)1e18 + 1;
//const ll MOD = 1000000007;
const ll MOD = 998244353;
const double pi = 3.14159265358979323846;
#define Sp(p) cout<<setprecision(15)<< fixed<<p<<endl;
int dx[4] = { -1,0, 1,0 }, dy[4] = { 0,1,0,-1 };
int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };
#define fio() cin.tie(0); ios::sync_with_stdio(false);
// #define mp make_pair
//#define endl '\n'


int n, k;
void u2abc(int u, int &a, int &b, int &c) {
  a = u / (k*k);
  u %= k*k;
  b = u / k;
  u %= k;
  c = u;
}

int abc2u(int a, int b, int c) {
  return a * k * k + b * k + c;
}

typedef vector<ll> vec;
typedef vector<vec> mat;

mat mul(mat &A, mat &B) {
	mat C(A.size(), vec(B[0].size()));
	for (int i = 0; i < A.size(); i++) {
		for (int k = 0; k < B.size(); k++) {
			for (int j = 0; j < B[0].size(); j++) {
              C[i][j] = (C[i][j] + A[i][k] * B[k][j] % MOD) % MOD;
			}
		}
	}
	return C;
}

vec mul(mat &A, vec &B) {
	vec C(A.size());
	for (int i = 0; i < A.size(); i++) {
		for (int j = 0; j < B.size(); j++) {
          (C[i] += A[i][j] * B[j] % MOD) % MOD;
		}
	}
	return C;
}

mat pow(mat A, ll n) {
	mat B(A.size(), vec(A.size()));
	for (int i = 0; i < A.size(); i++) {
		B[i][i] = 1;
	}
	while (n > 0) {
		if (n & 1)B = mul(B, A);
		A = mul(A, A);
		n >>= 1;
	}
	return B;
}


int main() {
  cin >> n >> k;

  int m = k * k * k;

  vll A(m, vl(m));

  
  rep (a, k) rep (b, k) rep (c, k) {
    int u = abc2u(a, b, c);
    int x = abc2u((k + a - 1) % k, b, c);
    int y = abc2u(a, (k + b - a) % k, c);
    int z = abc2u(a, b, (k + c - b) % k);
    A[u][x]++;
    A[u][y]++;
    A[u][z]++;
  }

  vec b(m);
  b[0] = 1;

  mat An = pow(A, n);
  vec bn = mul(An, b);
  
  ll ans = 0;
  rep (a, k) rep (b, k) {
    int u = abc2u(a, b, 0);
    ans += bn[u];
    ans %= MOD;
  }
  cout << ans << endl;
}
0