結果

問題 No.3030 Kruskal-Katona
ユーザー nikoro256
提出日時 2025-02-21 22:18:40
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,129 bytes
コンパイル時間 314 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 248,476 KB
最終ジャッジ日時 2025-02-21 22:18:55
合計ジャッジ時間 10,667 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

def cmb(n, r, p):
    if (r < 0) or (n < r):
        return 0
    r = min(r, n - r)
    return fact[n] * factinv[r] * factinv[n-r] % p

p=998244353
N = 10**6  # N は必要分だけ用意する
fact = [1, 1]  # fact[n] = (n! mod p)
factinv = [1, 1]  # factinv[n] = ((n!)^(-1) mod p)
inv = [0, 1]  # factinv 計算用S
 
for i in range(2, N + 1):
    fact.append((fact[-1] * i) % p)
    inv.append((-inv[p % i] * (p // i)) % p)
    factinv.append((factinv[-1] * inv[-1]) % p)

def solve(N,i):
    if i==1:
        return [N]
    l=N
    ans=[]
    last=10**5
    for i in range(i,0,-1):
        left,right=1,last
        while right-left>1:
            mid=(left+right)//2
            if cmb(mid,i,p)<=N:
                left=mid
            else:
                right=mid
        ans.append(left)
        N-=cmb(left,i,p)
        last = left
        if N==0:
            break
    assert N==0
    assert min(ans)>=1
    assert len(ans)==len(set(ans))
    return ans

N,i=map(int,input().split())
print(*solve(N,i))
exit()
N=10**8
import random
for i in range(1000):
    n=97
    i=random.randint(1,10)
    print(n,i,solve(n,i))
0