結果

問題 No.482 あなたの名は
ユーザー atuk721
提出日時 2017-02-18 06:24:37
言語 Haskell
(9.10.1)
結果
AC  
実行時間 282 ms / 2,000 ms
コード長 858 bytes
コンパイル時間 8,525 ms
コンパイル使用メモリ 201,472 KB
実行使用メモリ 12,672 KB
最終ジャッジ日時 2024-12-30 03:36:47
合計ジャッジ時間 15,247 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

import qualified Data.ByteString.Char8 as B
import qualified Data.Vector.Unboxed.Mutable as V
import Data.Int (Int64)

main :: IO ()
main = do
  (n, k) <- fmap toTuple $ B.getLine
  ds <- fmap B.words $ B.getLine
  vec <- V.unsafeNew n :: IO (V.IOVector Int)
  mapM_ ((\(i, v) -> V.unsafeWrite vec i v) . (\(i, v) -> (i, flip (-) 1 . read . B.unpack $ v))) $ zip [0..] ds
  result <- sortCount n k 0 vec
  putStrLn result

toTuple :: B.ByteString -> (Int, Int64)
toTuple = (\[x, y] -> (read . B.unpack $ x, read . B.unpack $ y)) . B.words

sortCount :: Int -> Int64 -> Int -> V.IOVector Int -> IO (String)
sortCount n k i _
  | k < 0  = return "NO"
  | n == i = return $ if odd k then "NO" else "YES"
sortCount n k i xs = do
  a1 <- V.unsafeRead xs i
  if a1 == i
     then sortCount n k (i + 1) xs
     else V.unsafeSwap xs i a1 >> sortCount n (k - 1) i xs
0