結果

問題 No.482 あなたの名は
コンテスト
ユーザー atuk721
提出日時 2017-02-17 08:49:06
言語 Haskell
(9.10.1)
結果
AC  
実行時間 254 ms / 2,000 ms
コード長 827 bytes
コンパイル時間 15,036 ms
コンパイル使用メモリ 182,540 KB
実行使用メモリ 12,800 KB
最終ジャッジ日時 2024-12-30 22:44:03
合計ジャッジ時間 9,805 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 Data.Array.Base (unsafeRead, unsafeWrite)
import Data.Int (Int64)
import Data.Array.IO (IOUArray, newListArray)

main :: IO ()
main = do
  (n, k) <- fmap ((\[a, b] -> (read . B.unpack $ a, read . B.unpack $ b)) . B.words) $ B.getLine
  ds <- fmap (map (flip (-) 1 . read . B.unpack) . B.words) $ B.getLine
  is <- newListArray (0, n - 1) ds
  result <- sortCount n k 0 is
  putStrLn result

sortCount :: Int -> Int64 -> Int -> IOUArray Int 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 <- unsafeRead xs i
  if a1 == i
     then sortCount n k (i + 1) xs
     else do
      a2 <- unsafeRead xs a1
      unsafeWrite xs i a2
      unsafeWrite xs a1 a1
      sortCount n (k - 1) i xs
0