QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#340038#4812. Counting SequencetylAC ✓2119ms943544kbC++141.7kb2024-02-28 13:10:562024-02-28 13:10:56

Judging History

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

  • [2024-02-28 13:10:56]
  • 评测
  • 测评结果:AC
  • 用时:2119ms
  • 内存:943544kb
  • [2024-02-28 13:10:56]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10, M = 800, mod = 998244353;
int n, c;
int dp[N][M], f[N], tmp[N], ans;
int inc(const int &a, const int &b) { return (a + b >= mod ? a + b - mod : a + b); }
int mul(const int &a, const int &b) { return 1ll * a * b % mod; }
void Inc(int &a, const int &b) { a = (a + b >= mod ? a + b - mod : a + b); }
int main() {
    // freopen("aosora.in", "r", stdin);
    // freopen("aosora.out", "w", stdout);
    cin>>n>>c;
    for(int i = 1; i <= min(n, M - 1); i++) dp[i][i] = 1;
    for(int i = 2; i <= n; i++)
        for(int j = 1; j <= min(i, M - 1); j++) {
            if(j > 1) Inc(dp[i][j], dp[i - j][j - 1]);
            if(j + 1 < M) Inc(dp[i][j], mul(dp[i - j][j + 1], c));
        }
    for(int i = 1; i < M; i++) Inc(ans, dp[n][i]);
    tmp[M] = 1;
    for(int i = M; i <= n; i++) Inc(tmp[i], dp[i - M][M - 1]);
    // f[0][0] = 1;
    // for(int i = 1; i < M; i++)
    //     for(int j = 0; j < N; j++) {
    //         f[i][j] = mul(f[i - 1][j], c);
    //         if(j >= i) Inc(f[i][j], f[i - 1][j - i]);
    //     }
    f[0] = 1;
    for(int k = 0; k < M; k++) {
        if(k) {
            for(int j = N - 1; j >= 0; j--) {
                f[j] = mul(f[j], c);
                if(j >= k) Inc(f[j], f[j - k]);
            }
        }
        for(int i = M; i <= n; i++) {
            for(int j = i; j <= n; j++) {
                if(i > M && j > i) break;
                int v = i == M ? tmp[j] : 1;
                int res = n - j - i * k + k * (k + 1) / 2;
                if(res >= 2 * N || res < 0 || (res & 1)) continue;
                res >>= 1;
                Inc(ans, mul(v, f[res]));
            }
        }
    }
    
    cout<<ans<<endl;
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 364ms
memory: 4888kb

input:

5 3

output:

8

result:

ok 1 number(s): "8"

Test #2:

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

input:

1 0

output:

1

result:

ok 1 number(s): "1"

Test #3:

score: 0
Accepted
time: 354ms
memory: 11192kb

input:

2022 39

output:

273239559

result:

ok 1 number(s): "273239559"

Test #4:

score: 0
Accepted
time: 372ms
memory: 4856kb

input:

1 998244352

output:

1

result:

ok 1 number(s): "1"

Test #5:

score: 0
Accepted
time: 351ms
memory: 4800kb

input:

1 12345678

output:

1

result:

ok 1 number(s): "1"

Test #6:

score: 0
Accepted
time: 354ms
memory: 4832kb

input:

20 998998

output:

643731701

result:

ok 1 number(s): "643731701"

Test #7:

score: 0
Accepted
time: 358ms
memory: 4780kb

input:

23 123

output:

947753998

result:

ok 1 number(s): "947753998"

Test #8:

score: 0
Accepted
time: 360ms
memory: 4944kb

input:

50 5555

output:

745339864

result:

ok 1 number(s): "745339864"

Test #9:

score: 0
Accepted
time: 356ms
memory: 4904kb

input:

60 6666

output:

690992218

result:

ok 1 number(s): "690992218"

Test #10:

score: 0
Accepted
time: 355ms
memory: 5116kb

input:

100 50

output:

169678588

result:

ok 1 number(s): "169678588"

Test #11:

score: 0
Accepted
time: 355ms
memory: 6396kb

input:

500 88888

output:

216149701

result:

ok 1 number(s): "216149701"

Test #12:

score: 0
Accepted
time: 356ms
memory: 7888kb

input:

1000 213456

output:

270989457

result:

ok 1 number(s): "270989457"

Test #13:

score: 0
Accepted
time: 366ms
memory: 10984kb

input:

2000 119988

output:

756425375

result:

ok 1 number(s): "756425375"

Test #14:

score: 0
Accepted
time: 369ms
memory: 14176kb

input:

3000 998244352

output:

71841227

result:

ok 1 number(s): "71841227"

Test #15:

score: 0
Accepted
time: 363ms
memory: 14196kb

input:

3000 555555555

output:

79880116

result:

ok 1 number(s): "79880116"

Test #16:

score: 0
Accepted
time: 374ms
memory: 18360kb

input:

4321 1234

output:

949603993

result:

ok 1 number(s): "949603993"

Test #17:

score: 0
Accepted
time: 380ms
memory: 20452kb

input:

5000 0

output:

5

result:

ok 1 number(s): "5"

Test #18:

score: 0
Accepted
time: 375ms
memory: 20308kb

input:

5000 88888777

output:

833064960

result:

ok 1 number(s): "833064960"

Test #19:

score: 0
Accepted
time: 376ms
memory: 20428kb

input:

5000 35557777

output:

696388498

result:

ok 1 number(s): "696388498"

Test #20:

score: 0
Accepted
time: 405ms
memory: 36040kb

input:

10000 123456

output:

434296902

result:

ok 1 number(s): "434296902"

Test #21:

score: 0
Accepted
time: 451ms
memory: 67280kb

input:

20000 555555

output:

34806915

result:

ok 1 number(s): "34806915"

Test #22:

score: 0
Accepted
time: 504ms
memory: 98744kb

input:

30000 777888999

output:

58443551

result:

ok 1 number(s): "58443551"

Test #23:

score: 0
Accepted
time: 640ms
memory: 161320kb

input:

50000 2

output:

90102905

result:

ok 1 number(s): "90102905"

Test #24:

score: 0
Accepted
time: 745ms
memory: 223816kb

input:

70000 77998866

output:

202638568

result:

ok 1 number(s): "202638568"

Test #25:

score: 0
Accepted
time: 905ms
memory: 317592kb

input:

100000 998244352

output:

360520717

result:

ok 1 number(s): "360520717"

Test #26:

score: 0
Accepted
time: 916ms
memory: 317596kb

input:

100000 555555555

output:

613886009

result:

ok 1 number(s): "613886009"

Test #27:

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

input:

150000 233333

output:

381065878

result:

ok 1 number(s): "381065878"

Test #28:

score: 0
Accepted
time: 1194ms
memory: 474212kb

input:

150000 20050117

output:

269891864

result:

ok 1 number(s): "269891864"

Test #29:

score: 0
Accepted
time: 1418ms
memory: 630424kb

input:

200000 114514

output:

262861613

result:

ok 1 number(s): "262861613"

Test #30:

score: 0
Accepted
time: 1421ms
memory: 630484kb

input:

200000 1919810

output:

77872388

result:

ok 1 number(s): "77872388"

Test #31:

score: 0
Accepted
time: 2092ms
memory: 943328kb

input:

300000 0

output:

12

result:

ok 1 number(s): "12"

Test #32:

score: 0
Accepted
time: 2007ms
memory: 943452kb

input:

300000 1

output:

298803948

result:

ok 1 number(s): "298803948"

Test #33:

score: 0
Accepted
time: 1894ms
memory: 943384kb

input:

300000 2

output:

106751203

result:

ok 1 number(s): "106751203"

Test #34:

score: 0
Accepted
time: 2024ms
memory: 943544kb

input:

300000 1234

output:

427045479

result:

ok 1 number(s): "427045479"

Test #35:

score: 0
Accepted
time: 2081ms
memory: 943392kb

input:

300000 2345

output:

441593553

result:

ok 1 number(s): "441593553"

Test #36:

score: 0
Accepted
time: 2106ms
memory: 943432kb

input:

300000 20041115

output:

580367993

result:

ok 1 number(s): "580367993"

Test #37:

score: 0
Accepted
time: 2074ms
memory: 943456kb

input:

300000 20050117

output:

579859619

result:

ok 1 number(s): "579859619"

Test #38:

score: 0
Accepted
time: 2021ms
memory: 943356kb

input:

300000 22223333

output:

596066085

result:

ok 1 number(s): "596066085"

Test #39:

score: 0
Accepted
time: 2058ms
memory: 943368kb

input:

300000 175846372

output:

660364393

result:

ok 1 number(s): "660364393"

Test #40:

score: 0
Accepted
time: 2067ms
memory: 943328kb

input:

299999 9999999

output:

954865020

result:

ok 1 number(s): "954865020"

Test #41:

score: 0
Accepted
time: 2019ms
memory: 943372kb

input:

299998 55556666

output:

904862432

result:

ok 1 number(s): "904862432"

Test #42:

score: 0
Accepted
time: 1445ms
memory: 601532kb

input:

190733 31756136

output:

880544587

result:

ok 1 number(s): "880544587"

Test #43:

score: 0
Accepted
time: 1414ms
memory: 625916kb

input:

198497 463488298

output:

185220207

result:

ok 1 number(s): "185220207"

Test #44:

score: 0
Accepted
time: 2119ms
memory: 943380kb

input:

299997 0

output:

16

result:

ok 1 number(s): "16"

Test #45:

score: 0
Accepted
time: 1280ms
memory: 531020kb

input:

168168 296157813

output:

798716760

result:

ok 1 number(s): "798716760"

Test #46:

score: 0
Accepted
time: 973ms
memory: 363112kb

input:

114514 1919810

output:

783513290

result:

ok 1 number(s): "783513290"