結果

問題 No.1035 Color Box
ユーザー ikdikd
提出日時 2020-05-02 02:17:16
言語 F#
(F# 4.0)
結果
AC  
実行時間 485 ms / 2,000 ms
コード長 1,249 bytes
コンパイル時間 19,559 ms
コンパイル使用メモリ 198,552 KB
実行使用メモリ 45,276 KB
最終ジャッジ日時 2024-12-25 23:31:46
合計ジャッジ時間 38,801 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.fsproj を復元しました (554 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
/home/judge/data/code/Main.fs(15,9): warning FS0025: この式のパターン マッチが不完全です たとえば、値 '[|_; _; _|]' はパターンに含まれないケースを示す可能性があります。 [/home/judge/data/code/main.fsproj]
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

// Learn more about F# at http://fsharp.org

open System

let rec mpow (a: int64) x m =
    if x = 1L then
        a
    elif x % 2L = 0L then
        mpow (a * a % m) (x / 2L) m
    else
        a * mpow a (x - 1L) m % m

[<EntryPoint>]
let main argv =
    let [|n; m|] = stdin.ReadLine().Split() |> Array.map int
    let l = [n; m] |> List.max
    let mo = 1000000007L
    // let mutable fact = [n; m] |> List.max |> Array.create 1
    let fact = 
        Array.concat[ [|1L|]; [| for i in 1L .. int64(l) -> i |] ]
            |> Array.mapFold (fun acc x -> acc * x % mo, acc * x % mo) 1L
            |> fst
    let inv = 
        Array.concat[ [| for i in 1L .. int64(l) -> i |]; [|1L|] ]
        |> Array.rev
        |> Array.mapFold (fun acc x -> acc * x % mo, acc * x % mo) (mpow fact.[l] (mo - 2L) mo)
        |> fst
        |> Array.rev
    let binom a b =
        fact.[a] * inv.[b] % mo * inv.[a - b] % mo
    let mutable ans = 0L
    for i = m downto 1 do
        if (m - i) % 2 = 0 then
            ans <- (ans + (binom m i) * (mpow (i |> int64) (n |> int64) mo) % mo) % mo
        else
            ans <- (ans - (binom m i) * (mpow (i |> int64) (n |> int64) mo) % mo + mo) % mo
    printfn "%d" ans
    0 // return an integer exit code
0