結果
問題 |
No.1100 Boxes
|
ユーザー |
![]() |
提出日時 | 2021-09-16 19:28:03 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 721 ms / 2,000 ms |
コード長 | 2,475 bytes |
コンパイル時間 | 167 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 19,840 KB |
最終ジャッジ日時 | 2024-06-29 15:26:59 |
合計ジャッジ時間 | 9,540 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 36 |
ソースコード
from math import gcd as GCD def Extended_Euclid(n,m): stack=[] while m: stack.append((n,m)) n,m=m,n%m if n>=0: x,y=1,0 else: x,y=-1,0 for i in range(len(stack)-1,-1,-1): n,m=stack[i] x,y=y,x-(n//m)*y return x,y class MOD: def __init__(self,p,e=1): self.p=p self.e=e self.mod=self.p**self.e def Pow(self,a,n): a%=self.mod if n>=0: return pow(a,n,self.mod) else: assert GCD(a,self.mod)==1 x=Extended_Euclid(a,self.mod)[0] return pow(x,-n,self.mod) def Build_Fact(self,N): assert N>=0 self.factorial=[1] self.cnt=[0]*(N+1) for i in range(1,N+1): ii=i self.cnt[i]=self.cnt[i-1] while ii%self.p==0: ii//=self.p self.cnt[i]+=1 self.factorial.append((self.factorial[-1]*ii)%self.mod) self.factorial_inve=[None]*(N+1) self.factorial_inve[-1]=self.Pow(self.factorial[-1],-1) for i in range(N-1,-1,-1): ii=i+1 while ii%self.p==0: ii//=self.p self.factorial_inve[i]=(self.factorial_inve[i+1]*ii)%self.mod def Fact(self,N): if N<0: return 0 return self.factorial[N]*pow(self.p,self.cnt[N],self.mod)%self.mod def Fact_Inve(self,N): if self.cnt[N]: return None return self.factorial_inve[N] def Comb(self,N,K,divisible_count=False): if K<0 or K>N: return 0 retu=self.factorial[N]*self.factorial_inve[K]*self.factorial_inve[N-K]%self.mod cnt=self.cnt[N]-self.cnt[N-K]-self.cnt[K] if divisible_count: return retu,cnt else: retu*=pow(self.p,cnt,self.mod) retu%=self.mod return retu def Montmort_Numbers(N,mod=0): montmort_numbers=[0]*(N+1) if N>=2: montmort_numbers[2]=1 for i in range(3,N+1): montmort_numbers[i]=(i-1)*(montmort_numbers[i-1]+montmort_numbers[i-2]) if mod: montmort_numbers[i]%=mod return montmort_numbers N,K=map(int,input().split()) ans=0 mod=998244353 MD=MOD(mod) MD.Build_Fact(K) for k in range(1,K): if k%2: ans+=pow(K-k,N,mod)*pow(2,k-1,mod)*MD.Comb(K,k) else: ans-=pow(K-k,N,mod)*pow(2,k-1,mod)*MD.Comb(K,k) ans%=mod print(ans)