Submission #17752380

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys
import numpy as np
import numba
from numba import njit, b1, i4, i8, f8
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
MOD = 998_244_353
@njit((i8, i8), cache=True)
def main(N, K):
    dp = np.zeros((N + 1, 2 * N + 1), np.int64)  # 個数、和
    dp[0, 0] = 1
    for n in range(1, N + 1):
        for s in range(n, 0, -1):
            x = dp[n - 1, s - 1]
            # 1 を使わない
            x += dp[n, 2 * s]
            if x >= MOD:
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

import sys
import numpy as np
import numba
from numba import njit, b1, i4, i8, f8

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

MOD = 998_244_353

@njit((i8, i8), cache=True)
def main(N, K):
    dp = np.zeros((N + 1, 2 * N + 1), np.int64)  # 個数、和
    dp[0, 0] = 1
    for n in range(1, N + 1):
        for s in range(n, 0, -1):
            x = dp[n - 1, s - 1]
            # 1 を使わない
            x += dp[n, 2 * s]
            if x >= MOD:
                x -= MOD
            dp[n, s] = x
    return dp[N, K]

a, b = map(int, read().split())

print(main(a, b))

Submission Info

Judge Result