結果

問題 No.2299 Antitypoglycemia
コンテスト
ユーザー mtwtkman
提出日時 2023-05-14 21:07:49
言語 Haskell
(9.14.1)
コンパイル:
ghc -rtsopts -with-rtsopts=-K1G -o a.out -O2 _filename_
実行:
./a.out
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 734 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,065 ms
コンパイル使用メモリ 195,456 KB
実行使用メモリ 19,200 KB
最終ジャッジ日時 2026-05-23 23:49:05
合計ジャッジ時間 3,590 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge1_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.14.1/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
Main.hs:28:29: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘head’
    (imported from Prelude, but defined in GHC.Internal.List):
    "This is a partial function, it throws an error on empty lists. Use pattern matching, 'Data.List.uncons' or 'Data.Maybe.listToMaybe' instead. Consider refactoring to use "Data.List.NonEmpty"."
   |
28 |   let [n, a, b] = map read [head vs, vs !! 1, vs !! 2]
   |                             ^^^^

[2 of 2] Linking a.out

ソースコード

diff #
raw source code

split :: String -> [String]
split = go ("", [])
  where
    go :: (String, [String]) -> String -> [String]
    go (a, acc) "" = acc <> [a]
    go (a, acc) (' ' : xs) = go ("", acc <> [a]) xs
    go (a, acc) (x : xs) = go (a <> [x], acc) xs

prompt :: IO [String]
prompt = do split <$> getLine

m = 998244353

fac :: Integer -> Integer
-- fac 0 = 1
-- fac n = (fac (n - 1) * n) `mod` m

fac n = foldr (\x acc -> (acc * x) `mod` m) 1 [1..n]

solve :: Integer -> Integer -> Integer -> Integer
solve n a b =
  let r = (fac n + ((m - 2) * fac (n - 1)) `mod` m) `mod` m
   in if a /= b then (r + fac (n - 2)) `mod` m else r

main :: IO ()
main = do
  vs <- prompt
  let [n, a, b] = map read [head vs, vs !! 1, vs !! 2]
  print $ solve n a b
0