AdventOfCode/2020/03_haskell.ipynb

97 lines
1.8 KiB
Text

{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"import System.IO\n",
"\n",
"handle <- openFile \"03.txt\" ReadMode \n",
"puzzle <- hGetContents handle \n",
"let plines = lines puzzle\n",
"\n",
"int x = read x :: Int"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Day 03"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"### Puzzle 1"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"294"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"length $ filter (\\(i, line) -> line !! mod (i*3) (length line) == '#') $ zip [0..] plines"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"### Puzzle 2"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"5774564250"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"slice lst x = [e | (i, e) <- zip [0..] lst, mod i x == 0]\n",
"count a b = length $ filter (\\(i, line) -> line !! mod (i*a) (length line) == '#') $ zip [0..] $ slice plines b\n",
"product [count a 1 | a <- [1,3,5,7]] * count 1 2"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Haskell",
"language": "haskell",
"name": "haskell"
},
"language_info": {
"codemirror_mode": "Haskell",
"file_extension": ".hs",
"mimetype": "text/x-haskell",
"name": "haskell",
"pygments_lexer": "Haskell",
"version": "8.8.4"
}
},
"nbformat": 4,
"nbformat_minor": 4
}