Submission #57538973

Source Code Expand

n, k = map(int, input().split())
a = list(map(int, input().split()))

MOD = 998244353
f = {}
f[0] = 1
t = 1
s = 0
for i in range(n):
    s += a[i]
    if s-k in f:
        x = t - f[s-k]
    else:
        x = t
    t += x
    x, t = x % MOD, t % MOD
    if s not in f:
        f[s] = x
    else:
        f[s] += x
    # print(i, s, s-k, t, x, f)
print(x%MOD)

Submission Info

Submission Time
Task E - Avoid K Partition
User yefllower
Language Python (PyPy 3.10-v7.3.12)
Score 475
Code Size 385 Byte
Status AC
Exec Time 121 ms
Memory 138412 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 475 / 475
Status
AC × 3
AC × 37
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_small_00.txt, 01_small_01.txt, 01_small_02.txt, 01_small_03.txt, 01_small_04.txt, 01_small_05.txt, 01_small_06.txt, 01_small_07.txt, 01_small_08.txt, 01_small_09.txt, 02_random_00.txt, 02_random_01.txt, 02_random_02.txt, 02_random_03.txt, 02_random_04.txt, 02_random_05.txt, 02_random_06.txt, 02_random_07.txt, 02_random_08.txt, 02_random_09.txt, 03_max_00.txt, 03_max_01.txt, 03_max_02.txt, 03_max_03.txt, 03_max_04.txt, 03_max_05.txt, 03_max_06.txt, 03_max_07.txt, 03_max_08.txt, 03_max_09.txt, 04_min_00.txt, 04_min_01.txt, 05_hack_00.txt, 05_hack_01.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 57 ms 76284 KiB
00_sample_01.txt AC 57 ms 76560 KiB
00_sample_02.txt AC 57 ms 76288 KiB
01_small_00.txt AC 57 ms 76448 KiB
01_small_01.txt AC 57 ms 76448 KiB
01_small_02.txt AC 57 ms 76588 KiB
01_small_03.txt AC 57 ms 76224 KiB
01_small_04.txt AC 58 ms 76592 KiB
01_small_05.txt AC 58 ms 76404 KiB
01_small_06.txt AC 58 ms 76680 KiB
01_small_07.txt AC 57 ms 76556 KiB
01_small_08.txt AC 58 ms 76648 KiB
01_small_09.txt AC 58 ms 76420 KiB
02_random_00.txt AC 100 ms 129704 KiB
02_random_01.txt AC 62 ms 81820 KiB
02_random_02.txt AC 88 ms 114616 KiB
02_random_03.txt AC 79 ms 98528 KiB
02_random_04.txt AC 94 ms 119832 KiB
02_random_05.txt AC 68 ms 88816 KiB
02_random_06.txt AC 71 ms 95132 KiB
02_random_07.txt AC 80 ms 101196 KiB
02_random_08.txt AC 83 ms 107356 KiB
02_random_09.txt AC 75 ms 94544 KiB
03_max_00.txt AC 118 ms 138340 KiB
03_max_01.txt AC 85 ms 109976 KiB
03_max_02.txt AC 120 ms 138332 KiB
03_max_03.txt AC 89 ms 109928 KiB
03_max_04.txt AC 119 ms 138412 KiB
03_max_05.txt AC 86 ms 110236 KiB
03_max_06.txt AC 117 ms 138324 KiB
03_max_07.txt AC 89 ms 110140 KiB
03_max_08.txt AC 121 ms 137996 KiB
03_max_09.txt AC 89 ms 109904 KiB
04_min_00.txt AC 58 ms 76344 KiB
04_min_01.txt AC 57 ms 76664 KiB
05_hack_00.txt AC 79 ms 109892 KiB
05_hack_01.txt AC 83 ms 109608 KiB