結果

問題 No.1035 Color Box
コンテスト
ユーザー ikd
提出日時 2020-05-02 02:26:21
言語 F#
(F# 4.0)
結果
AC  
実行時間 472 ms / 2,000 ms
コード長 1,191 bytes
コンパイル時間 15,208 ms
コンパイル使用メモリ 199,528 KB
実行使用メモリ 45,148 KB
最終ジャッジ日時 2024-12-26 00:00:07
合計ジャッジ時間 30,344 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.fsproj を復元しました (494 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
/home/judge/data/code/Main.fs(12,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 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