結果

問題 No.2846 Birthday Cake
ユーザー detteiuudetteiuu
提出日時 2024-08-23 23:02:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,928 ms / 2,000 ms
コード長 694 bytes
コンパイル時間 432 ms
コンパイル使用メモリ 82,324 KB
実行使用メモリ 79,332 KB
最終ジャッジ日時 2024-08-23 23:03:12
合計ジャッジ時間 20,904 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**6)
import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
from math import lcm

K, N = map(int, input().split())

LCM = [1]
for i in range(2, 25):
    LCM.append(lcm(LCM[-1], i))
L = LCM[-1]
P = [1]
for i in range(2, 25):
    P.append(P[-1]*i)

def func(n, R, c, a):
    if n == 0:
        return a
    ans = 0
    for i in range(min(n+1, c+1)):
        r = R-L//n*i
        if r < 0:
            break
        if n >= 2 and r%(L//LCM[n-2]) == 0 or n == 1 and r == 0 and c == i:
            cnt[n-1] += i
            ans += func(n-1, r, c-i, a//P[i-1] if i >= 1 else a)
            cnt[n-1] -= i
    return ans

cnt = [0]*N
print(func(N, L, K, P[K-1]))
0