結果
問題 |
No.118 門松列(2)
|
ユーザー |
![]() |
提出日時 | 2018-06-05 21:08:44 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 806 bytes |
コンパイル時間 | 87 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 28,624 KB |
最終ジャッジ日時 | 2024-06-30 10:06:03 |
合計ジャッジ時間 | 6,063 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 4 WA * 22 |
ソースコード
mod=10**9+7 g1 = [1, 1] # 元テーブル g2 = [1, 1] #逆元テーブル inverse = [0, 1] #逆元テーブル計算用テーブル for i in range(2,100001): g1.append( ( g1[-1] * i ) % mod ) inverse.append( ( -inverse[mod % i] * (mod//i) ) % mod ) g2.append( (g2[-1] * inverse[-1]) % mod ) #以下のcom関数を用いて組み合わせを計算 def com(n, r, mod): if ( r<0 or r>n ): return 0 r = min(r, n-r) return g1[n] * g2[r] * g2[n-r] % mod n=int(input()) a=[int(i) for i in input().split()] syurui=0 times=1 count=1 for i in range(1,n): if a[i]==a[i-1]: count+=1 else : times*=(count) count=1 syurui+=1 else: times*=count syurui+=1 if syurui>=3: print(com(syurui,3,mod) * times % mod) else: print(0)