結果

問題 No.39 桁の数字を入れ替え
ユーザー tjnttjnt
提出日時 2021-11-03 16:10:37
言語 Haskell
(9.8.2)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 728 bytes
コンパイル時間 5,004 ms
コンパイル使用メモリ 168,832 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-13 06:11:06
合計ジャッジ時間 5,862 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 1 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 1 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 2 ms
5,248 KB
testcase_14 AC 1 ms
5,248 KB
testcase_15 AC 2 ms
5,248 KB
testcase_16 AC 1 ms
5,248 KB
testcase_17 AC 1 ms
5,248 KB
testcase_18 AC 1 ms
5,248 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 Data.List
import Data.Char (digitToInt)
import Data.Maybe (fromJust)

swap :: Int -> Int -> [a] -> [a]
swap i j a = go 0 a
  where
    go _ [] = []
    go k (x:xs)
      | k == i = (a !! j) : go (k+1) xs
      | k == j = (a !! i) : go (k+1) xs
      | otherwise = x : go (k+1) xs

solve :: [Int] -> [Int]
solve a = go a rs
    where
      rs = reverse . sort $ a
      go [] [] = a
      go _  [] = a
      go []  _ = a
      go (x:xs) (r:rs)
        | x == r = go xs rs
        | otherwise =
            let i = fromJust $ elemIndex x a
                j = last $ elemIndices r a
             in swap i j a

main :: IO ()
main = do
    n <- map digitToInt <$> getLine :: IO [Int]
    putStrLn $ concatMap show $ solve n
0