Inversiot

input/code.hs:2:1: Warning:
    Top-level binding with no type signature:
      deltaAfterIterations :: forall a. Integral a => a -> a -> a

input/code.hs:2:64: Warning:
    Defaulting the following constraint(s) to type `Integer'
      (Integral b0) arising from a use of `^' at input/code.hs:2:64
      (Num b0) arising from the literal `2' at input/code.hs:2:65
    In the first argument of `(+)', namely `iterations ^ 2'
    In the first argument of `div', namely
      `(iterations ^ 2 + iterations)'
    In the second argument of `(-)', namely
      `((iterations ^ 2 + iterations) `div` 2)'

input/code.hs:6:20: Warning:
    Defaulting the following constraint(s) to type `Double'
      (Floating a0) arising from a use of `sqrt' at input/code.hs:6:20-23
      (Fractional a0) arising from a use of `/' at input/code.hs:5:75
      (Num a0)
        arising from a use of `fromIntegral' at input/code.hs:6:25-36
      (RealFrac a0)
        arising from a use of `floor' at input/code.hs:5:30-34...

deltaAfterIterations iterations n = iterations*n - ((iterations^2 + iterations) `div` 2)

iterationsForDelta :: Int -> Int -> Int
iterationsForDelta delta n = floor (((fromIntegral (2*n - 1)) - floating) / 2)
  where floating = sqrt(fromIntegral (4*(n^2) - 4*n + 1 - 8*delta))

main = do
  [n, k] <- do
    l <- getLine
    return ((map read) (words l)) :: IO [Int]
  let even = (deltaAfterIterations (iterationsForDelta k n) n) == k
  let movedFromBeginning = iterationsForDelta k n
  if even
    then do
      putStr (unwords (map (show) [(movedFromBeginning+1)..n]))
      putStr " "
      putStrLn (unwords (map (\x -> show (movedFromBeginning - x + 1)) [1..movedFromBeginning]))
    else do
      let index = k - (deltaAfterIterations (iterationsForDelta k n) n)
      putStr (unwords (map (show) ((take index [(movedFromBeginning+2)..n]))))
      putStr " "
      putStr (show (movedFromBeginning+1))
      putStr " "
      putStr (unwords (map (show) (drop index [(movedFromBeginning+2)..n])))
      putStr " "
      putStrLn (unwords (map (\x -> show (movedFromBeginning - x + 1)) [1..(movedFromBeginning)]))
  

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED