import Data.List.Split
import Data.List

main :: IO ()
main = do
  input <- readCords <$> map (map (splitOn ",") . words) . lines <$> getContents
  print $ part1 input
  print $ part2 input

readCords :: [[[String]]]-> [((Int, Int), (Int, Int))]
readCords xs = [ ((read x1, read y1), (read x2, read y2)) | [[x1,y1],_,[x2,y2]] <- xs]

count2LineOverlap :: [(Int, Int)] -> Int
count2LineOverlap xs = length $ filter (>=2) $ map length $ group $ sort xs         

part1 :: [((Int, Int), (Int, Int))] -> Int
part1 xs = count2LineOverlap input
    where
      input = collectPoints $ filterLines xs
      collectPoints = foldl (\x y -> x ++ coverPoints y) []
      coverPoints ((x1,y1),(x2,y2)) = [(x,y) | x <-[a..b], y <- [c..d]] 
        where
          (a,b) = if x1 < x2 then (x1,x2) else (x2,x1)
          (c,d) = if y1 < y2 then (y1,y2) else (y2,y1)
      filterLines = filter (\((x1,y1),(x2,y2)) -> x1 == x2 || y1 == y2)

part2 :: [((Int, Int), (Int, Int))] -> Int      
part2 xs = count2LineOverlap input
    where
      slope (x1,y1) (x2,y2) = abs (y1-y2) == abs (x1-x2)
      input = collectPoints $ filterLines xs
      collectPoints = foldl (\x y -> x ++ coverPoints y) []
      coverPoints ((x1,y1),(x2,y2))
        | x1==x2 || y1==y2 = [(x,y) | x <-[a..b], y <- [c..d]]
        | slope (x1,y1) (x2,y2) = [(x,y) | x <-[a..b], y <- [c..d], slope (x1,y1) (x,y)]
        where
          (a,b) = if x1 < x2 then (x1,x2) else (x2,x1)
          (c,d) = if y1 < y2 then (y1,y2) else (y2,y1)
      filterLines = filter (\((x1,y1),(x2,y2)) -> x1 == x2 || y1 == y2 || slope (x1,y1) (x2,y2))  

Haskell online compiler

Write, Run & Share Haskell code online using OneCompiler's Haskell online compiler for free. It's one of the robust, feature-rich online compilers for Haskell language, running the latest Haskell version 8.6. Getting started with the OneCompiler's Haskell editor is easy and fast. The editor shows sample boilerplate code when you choose language as Haskell and start coding.

Taking inputs (stdin)

OneCompiler's Haskell online editor supports stdin and users can give inputs to programs using the STDIN textbox under the I/O tab. Following is a sample Haskell program which takes name as input and prints hello message with your name.

main = do  
    name <- getLine  
    putStrLn ("Hello " ++ name ++ ", Happy learning!") 

About Haskell

Haskell is purely a functional programming language which was introduced in 1990's.

Key Features

  • Haskell is both compiled and interpreted
  • Lazy language as the results are computed only if required
  • Pure functions
  • Pattern matching on data structures
  • Emphasizes on what to do but not on how to do
  • Glasgow Haskell Compiler (GHC), most widely used Haskell compiler also written in Haskell.
  • Data is immutable

Syntax help

Data Types

Data-typeDescription
NumbersHaskell is intelligent to identify numbers without specifying data type
CharactersHaskell is intelligent to identify characters and strings without specifying data type
TupleTo declare multiple values in a single data type. Tuples are represented in single paranthesis. For example (10, 20, 'apple')
BooleanTo represent boolean values, true or false
ListTo declare same type of values in a single data type. Lists are represented in square braces.For example [1, 2, 3] or `['a','b','c','d']

Control statements

If-Else / Nested If-Else:

When ever you want to perform a set of operations based on a condition or set of conditions, then If-Else/ Nested-If-Else are used.

Example:

main = do   
   let age = 21 
   if age > 18 
      then putStrLn "Adult" 
   else putStrLn "child"

Functions

Function is a sub-routine which contains set of statements. Usually functions are written when multiple calls are required to same set of statements which increases re-usuability and modularity. Functions play an important role in Haskell, since it is a purely functional language.

Example

multiply :: Integer -> Integer -> Integer   --declaration of a function 
multiply x1 x2 =  x1 * x2                   --definition of a function

main = do 
   putStrLn "Multiplication value is:"  
   print(multiply 10 5)    --calling a function