結果
| 問題 |
No.1655 123 Swaps
|
| コンテスト | |
| ユーザー |
kozy
|
| 提出日時 | 2021-08-21 01:18:29 |
| 言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 2,671 bytes |
| コンパイル時間 | 112 ms |
| コンパイル使用メモリ | 13,440 KB |
| 実行使用メモリ | 200,824 KB |
| 最終ジャッジ日時 | 2024-10-14 10:05:20 |
| 合計ジャッジ時間 | 24,975 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 12 TLE * 1 -- * 29 |
ソースコード
import numpy as np
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
mod=924844033
inv2=pow(2,mod-2,mod)
def fft_len(N):
fftlen=1
beki=0
while fftlen<N:
fftlen*=2
beki+=1
return fftlen,beki
def FPStime(L,R):
fftlen,beki=fft_len(len(L)+len(R))
FS=np.fft.rfft(L, fftlen)
FT=np.fft.rfft(R, fftlen)
Fh=FS*FT
h = np.fft.irfft(Fh, fftlen)
h = np.rint(h).astype(np.int64)
return h[:len(L) + len(R) - 1]
def FPStime2(L,R,mod):
f1,f2=np.divmod(L,1<<15)
g1,g2=np.divmod(R,1<<15)
a=FPStime(f1,g1)%mod
c=FPStime(f2,g2)%mod
b=(FPStime(f1+f2,g1+g2)-(a+c))%mod
h=(a<<30)+(b<<15)+c
return h%mod
def FPSinv(f):
"""
1/f=gをreturn
"""
f=np.array(f, np.int64)
this_len,this_roop=fft_len(len(f))
c=pow(int(f[0]),-1,mod)
g=np.array([c], np.int64)
a=1
for i in range(this_roop):
a*=2
S=FPStime2(g,f,mod)[:a]
S*=-1
S[0]+=2
S=[i%mod for i in S]
g=FPStime2(g,S,mod)
g=g[:a]
g=[i%mod for i in g]
return g[:len(f)+1]
def FPSsqrt(f):
"""
sqrt(f)=gをreturn
"""
f=np.array(f, np.float64)
g=np.array([f[0]**0.5], np.float64)
this_len,this_roop=fft_len(len(f))
a=1
for i in range(this_roop):
a*=2
g=FPSplus(FPStime(f,FPSinv(g)),g)
g=g*inv2
g=g[:a]
g=[i%mod for i in g]
return g[:len(f)]
def keisu(P,Q,N):#P(x)/Q(x)のx^Nの係数
Q2=list()
for i in range(len(Q)):
if N==0:
return P[0]
if i%2==0:
Q2.append(Q[i])
else:
Q2.append(-Q[i])
S=FPStime2(Q,Q2,mod)
V=list()
for i in range(0,len(S),2):
V.append(S[i])
Ue=list()
Uo=list()
P=FPStime2(P,Q2,mod)
for i in range(len(P)):
if i%2==0:
Ue.append(P[i])
else:
Uo.append(P[i])
if N%2==0:
return keisu(Ue,V,N//2)
else:
return keisu(Uo,V,N//2)
g1 = [1, 1]
g2 = [1, 1]
inverse = [0, 1]
N=5*(10**5)
for i in range( 2, N + 1 ):
g1.append( ( g1[-1] * i ) % mod )
inverse.append( ( -inverse[mod % i] * (mod//i) ) % mod )
g2.append( (g2[-1] * inverse[-1]) % mod )
a,b,c=map(int,input().split())
A=a
B=b
C=c
N=a+b+c
if N%2==1:
print(0)
exit()
L=[[0]*(A+1) for i in range(3)]
R=[[0]*(B+1) for i in range(3)]
for x in range(A+1):
L[x%3][x]=g2[x]*g2[A-x]%mod
for y in range(B+1):
R[y%3][y]=g2[y]*g2[B-y]%mod
C=[0]*(A+B+1)
for x in range(3):
S=FPStime2(L[x],R[((B+2*x-A)*2)%3],mod)
for i in range(len(S)):
C[i]+=S[i]
C[i]%=mod
for i in range(A+B+1):
C[i]*=(g1[(N//2)]**2)*g2[(N//2)-i]*g2[(N//2)-A-B+i]%mod
C[i]%=mod
ans=0
for i in range((N//2)-c,(N//2)+1):
if 0<=i<=A+B:
ans+=C[i]
ans%=mod
print(ans)
kozy