QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#687106#9479. And DNAreal_sigma_team#AC ✓1ms3848kbC++232.3kb2024-10-29 17:17:112024-10-29 17:17:11

Judging History

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

  • [2024-10-29 17:17:11]
  • 评测
  • 测评结果:AC
  • 用时:1ms
  • 内存:3848kb
  • [2024-10-29 17:17:11]
  • 提交

answer

#include<bits/stdc++.h>

const int mod = 998244353;

int add(int a, int b) {
	return a + b >= mod ? a + b - mod : a + b;
} 
int sub(int a, int b) {
	return a >= b ? a - b : a + mod - b;
}
int mul(int a, int b) {
	return 1ll * a * b % mod;
}
int power(int a, int b) {
	int res = 1;
	for (; b; b >>= 1, a = mul(a, a)) if (b & 1) res = mul(res, a);
		return res;
}
int inv(int x) {
	return x == 1 ? 1 : sub(0, mul(mod / x, inv(mod % x)));
}

std::vector<std::vector<int>> mul(std::vector<std::vector<int>> a,
									std::vector<std::vector<int>> b) {
	int n = a.size();
	int m = a[0].size();
	int k = b[0].size();
	std::vector c(n, std::vector(k, 0));
	for (int i = 0; i < n; ++i) {
		for (int j = 0; j < m; ++j) {
			for (int l = 0; l < k; ++l) {
				c[i][l] = add(c[i][l], mul(a[i][j], b[j][l]));
			}
		}
	}
	return c;
}

std::vector<std::vector<int>> power(std::vector<std::vector<int>> a, int b) {
	int n = a.size();
	std::vector ret(n, std::vector(n, 0));
	for (int i = 0; i < n;  ++i) ret[i][i] = 1;
	for (; b; b >>= 1, a = mul(a, a)) if (b & 1) ret = mul(ret, a);
	return ret;
}

int main() {
	std::cin.tie(nullptr)->sync_with_stdio(false);

	int n, m;
	std::cin >> n >> m;
	std::vector cal(4, std::vector(4, 0));
	for (int i = 0; i < 4; ++i) {
		for (int z = 0; z < 2; ++z) {
			int x = i & 1, y = i >> 1;
			if ((x & z) + y == 1) {
				cal[i][i / 2 + z * 2] = 1;
			}
		}
	}
	int shusha = 0;
	for (int c = 0; c < 4; ++c) {
		std::vector sheesh(1, std::vector(4, 0));
		sheesh[0][c] = 1;
		sheesh = mul(sheesh, power(cal, n - 2));
		for (int i = 0; i < 4; ++i) {
			int x = i & 1, y = i >> 1, z = c & 1, l = c >> 1;
			if ((x & z) + y == 1 && (y & l) + z == 1) shusha = add(shusha, sheesh[0][i]);
		}
	}
	std::array<int, 2> dp;
	dp[0] = 1;
	dp[1] = 0;
	for (int i = 0; i <= 30; ++i) {
		std::array<int, 2> nw = {0, 0};
		for (int x = 0; x < 2; ++x) {
			int c = (m >> i & 1) - x;
			for (int q = 0; q <= 2; ++q) {
				int v;
				if (q == 1) v = shusha;
				else v = 1;
				int y = c - q;
				// std::cout << i << ' ' << x << ' ' << q << ' ' << y << ' ' << v << ' ' << dp[x] << std::endl;
				if (y % 2 != 0) continue;
				y = -y / 2;
				nw[y] = add(nw[y], mul(v, dp[x]));
			}
		}
		dp = nw;
		// std::cout << dp[0] << ' ' << dp[1] << std::endl;
	}
	std::cout << dp[0];
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3844kb

input:

3 2

output:

4

result:

ok "4"

Test #2:

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

input:

3 0

output:

1

result:

ok "1"

Test #3:

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

input:

100 100

output:

343406454

result:

ok "343406454"

Test #4:

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

input:

686579306 119540831

output:

260602195

result:

ok "260602195"

Test #5:

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

input:

26855095 796233790

output:

492736984

result:

ok "492736984"

Test #6:

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

input:

295310488 262950628

output:

503008241

result:

ok "503008241"

Test #7:

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

input:

239670714 149827706

output:

994702969

result:

ok "994702969"

Test #8:

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

input:

790779949 110053353

output:

898252188

result:

ok "898252188"

Test #9:

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

input:

726600542 795285932

output:

183726777

result:

ok "183726777"

Test #10:

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

input:

957970519 585582861

output:

298814018

result:

ok "298814018"

Test #11:

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

input:

93349859 634036506

output:

110693443

result:

ok "110693443"

Test #12:

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

input:

453035113 34126396

output:

306244220

result:

ok "306244220"

Test #13:

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

input:

31994526 100604502

output:

930620701

result:

ok "930620701"

Test #14:

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

input:

234760741 249817734

output:

440195858

result:

ok "440195858"

Test #15:

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

input:

542621111 646412689

output:

207377992

result:

ok "207377992"

Test #16:

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

input:

28492783 602632297

output:

65234506

result:

ok "65234506"

Test #17:

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

input:

213500301 768820204

output:

540205113

result:

ok "540205113"

Test #18:

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

input:

697808101 753041955

output:

93561295

result:

ok "93561295"

Test #19:

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

input:

585126464 450455977

output:

91109717

result:

ok "91109717"

Test #20:

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

input:

236696315 482334538

output:

925575199

result:

ok "925575199"

Test #21:

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

input:

632719214 298704996

output:

358926097

result:

ok "358926097"

Test #22:

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

input:

869119333 933404114

output:

318501108

result:

ok "318501108"

Test #23:

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

input:

6977994 814763202

output:

585332987

result:

ok "585332987"

Test #24:

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

input:

3 1

output:

3

result:

ok "3"

Test #25:

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

input:

3 1000000000

output:

279679226

result:

ok "279679226"

Test #26:

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

input:

4 0

output:

1

result:

ok "1"

Test #27:

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

input:

4 1

output:

2

result:

ok "2"

Test #28:

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

input:

4 1000000000

output:

1755648

result:

ok "1755648"

Test #29:

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

input:

1000000000 0

output:

1

result:

ok "1"

Test #30:

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

input:

1000000000 1

output:

696798313

result:

ok "696798313"

Test #31:

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

input:

1000000000 1000000000

output:

703786301

result:

ok "703786301"

Extra Test:

score: 0
Extra Test Passed