QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#326165#4812. Counting SequenceTokido_SayaAC ✓4901ms205004kbC++143.0kb2024-02-12 14:17:532024-02-12 14:17:53

Judging History

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

  • [2024-02-12 14:17:53]
  • 评测
  • 测评结果:AC
  • 用时:4901ms
  • 内存:205004kb
  • [2024-02-12 14:17:53]
  • 提交

answer

// Sea, Your Memory
#include<bits/stdc++.h>
#define LL long long
#define DB double
#define MOD 998244353
#define ls(x) x << 1
#define rs(x) x << 1 | 1
#define lowbit(x) x & (-x)
#define PII pair<int, int>
#define MP make_pair
#define VI vector<int>
#define VII vector<int>::iterator
#define all(x) x.begin(), x.end()
#define EB emplace_back
#define SI set<int>
#define SII set<int>::iterator
#define QI queue<int>
#define fi first
#define se second
using namespace std;
template<typename T> void chkmn(T &a, const T &b) { (a > b) && (a = b); }
template<typename T> void chkmx(T &a, const T &b) { (a < b) && (a = b); }
int inc(const int &a, const int &b) { return a + b >= MOD ? a + b - MOD : a + b; }
int dec(const int &a, const int &b) { return a - b < 0 ? a - b + MOD : a - b; }
int mul(const int &a, const int &b) { return 1LL * a * b % MOD; }
int sqr(const int &a) { return 1LL * a * a % MOD; }
void Inc(int &a, const int &b) { ((a += b) >= MOD) && (a -= MOD); }
void Dec(int &a, const int &b) { ((a -= b) < 0) && (a += MOD); }
void Mul(int &a, const int &b) { a = 1LL * a * b % MOD; }
void Sqr(int &a) { a = 1LL * a * a % MOD; }
int qwqmi(int x, int k = MOD - 2)
{
    int res = 1;
    while(k)
    {
        if(k & 1) Mul(res, x);
        Sqr(x), k >>= 1;
    }
    return res;
}
template<typename T> void read(T &x)
{
	x = 0;
	int f = 1;
	char ch = getchar();
	while(!isdigit(ch))
	{
		if(ch == '-')
			f = -1;
		ch = getchar();
	}
	while(isdigit(ch))
	{
		x = (x << 1) + (x << 3) + (ch ^ 48);
		ch = getchar();
	}
	x = x * f;
}
const int N = 3e5 + 5;
const int B = 1933;
const int B2 = 2 * B + 5;
const int _B = N / B + 10;
int n, C;
int f[B2][B2];
int Solve1()
{
	for(int i = 1; i < B; ++i)
		f[i][i] = 1;
	for(int s = 1; s <= n; ++s)
	{
		for(int x = 1; x <= 2 * B && s - x > 0; ++x)
		{
			int t = s - x, lst = t % B2, now = s % B2;
			f[now][x] = inc(mul(C, f[lst][x + 1]), f[lst][x - 1]);
		}
	}
	int res = 0;
	for(int x = 1; x <= 2 * B; ++x)
		Inc(res, f[n % B2][x]);
//	cerr << "res1 : " << res << '\n';
	return res;
}
int g[2][_B * _B * 2 + 5][_B * 2 + 5];
int Solve2()
{
	int now = 1, lst = 0;
	g[1][_B * _B][_B] = 1;
	int res = (n >= B);
	for(int i = 2; i <= _B; ++i)
	{
		swap(now, lst);
		memset(g[now], 0, sizeof(g[now]));
		for(int j = -(i - 1) * (i - 2) / 2; j <= (i - 1) * (i - 2) / 2; ++j)
			for(int k = 2 - i; k <= i - 2; ++k)
			{
				if(!g[lst][j + _B * _B][k + _B]) continue;
				Inc(g[now][j + k + 1 + _B * _B][k + 1 + _B], g[lst][j + _B * _B][k + _B]);
				Inc(g[now][j + k - 1 + _B * _B][k - 1 + _B], mul(C, g[lst][j + _B * _B][k + _B]));
			}
		for(int j = -i * (i - 1) / 2; j <= i * (i - 1) / 2; ++j)
			if((n - j) % i == 0 && (n - j) / i >= B) 
				for(int k = 1 - i; k <= i - 1; ++k)
					Inc(res, g[now][j + _B * _B][k + _B]);
	}
//	cerr << "res2 : " << res << '\n';
	return res;
}
int main()
{
	read(n), read(C);
	int ans = inc(Solve1(), Solve2());
	printf("%d\n", ans);
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1238ms
memory: 177004kb

input:

5 3

output:

8

result:

ok 1 number(s): "8"

Test #2:

score: 0
Accepted
time: 1112ms
memory: 176056kb

input:

1 0

output:

1

result:

ok 1 number(s): "1"

Test #3:

score: 0
Accepted
time: 1220ms
memory: 177204kb

input:

2022 39

output:

273239559

result:

ok 1 number(s): "273239559"

Test #4:

score: 0
Accepted
time: 1238ms
memory: 177604kb

input:

1 998244352

output:

1

result:

ok 1 number(s): "1"

Test #5:

score: 0
Accepted
time: 1218ms
memory: 175636kb

input:

1 12345678

output:

1

result:

ok 1 number(s): "1"

Test #6:

score: 0
Accepted
time: 1226ms
memory: 175716kb

input:

20 998998

output:

643731701

result:

ok 1 number(s): "643731701"

Test #7:

score: 0
Accepted
time: 1224ms
memory: 176804kb

input:

23 123

output:

947753998

result:

ok 1 number(s): "947753998"

Test #8:

score: 0
Accepted
time: 1198ms
memory: 175908kb

input:

50 5555

output:

745339864

result:

ok 1 number(s): "745339864"

Test #9:

score: 0
Accepted
time: 1219ms
memory: 177440kb

input:

60 6666

output:

690992218

result:

ok 1 number(s): "690992218"

Test #10:

score: 0
Accepted
time: 1205ms
memory: 175776kb

input:

100 50

output:

169678588

result:

ok 1 number(s): "169678588"

Test #11:

score: 0
Accepted
time: 1192ms
memory: 177180kb

input:

500 88888

output:

216149701

result:

ok 1 number(s): "216149701"

Test #12:

score: 0
Accepted
time: 1223ms
memory: 176556kb

input:

1000 213456

output:

270989457

result:

ok 1 number(s): "270989457"

Test #13:

score: 0
Accepted
time: 1221ms
memory: 178560kb

input:

2000 119988

output:

756425375

result:

ok 1 number(s): "756425375"

Test #14:

score: 0
Accepted
time: 1270ms
memory: 192880kb

input:

3000 998244352

output:

71841227

result:

ok 1 number(s): "71841227"

Test #15:

score: 0
Accepted
time: 1249ms
memory: 191852kb

input:

3000 555555555

output:

79880116

result:

ok 1 number(s): "79880116"

Test #16:

score: 0
Accepted
time: 1265ms
memory: 204880kb

input:

4321 1234

output:

949603993

result:

ok 1 number(s): "949603993"

Test #17:

score: 0
Accepted
time: 1153ms
memory: 204896kb

input:

5000 0

output:

5

result:

ok 1 number(s): "5"

Test #18:

score: 0
Accepted
time: 1291ms
memory: 204892kb

input:

5000 88888777

output:

833064960

result:

ok 1 number(s): "833064960"

Test #19:

score: 0
Accepted
time: 1243ms
memory: 204820kb

input:

5000 35557777

output:

696388498

result:

ok 1 number(s): "696388498"

Test #20:

score: 0
Accepted
time: 1396ms
memory: 204936kb

input:

10000 123456

output:

434296902

result:

ok 1 number(s): "434296902"

Test #21:

score: 0
Accepted
time: 1455ms
memory: 204812kb

input:

20000 555555

output:

34806915

result:

ok 1 number(s): "34806915"

Test #22:

score: 0
Accepted
time: 1561ms
memory: 205004kb

input:

30000 777888999

output:

58443551

result:

ok 1 number(s): "58443551"

Test #23:

score: 0
Accepted
time: 1783ms
memory: 204900kb

input:

50000 2

output:

90102905

result:

ok 1 number(s): "90102905"

Test #24:

score: 0
Accepted
time: 2036ms
memory: 204880kb

input:

70000 77998866

output:

202638568

result:

ok 1 number(s): "202638568"

Test #25:

score: 0
Accepted
time: 2384ms
memory: 204864kb

input:

100000 998244352

output:

360520717

result:

ok 1 number(s): "360520717"

Test #26:

score: 0
Accepted
time: 2412ms
memory: 204896kb

input:

100000 555555555

output:

613886009

result:

ok 1 number(s): "613886009"

Test #27:

score: 0
Accepted
time: 2997ms
memory: 204944kb

input:

150000 233333

output:

381065878

result:

ok 1 number(s): "381065878"

Test #28:

score: 0
Accepted
time: 2999ms
memory: 204888kb

input:

150000 20050117

output:

269891864

result:

ok 1 number(s): "269891864"

Test #29:

score: 0
Accepted
time: 3599ms
memory: 204820kb

input:

200000 114514

output:

262861613

result:

ok 1 number(s): "262861613"

Test #30:

score: 0
Accepted
time: 3651ms
memory: 205000kb

input:

200000 1919810

output:

77872388

result:

ok 1 number(s): "77872388"

Test #31:

score: 0
Accepted
time: 4721ms
memory: 204816kb

input:

300000 0

output:

12

result:

ok 1 number(s): "12"

Test #32:

score: 0
Accepted
time: 4791ms
memory: 204904kb

input:

300000 1

output:

298803948

result:

ok 1 number(s): "298803948"

Test #33:

score: 0
Accepted
time: 4812ms
memory: 204872kb

input:

300000 2

output:

106751203

result:

ok 1 number(s): "106751203"

Test #34:

score: 0
Accepted
time: 4846ms
memory: 204900kb

input:

300000 1234

output:

427045479

result:

ok 1 number(s): "427045479"

Test #35:

score: 0
Accepted
time: 4847ms
memory: 204812kb

input:

300000 2345

output:

441593553

result:

ok 1 number(s): "441593553"

Test #36:

score: 0
Accepted
time: 4860ms
memory: 204892kb

input:

300000 20041115

output:

580367993

result:

ok 1 number(s): "580367993"

Test #37:

score: 0
Accepted
time: 4833ms
memory: 204940kb

input:

300000 20050117

output:

579859619

result:

ok 1 number(s): "579859619"

Test #38:

score: 0
Accepted
time: 4819ms
memory: 204884kb

input:

300000 22223333

output:

596066085

result:

ok 1 number(s): "596066085"

Test #39:

score: 0
Accepted
time: 4799ms
memory: 204940kb

input:

300000 175846372

output:

660364393

result:

ok 1 number(s): "660364393"

Test #40:

score: 0
Accepted
time: 4836ms
memory: 204936kb

input:

299999 9999999

output:

954865020

result:

ok 1 number(s): "954865020"

Test #41:

score: 0
Accepted
time: 4901ms
memory: 204892kb

input:

299998 55556666

output:

904862432

result:

ok 1 number(s): "904862432"

Test #42:

score: 0
Accepted
time: 3494ms
memory: 204816kb

input:

190733 31756136

output:

880544587

result:

ok 1 number(s): "880544587"

Test #43:

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

input:

198497 463488298

output:

185220207

result:

ok 1 number(s): "185220207"

Test #44:

score: 0
Accepted
time: 4709ms
memory: 204820kb

input:

299997 0

output:

16

result:

ok 1 number(s): "16"

Test #45:

score: 0
Accepted
time: 3186ms
memory: 204936kb

input:

168168 296157813

output:

798716760

result:

ok 1 number(s): "798716760"

Test #46:

score: 0
Accepted
time: 2566ms
memory: 204892kb

input:

114514 1919810

output:

783513290

result:

ok 1 number(s): "783513290"