結果

問題 No.482 あなたの名は
ユーザー atuk721
提出日時 2017-02-18 06:37:06
言語 Haskell
(9.10.1)
結果
AC  
実行時間 248 ms / 2,000 ms
コード長 856 bytes
コンパイル時間 4,982 ms
コンパイル使用メモリ 204,672 KB
実行使用メモリ 12,800 KB
最終ジャッジ日時 2024-12-14 10:48:26
合計ジャッジ時間 10,758 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 <- B.getLine
  vec <- V.unsafeNew n
  mapM_ ((\(i, v) -> V.unsafeWrite vec i v) . convert) . zip [0..] $ B.words ds
  result <- proc n k 0 vec
  putStrLn result

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

convert :: (Int, B.ByteString) -> (Int, Int)
convert (i, v) = (i, flip (-) 1 . read . B.unpack $ v)

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