QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#767128#9479. And DNApurplevineAC ✓1ms4104kbC++201.1kb2024-11-20 19:53:412024-11-20 19:53:41

Judging History

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

  • [2024-11-20 19:53:41]
  • 评测
  • 测评结果:AC
  • 用时:1ms
  • 内存:4104kb
  • [2024-11-20 19:53:41]
  • 提交

answer

#include <bits/stdc++.h>

const int mod = 998244353;

struct Mat {
  int a[3][3];
  Mat() { memset(a, 0, sizeof(a)); }
  Mat operator * (const Mat &t) const {
    Mat ans;
    for (int i : {0, 1, 2}) for (int j : {0, 1, 2}) for (int k : {0, 1, 2})
      (ans.a[i][j] += 1ll * a[i][k] * t.a[k][j] % mod) %= mod;
    return ans;
  }
};

int main() {
  int n, m; scanf("%d %d", &n, &m);
  Mat s, t;
  for (int i = 0; i < 3; i++) s.a[i][i] = 1;
  t.a[0][1] = t.a[0][2] = t.a[1][0] = t.a[2][1] = 1;
  for (int k = n - 2; k; k >>= 1) {
    if (k & 1) s = s * t;
    t = t * t;
  }
  int res = ((long long) s.a[0][0] + s.a[0][2] + s.a[1][0] + s.a[1][1] + s.a[2][1]) % mod;
  // printf("%d\n", res);
  std::vector<std::vector<int>> f(31, std::vector<int>(2));
  if (m & 1) f[0][0] = res;
  else f[0][0] = f[0][1] = 1;
  for (int i = 1; i <= 30; i++) {
    if (m & (1 << i)) {
      f[i][0] = (1ll * res * f[i - 1][0] % mod + f[i - 1][1]) % mod;
      f[i][1] = f[i - 1][1];
    } else {
      f[i][0] = f[i - 1][0];
      f[i][1] = (1ll * res * f[i - 1][1] + f[i - 1][0]) % mod;
    }
  }
  printf("%d\n", f[30][0]);
} 

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

3 2

output:

4

result:

ok "4"

Test #2:

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

input:

3 0

output:

1

result:

ok "1"

Test #3:

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

input:

100 100

output:

343406454

result:

ok "343406454"

Test #4:

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

input:

686579306 119540831

output:

260602195

result:

ok "260602195"

Test #5:

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

input:

26855095 796233790

output:

492736984

result:

ok "492736984"

Test #6:

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

input:

295310488 262950628

output:

503008241

result:

ok "503008241"

Test #7:

score: 0
Accepted
time: 1ms
memory: 3840kb

input:

239670714 149827706

output:

994702969

result:

ok "994702969"

Test #8:

score: 0
Accepted
time: 1ms
memory: 3872kb

input:

790779949 110053353

output:

898252188

result:

ok "898252188"

Test #9:

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

input:

726600542 795285932

output:

183726777

result:

ok "183726777"

Test #10:

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

input:

957970519 585582861

output:

298814018

result:

ok "298814018"

Test #11:

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

input:

93349859 634036506

output:

110693443

result:

ok "110693443"

Test #12:

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

input:

453035113 34126396

output:

306244220

result:

ok "306244220"

Test #13:

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

input:

31994526 100604502

output:

930620701

result:

ok "930620701"

Test #14:

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

input:

234760741 249817734

output:

440195858

result:

ok "440195858"

Test #15:

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

input:

542621111 646412689

output:

207377992

result:

ok "207377992"

Test #16:

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

input:

28492783 602632297

output:

65234506

result:

ok "65234506"

Test #17:

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

input:

213500301 768820204

output:

540205113

result:

ok "540205113"

Test #18:

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

input:

697808101 753041955

output:

93561295

result:

ok "93561295"

Test #19:

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

input:

585126464 450455977

output:

91109717

result:

ok "91109717"

Test #20:

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

input:

236696315 482334538

output:

925575199

result:

ok "925575199"

Test #21:

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

input:

632719214 298704996

output:

358926097

result:

ok "358926097"

Test #22:

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

input:

869119333 933404114

output:

318501108

result:

ok "318501108"

Test #23:

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

input:

6977994 814763202

output:

585332987

result:

ok "585332987"

Test #24:

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

input:

3 1

output:

3

result:

ok "3"

Test #25:

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

input:

3 1000000000

output:

279679226

result:

ok "279679226"

Test #26:

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

input:

4 0

output:

1

result:

ok "1"

Test #27:

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

input:

4 1

output:

2

result:

ok "2"

Test #28:

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

input:

4 1000000000

output:

1755648

result:

ok "1755648"

Test #29:

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

input:

1000000000 0

output:

1

result:

ok "1"

Test #30:

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

input:

1000000000 1

output:

696798313

result:

ok "696798313"

Test #31:

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

input:

1000000000 1000000000

output:

703786301

result:

ok "703786301"

Extra Test:

score: 0
Extra Test Passed