# 99 questions/Solutions/3

### From HaskellWiki

< 99 questions | Solutions(Difference between revisions)

(10 intermediate revisions by 7 users not shown) | |||

Line 13: | Line 13: | ||

<haskell> | <haskell> | ||

elementAt :: [a] -> Int -> a | elementAt :: [a] -> Int -> a | ||

− | elementAt list i = list !! (i-1) | + | elementAt list i = list !! (i-1) |

</haskell> | </haskell> | ||

Line 20: | Line 20: | ||

<haskell> | <haskell> | ||

elementAt' :: [a] -> Int -> a | elementAt' :: [a] -> Int -> a | ||

− | elementAt' (x:_) 1 = x | + | elementAt' (x:_) 1 = x |

− | elementAt' [] _ = error "Index out of bounds" | + | elementAt' [] _ = error "Index out of bounds" |

elementAt' (_:xs) k | elementAt' (_:xs) k | ||

− | + | | k < 1 = error "Index out of bounds" | |

− | + | | otherwise = elementAt' xs (k - 1) | |

</haskell> | </haskell> | ||

Line 31: | Line 31: | ||

<haskell> | <haskell> | ||

elementAt'' :: [a] -> Int -> a | elementAt'' :: [a] -> Int -> a | ||

− | elementAt'' (x:_) 1 = x | + | elementAt'' (x:_) 1 = x |

− | elementAt'' (_:xs) i = elementAt xs (i - 1) | + | elementAt'' (_:xs) i = elementAt'' xs (i - 1) |

− | elementAt'' _ _ = error "Index out of bounds" | + | elementAt'' _ _ = error "Index out of bounds" |

</haskell> | </haskell> | ||

'''This does not work correctly on invalid indexes and infinite lists, e.g.:''' | '''This does not work correctly on invalid indexes and infinite lists, e.g.:''' | ||

Line 43: | Line 43: | ||

<haskell> | <haskell> | ||

− | + | elementAt'' xs n | |

− | -- Main> | + | | length xs < n = error "Index out of bounds" |

− | -- | + | | otherwise = fst . last $ zip xs [1..n] |

+ | |||

+ | elementAt''' xs n = head $ foldr ($) xs | ||

+ | $ replicate (n - 1) tail | ||

+ | -- Negative indices not handled correctly: | ||

+ | -- Main> elementAt''' "haskell" (-1) | ||

+ | -- 'h' | ||

+ | |||

+ | elementAt'''' xs n | ||

+ | | length xs < n = error "Index out of bounds" | ||

+ | | otherwise = last $ take n xs | ||

+ | |||

+ | elementAt''''' xs n | ||

+ | | length xs < n = error "Index out of bounds" | ||

+ | | otherwise = head . reverse $ take n xs | ||

+ | |||

+ | elementAt'''''' xs n | ||

+ | | length xs < n = error "Index out of bounds" | ||

+ | | otherwise = head $ drop (n - 1) xs | ||

+ | |||

</haskell> | </haskell> | ||

+ | or <hask>elementAt_w'</hask> correctly in point-free style: | ||

<haskell> | <haskell> | ||

− | elementAt_w' | + | elementAt_w'pf = (last .) . take . (+ 1) |

− | + | ||

− | + | ||

</haskell> | </haskell> | ||

+ | Pedantic note: the above definition of <hask>elementAt_w'pf</hask> does not conform to the order of arguments specified by the question, but the following does: | ||

<haskell> | <haskell> | ||

− | + | elementAt_w'pf' = flip $ (last .) . take . (+ 1) | |

</haskell> | </haskell> | ||

+ | |||

+ | |||

+ | [[Category:Programming exercise spoilers]] |

## Latest revision as of 02:32, 15 June 2016

(*) Find the K'th element of a list. The first element in the list is number 1.

This is (almost) the infix operator !! in Prelude, which is defined as:

(!!) :: [a] -> Int -> a (x:_) !! 0 = x (_:xs) !! n = xs !! (n-1)

Except this doesn't quite work, because !! is zero-indexed, and element-at should be one-indexed. So:

elementAt :: [a] -> Int -> a elementAt list i = list !! (i-1)

Or without using the infix operator:

elementAt' :: [a] -> Int -> a elementAt' (x:_) 1 = x elementAt' [] _ = error "Index out of bounds" elementAt' (_:xs) k | k < 1 = error "Index out of bounds" | otherwise = elementAt' xs (k - 1)

Alternative version:

elementAt'' :: [a] -> Int -> a elementAt'' (x:_) 1 = x elementAt'' (_:xs) i = elementAt'' xs (i - 1) elementAt'' _ _ = error "Index out of bounds"

**This does not work correctly on invalid indexes and infinite lists, e.g.:**

elementAt'' [1..] 0

A few more solutions using prelude functions:

elementAt'' xs n | length xs < n = error "Index out of bounds" | otherwise = fst . last $ zip xs [1..n] elementAt''' xs n = head $ foldr ($) xs $ replicate (n - 1) tail -- Negative indices not handled correctly: -- Main> elementAt''' "haskell" (-1) -- 'h' elementAt'''' xs n | length xs < n = error "Index out of bounds" | otherwise = last $ take n xs elementAt''''' xs n | length xs < n = error "Index out of bounds" | otherwise = head . reverse $ take n xs elementAt'''''' xs n | length xs < n = error "Index out of bounds" | otherwise = head $ drop (n - 1) xs

elementAt_w'

elementAt_w'pf = (last .) . take . (+ 1)

elementAt_w'pf

elementAt_w'pf' = flip $ (last .) . take . (+ 1)