結果
問題 | No.2383 Naphthol |
ユーザー |
![]() |
提出日時 | 2023-07-20 01:05:35 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 239 ms / 2,000 ms |
コード長 | 881 bytes |
コンパイル時間 | 117 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 34,560 KB |
最終ジャッジ日時 | 2024-09-19 22:46:55 |
合計ジャッジ時間 | 5,990 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
N,K=map(int,input().split()) mod=998244353 FACT=[1] for i in range(1,3*10**5+1): FACT.append(FACT[-1]*i%mod) FACT_INV=[pow(FACT[-1],mod-2,mod)] for i in range(3*10**5,0,-1): FACT_INV.append(FACT_INV[-1]*i%mod) FACT_INV.reverse() def Combi(a,b): if 0<=b<=a: return FACT[a]*FACT_INV[b]%mod*FACT_INV[a-b]%mod else: return 0 if N==1: ANS=[1,1,3,3,3,1,1] print(ANS[K]) exit() ANS=Combi(2*N+4,K) # 回転 if K%2==0: ANS+=Combi(N+2,K//2) # 横反転 if N%2==0: if K%2==0: ANS+=Combi(N+2,K//2) else: # N+3のうち 1:1のものが二つ、他は1:2 # 0個 if K%2==0: ANS+=Combi(N+1,K//2) # 1個 if K%2==1: ANS+=Combi(N+1,(K-1)//2)*2 # 2個 if K%2==0: ANS+=Combi(N+1,(K-2)//2) # 縦反転 if K%2==0: ANS+=Combi(N+2,K//2) print((ANS*pow(4,mod-2,mod)%mod)%mod)