2와 0 열에서 '2020' 찾기

5319 단어 2020하스켈
무작위로 늘어선 '2'와 '0' 열에서 '2020'을 찾아 하이라이트한다.

연하장을 위해 PostScript로 작성한 것을 Haskell로 다시 작성했습니다. 좀 더 다양한 글쓰기를 할 수 있을 것 같다. 쉘 스크립트라든지라도 쓸 수 있을 것 같다.
module Main where

import System.Environment (getArgs)
import System.Random
import Data.List

scan2020 :: [Int] -> [[Int]]
scan2020 [] = []
scan2020 ls@(x:xs)
  | isPrefixOf the2020 ls = the2020 : (scan2020 $ drop 4 ls)
  | otherwise = [x] : scan2020 xs
  where the2020 = [2, 0, 2, 0]

hl2020 :: [[Int]] -> [String]
hl2020 [] = []
hl2020 (x:xs)
  | length x == 4 =
      concat ["\ESC[31m", intercalate " " $ map show x, "\ESC[m"] : hl2020 xs
  | otherwise = concatMap show x : hl2020 xs

main :: IO ()
main = do
  a <- fmap (read . head) getArgs
  s <- getStdGen
  putStrLn $ intercalate " " $ hl2020 $ take a $ scan2020 $ twoandzero s
  where
    twoandzero s = map ((*2) . flip mod 2) (randoms s :: [Int])

좋은 웹페이지 즐겨찾기