結果
問題 | No.1516 simple 門松列 problem Re:MASTER |
ユーザー | ayoiyouyoeyo |
提出日時 | 2021-05-21 21:55:08 |
言語 | C++14 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 813 ms / 6,000 ms |
コード長 | 2,908 bytes |
コンパイル時間 | 3,821 ms |
コンパイル使用メモリ | 174,412 KB |
実行使用メモリ | 7,040 KB |
最終ジャッジ日時 | 2024-04-18 15:15:18 |
合計ジャッジ時間 | 7,655 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
6,812 KB |
testcase_01 | AC | 58 ms
6,940 KB |
testcase_02 | AC | 348 ms
6,940 KB |
testcase_03 | AC | 3 ms
6,940 KB |
testcase_04 | AC | 2 ms
6,940 KB |
testcase_05 | AC | 3 ms
6,944 KB |
testcase_06 | AC | 15 ms
6,940 KB |
testcase_07 | AC | 58 ms
6,940 KB |
testcase_08 | AC | 85 ms
6,940 KB |
testcase_09 | AC | 2 ms
6,944 KB |
testcase_10 | AC | 18 ms
6,940 KB |
testcase_11 | AC | 3 ms
6,940 KB |
testcase_12 | AC | 2 ms
6,944 KB |
testcase_13 | AC | 2 ms
6,940 KB |
testcase_14 | AC | 700 ms
7,040 KB |
testcase_15 | AC | 326 ms
6,940 KB |
testcase_16 | AC | 155 ms
6,940 KB |
testcase_17 | AC | 56 ms
6,940 KB |
testcase_18 | AC | 15 ms
6,940 KB |
testcase_19 | AC | 7 ms
6,940 KB |
testcase_20 | AC | 811 ms
7,040 KB |
testcase_21 | AC | 813 ms
6,940 KB |
ソースコード
#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; }