結果

問題 No.1510 Simple Integral
コンテスト
ユーザー 정현우
提出日時 2022-01-15 02:02:59
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,206 bytes
コンパイル時間 298 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 24,784 KB
最終ジャッジ日時 2024-11-20 19:13:46
合計ジャッジ時間 107,223 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 TLE * 2
other AC * 10 TLE * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#(1/pi)*(1/N)*Integral[1/(x^2+A^2)^m,{x,-Infinity,Infinity}]
#=>cal(N,A,m)

mod=998244353

def minimize(lli):
    if len(lli)==1:
        return lli
    li=lli[:]
    li.sort(key=lambda x:x[1])
    
    while True:
        if li[-1][1]==li[-2][1] and li[-1][2]==li[-2][2]:
            li[-2]=(li[-1][0]*li[-2][0]*pow(li[-1][0]+li[-2][0],-1,mod)%mod, li[-1][1],li[-1][2])
            del li[-1]
    return li


def cal(N,A,m):
    numer=1
    denom=N*pow(A,2*m-1,mod)*pow(4,m-1,mod)%mod
    for i in range(m,2*m-1):
        numer*=i
        numer%=mod
    for i in range(1,m):
        denom*=i
        denom%=mod
    
    return numer*pow(denom,-1,mod)%mod


def change(tuple_1,A2):
    N1,A1,m1=tuple_1
    if A1==A2:
        return [(N1,A1,m1+1)]
    if m1==1:
        return [(N1*(A2**2-A1**2)%mod,A1,1),(-N1*(A2**2-A1**2)%mod,A2,1)]
    else:
        tmp=change((-N1*(A2**2-A1**2)%mod, A1, m1-1), A2)
        return [(N1*(A2**2-A1**2)%mod,A1,m1)]+tmp

n=int(input())
a=list(map(int,input().split()))
a.sort()

li=[(1,a[0],1)]
for i in range(1,n):
    ret=[]
    for tu in li:
        ret+=change(tu,a[i])
    li=ret[:]
    li=minimize(li)
#print(li)
ans=0
for tu in li:
    ans+=cal(*tu)
print(ans%mod)
0