Submission #3771124

---

Source Code Expand

import Control.Arrow
import Control.Monad
import Data.Maybe
import Data.List

type Hour = Int
type Minute = Int
data Time = Time {_hour::Hour, _minute::Minute} deriving (Show, Eq)
instance Ord Time where
    compare (Time h1 m1) (Time h2 m2) = do
        let h = compare h1 h2
        if h == EQ then compare m1 m2
        else h
data RainTime = RainTime {_start::Time, _end::Time} deriving (Show, Eq)
instance Ord RainTime where
    compare (RainTime s1 e1) (RainTime s2 e2) = do
        let s = compare s1 s2
        if s == EQ then compare e1 e2
        else s
main = do
    time <- read <$> getLine :: IO Int
    rains <- map strToRain <$> replicateM time getLine
    putStrLn (foldRains >>> map formatRain >>> intercalate "\n" $ rains)
addRainTime f@(RainTime s1 e1) (RainTime s2 e2) = do
    if s2 <= e1 then Just f{_end = maximum[e1,e2]}
    else Nothing

strToRain str = do
    let (s,_e) = span (/= '-') str
    let e = tail _e
    RainTime (toStartTime s) (toEndTime e)

toStartTime str = toTime str (\x -> div x 5 * 5)
toEndTime str = toTime str (\x -> div (x + 4) 5 * 5)
toTime str f = do
    let min = drop 2 >>> read >>> f $ str
    let hou = take 2 >>> read $ str
    if min >= 60 then Time (hou + 1) 0
    else Time hou min 

foldRains arr = do
    let (rem, _arr) = foldl' func (head sorted, []) $ tail sorted
    reverse $ rem:_arr
    where func (now, confirmed) coming = do
            let newTime = addRainTime now coming
            if isJust newTime then (fromJust newTime, confirmed)
            else (coming, now:confirmed)
          sorted = sort arr

formatTime (Time h m) = (paddingZero h) ++ (paddingZero m)
    where paddingZero str = reverse >>> take 2 >>> reverse $ "00" ++ (show str)
formatRain (RainTime s e) = (formatTime s) ++ "-" ++ (formatTime e)

Submission Info

Judge Result