QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#659274#9479. And DNAucup-team3691#AC ✓0ms3856kbC++202.4kb2024-10-19 19:27:072024-10-19 19:27:09

Judging History

你现在查看的是最新测评结果

  • [2024-10-19 19:27:09]
  • 评测
  • 测评结果:AC
  • 用时:0ms
  • 内存:3856kb
  • [2024-10-19 19:27:07]
  • 提交

answer

#include <iostream>
#include <queue>
#include <stack>
#include <vector>
#include <deque>
#include <set>
#include <map>
#include <unordered_map>
#include <cassert>
#include <algorithm>
#include <cmath>
#include <random>
#include <ctime>
#include <cstdlib>
#include <chrono>

using namespace std;

const int MOD = 998244353;

struct matrice {
  int n, m;
  int a[4][4];

  matrice(int n_, int m_) : n(n_), m(m_) {
    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < m; ++j)
        a[i][j] = 0;
    }
  }

  int* operator [] (int i) {
    return a[i];
  }

  matrice& operator *= (matrice &rhs) {
    matrice tmp(n, rhs.m);
    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < m; ++j) {
        for (int k = 0; k < rhs.m; ++k) {
          tmp[i][k] = (tmp[i][k] + 1LL * a[i][j] * rhs[j][k]) % MOD;
        }
      }
    }

    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < m; ++j) {
        a[i][j] = tmp[i][j];
      }
    }

    n = tmp.n;
    m = tmp.m;

    return *this;
  }
};

int ct;
map<int, int> dp;

bool is_ok(int x) {
  if ((x & 1) && (x & 4) && (x & 2))
    return false;

  if ((x & 1) && (x & 4))
    return true;

  if (x & 2)
    return true;

  return false;
}

int get_ct(int n) {
  vector<matrice> dp;
  for (int i = 0; i < 4; ++i) {
    dp.emplace_back(matrice(1, 4));
    dp[i][0][i] = 1;
  }

  matrice t(4, 4);

  t[1][2] = 1;
  t[1][3] = 1;
  t[2][1] = 1;
  t[3][2] = 1;

  n -= 2;
  for (int p = 1; p <= n; p <<= 1) {
    if (p & n) {
      for (int i = 0; i < 4; ++i)
        dp[i] *= t;
    }
    t *= t;
  }

  int ans = 0;
  for (int i = 0; i < 4; ++i) {
    for (int j = 0; j < 4; ++j) {
      if (is_ok((i) | ((j & 1) << 2)) && is_ok(((i & 2) >> 1) | (j << 1))) {
        ans = (ans + dp[i][0][j]) % MOD;
      }
    }
  }

  return ans;
}

int solve(int m) {
  if (m < 0)
    return 0;
  if (m == 0)
    return 1;

  if (dp.count(m)) {
    return dp[m];
  }

  if (m % 2 == 0) {
    return dp[m] = (solve(m / 2) + solve(m / 2 - 1)) % MOD;
  }

  return dp[m] = 1LL * solve(m / 2) * ct % MOD;
}

void solve() {
  int n, m;
  cin >> n >> m;

  ct = get_ct(n);

  cout << solve(m) << "\n";
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(NULL);

  int t = 1;
  while (t--)
    solve();

  return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3824kb

input:

3 2

output:

4

result:

ok "4"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3576kb

input:

3 0

output:

1

result:

ok "1"

Test #3:

score: 0
Accepted
time: 0ms
memory: 3776kb

input:

100 100

output:

343406454

result:

ok "343406454"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3604kb

input:

686579306 119540831

output:

260602195

result:

ok "260602195"

Test #5:

score: 0
Accepted
time: 0ms
memory: 3568kb

input:

26855095 796233790

output:

492736984

result:

ok "492736984"

Test #6:

score: 0
Accepted
time: 0ms
memory: 3576kb

input:

295310488 262950628

output:

503008241

result:

ok "503008241"

Test #7:

score: 0
Accepted
time: 0ms
memory: 3852kb

input:

239670714 149827706

output:

994702969

result:

ok "994702969"

Test #8:

score: 0
Accepted
time: 0ms
memory: 3624kb

input:

790779949 110053353

output:

898252188

result:

ok "898252188"

Test #9:

score: 0
Accepted
time: 0ms
memory: 3568kb

input:

726600542 795285932

output:

183726777

result:

ok "183726777"

Test #10:

score: 0
Accepted
time: 0ms
memory: 3512kb

input:

957970519 585582861

output:

298814018

result:

ok "298814018"

Test #11:

score: 0
Accepted
time: 0ms
memory: 3780kb

input:

93349859 634036506

output:

110693443

result:

ok "110693443"

Test #12:

score: 0
Accepted
time: 0ms
memory: 3824kb

input:

453035113 34126396

output:

306244220

result:

ok "306244220"

Test #13:

score: 0
Accepted
time: 0ms
memory: 3604kb

input:

31994526 100604502

output:

930620701

result:

ok "930620701"

Test #14:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

234760741 249817734

output:

440195858

result:

ok "440195858"

Test #15:

score: 0
Accepted
time: 0ms
memory: 3780kb

input:

542621111 646412689

output:

207377992

result:

ok "207377992"

Test #16:

score: 0
Accepted
time: 0ms
memory: 3856kb

input:

28492783 602632297

output:

65234506

result:

ok "65234506"

Test #17:

score: 0
Accepted
time: 0ms
memory: 3600kb

input:

213500301 768820204

output:

540205113

result:

ok "540205113"

Test #18:

score: 0
Accepted
time: 0ms
memory: 3572kb

input:

697808101 753041955

output:

93561295

result:

ok "93561295"

Test #19:

score: 0
Accepted
time: 0ms
memory: 3572kb

input:

585126464 450455977

output:

91109717

result:

ok "91109717"

Test #20:

score: 0
Accepted
time: 0ms
memory: 3856kb

input:

236696315 482334538

output:

925575199

result:

ok "925575199"

Test #21:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

632719214 298704996

output:

358926097

result:

ok "358926097"

Test #22:

score: 0
Accepted
time: 0ms
memory: 3856kb

input:

869119333 933404114

output:

318501108

result:

ok "318501108"

Test #23:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

6977994 814763202

output:

585332987

result:

ok "585332987"

Test #24:

score: 0
Accepted
time: 0ms
memory: 3844kb

input:

3 1

output:

3

result:

ok "3"

Test #25:

score: 0
Accepted
time: 0ms
memory: 3580kb

input:

3 1000000000

output:

279679226

result:

ok "279679226"

Test #26:

score: 0
Accepted
time: 0ms
memory: 3852kb

input:

4 0

output:

1

result:

ok "1"

Test #27:

score: 0
Accepted
time: 0ms
memory: 3852kb

input:

4 1

output:

2

result:

ok "2"

Test #28:

score: 0
Accepted
time: 0ms
memory: 3852kb

input:

4 1000000000

output:

1755648

result:

ok "1755648"

Test #29:

score: 0
Accepted
time: 0ms
memory: 3624kb

input:

1000000000 0

output:

1

result:

ok "1"

Test #30:

score: 0
Accepted
time: 0ms
memory: 3520kb

input:

1000000000 1

output:

696798313

result:

ok "696798313"

Test #31:

score: 0
Accepted
time: 0ms
memory: 3608kb

input:

1000000000 1000000000

output:

703786301

result:

ok "703786301"

Extra Test:

score: 0
Extra Test Passed