QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#262185#4812. Counting SequencezhouhuanyiAC ✓2451ms20132kbC++141.4kb2023-11-23 16:26:592023-11-23 16:27:00

Judging History

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

  • [2023-11-23 16:27:00]
  • 评测
  • 测评结果:AC
  • 用时:2451ms
  • 内存:20132kb
  • [2023-11-23 16:26:59]
  • 提交

answer

#include<iostream>
#include<cstdio>
#include<cmath>
#define M 2000
#define N 400000
#define mod 998244353
using namespace std;
int read()
{
    char c=0;
    int sum=0;
    while (c<'0'||c>'9') c=getchar();
    while ('0'<=c&&c<='9') sum=sum*10+c-'0',c=getchar();
    return sum;
}
int n,w,op,Base,Base2,F[M+1],dp[2][N+1],dp2[M+1][M+1],ans;
void Adder(int &x,int d)
{
    x+=d;
    if (x>=mod) x-=mod;
    return;
}
int main()
{
    int st;
    n=read(),w=read(),Base=sqrt(2*n),Base2=sqrt(2*n)+sqrt(n),dp[0][0]=1;
    //st>Base
    for (int i=0;i<=Base;++i) F[i]=(i*(i+1))>>1;
    for (int i=1;i<=Base;++i)
    {
	for (int j=Base+1;j<=n;++j)
	{
	    st=n-i*j+F[i-1];
	    if ((!(st&1))&&st>=0&&(st>>1)<=F[i-1]) Adder(ans,dp[op][st>>1]);
	}
	op^=1;
	for (int j=0;j<=F[i];++j) dp[op][j]=0;
	for (int j=0;j<=F[i];++j) Adder(dp[op][j],1ll*dp[!op][j]*w%mod);
	for (int j=i;j<=F[i];++j) Adder(dp[op][j],dp[!op][j-i]);
    }
    //st<=Base
    op=0;
    for (int i=1;i<=n;++i)
    {
	op=(op+1)%(Base2+1);
	for (int j=1;j<=min(i,Base2);++j) dp2[op][j]=0;
	if (i<=Base) Adder(dp2[op][i],1);
	for (int j=2;j<=min(i,Base2);++j) Adder(dp2[op][j],dp2[(op+Base2+1-j)%(Base2+1)][j-1]);
	for (int j=1;j<=min(i,Base2-1);++j) Adder(dp2[op][j],1ll*dp2[(op+Base2+1-j)%(Base2+1)][j+1]*w%mod);
    }
    for (int i=1;i<=min(n,Base2);++i) Adder(ans,dp2[op][i]);
    printf("%d\n",ans);
    return 0;
}


Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

5 3

output:

8

result:

ok 1 number(s): "8"

Test #2:

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

input:

1 0

output:

1

result:

ok 1 number(s): "1"

Test #3:

score: 0
Accepted
time: 3ms
memory: 8104kb

input:

2022 39

output:

273239559

result:

ok 1 number(s): "273239559"

Test #4:

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

input:

1 998244352

output:

1

result:

ok 1 number(s): "1"

Test #5:

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

input:

1 12345678

output:

1

result:

ok 1 number(s): "1"

Test #6:

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

input:

20 998998

output:

643731701

result:

ok 1 number(s): "643731701"

Test #7:

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

input:

23 123

output:

947753998

result:

ok 1 number(s): "947753998"

Test #8:

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

input:

50 5555

output:

745339864

result:

ok 1 number(s): "745339864"

Test #9:

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

input:

60 6666

output:

690992218

result:

ok 1 number(s): "690992218"

Test #10:

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

input:

100 50

output:

169678588

result:

ok 1 number(s): "169678588"

Test #11:

score: 0
Accepted
time: 2ms
memory: 9740kb

input:

500 88888

output:

216149701

result:

ok 1 number(s): "216149701"

Test #12:

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

input:

1000 213456

output:

270989457

result:

ok 1 number(s): "270989457"

Test #13:

score: 0
Accepted
time: 3ms
memory: 8060kb

input:

2000 119988

output:

756425375

result:

ok 1 number(s): "756425375"

Test #14:

score: 0
Accepted
time: 2ms
memory: 8256kb

input:

3000 998244352

output:

71841227

result:

ok 1 number(s): "71841227"

Test #15:

score: 0
Accepted
time: 3ms
memory: 9884kb

input:

3000 555555555

output:

79880116

result:

ok 1 number(s): "79880116"

Test #16:

score: 0
Accepted
time: 5ms
memory: 8156kb

input:

4321 1234

output:

949603993

result:

ok 1 number(s): "949603993"

Test #17:

score: 0
Accepted
time: 7ms
memory: 8408kb

input:

5000 0

output:

5

result:

ok 1 number(s): "5"

Test #18:

score: 0
Accepted
time: 7ms
memory: 8460kb

input:

5000 88888777

output:

833064960

result:

ok 1 number(s): "833064960"

Test #19:

score: 0
Accepted
time: 3ms
memory: 8368kb

input:

5000 35557777

output:

696388498

result:

ok 1 number(s): "696388498"

Test #20:

score: 0
Accepted
time: 15ms
memory: 8312kb

input:

10000 123456

output:

434296902

result:

ok 1 number(s): "434296902"

Test #21:

score: 0
Accepted
time: 41ms
memory: 10856kb

input:

20000 555555

output:

34806915

result:

ok 1 number(s): "34806915"

Test #22:

score: 0
Accepted
time: 75ms
memory: 11252kb

input:

30000 777888999

output:

58443551

result:

ok 1 number(s): "58443551"

Test #23:

score: 0
Accepted
time: 151ms
memory: 10288kb

input:

50000 2

output:

90102905

result:

ok 1 number(s): "90102905"

Test #24:

score: 0
Accepted
time: 266ms
memory: 12644kb

input:

70000 77998866

output:

202638568

result:

ok 1 number(s): "202638568"

Test #25:

score: 0
Accepted
time: 449ms
memory: 12860kb

input:

100000 998244352

output:

360520717

result:

ok 1 number(s): "360520717"

Test #26:

score: 0
Accepted
time: 444ms
memory: 14180kb

input:

100000 555555555

output:

613886009

result:

ok 1 number(s): "613886009"

Test #27:

score: 0
Accepted
time: 842ms
memory: 15368kb

input:

150000 233333

output:

381065878

result:

ok 1 number(s): "381065878"

Test #28:

score: 0
Accepted
time: 833ms
memory: 15040kb

input:

150000 20050117

output:

269891864

result:

ok 1 number(s): "269891864"

Test #29:

score: 0
Accepted
time: 1268ms
memory: 16172kb

input:

200000 114514

output:

262861613

result:

ok 1 number(s): "262861613"

Test #30:

score: 0
Accepted
time: 1293ms
memory: 15344kb

input:

200000 1919810

output:

77872388

result:

ok 1 number(s): "77872388"

Test #31:

score: 0
Accepted
time: 2380ms
memory: 18012kb

input:

300000 0

output:

12

result:

ok 1 number(s): "12"

Test #32:

score: 0
Accepted
time: 2437ms
memory: 19800kb

input:

300000 1

output:

298803948

result:

ok 1 number(s): "298803948"

Test #33:

score: 0
Accepted
time: 2404ms
memory: 20132kb

input:

300000 2

output:

106751203

result:

ok 1 number(s): "106751203"

Test #34:

score: 0
Accepted
time: 2422ms
memory: 19852kb

input:

300000 1234

output:

427045479

result:

ok 1 number(s): "427045479"

Test #35:

score: 0
Accepted
time: 2343ms
memory: 18392kb

input:

300000 2345

output:

441593553

result:

ok 1 number(s): "441593553"

Test #36:

score: 0
Accepted
time: 2451ms
memory: 19832kb

input:

300000 20041115

output:

580367993

result:

ok 1 number(s): "580367993"

Test #37:

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

input:

300000 20050117

output:

579859619

result:

ok 1 number(s): "579859619"

Test #38:

score: 0
Accepted
time: 2340ms
memory: 17984kb

input:

300000 22223333

output:

596066085

result:

ok 1 number(s): "596066085"

Test #39:

score: 0
Accepted
time: 2398ms
memory: 19800kb

input:

300000 175846372

output:

660364393

result:

ok 1 number(s): "660364393"

Test #40:

score: 0
Accepted
time: 2316ms
memory: 17968kb

input:

299999 9999999

output:

954865020

result:

ok 1 number(s): "954865020"

Test #41:

score: 0
Accepted
time: 2389ms
memory: 17728kb

input:

299998 55556666

output:

904862432

result:

ok 1 number(s): "904862432"

Test #42:

score: 0
Accepted
time: 1206ms
memory: 15368kb

input:

190733 31756136

output:

880544587

result:

ok 1 number(s): "880544587"

Test #43:

score: 0
Accepted
time: 1242ms
memory: 18004kb

input:

198497 463488298

output:

185220207

result:

ok 1 number(s): "185220207"

Test #44:

score: 0
Accepted
time: 2379ms
memory: 19828kb

input:

299997 0

output:

16

result:

ok 1 number(s): "16"

Test #45:

score: 0
Accepted
time: 976ms
memory: 15120kb

input:

168168 296157813

output:

798716760

result:

ok 1 number(s): "798716760"

Test #46:

score: 0
Accepted
time: 550ms
memory: 12940kb

input:

114514 1919810

output:

783513290

result:

ok 1 number(s): "783513290"