day3.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "day3.ipynb",
      "provenance": [],
      "collapsed_sections": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "_6pLOZKYAmzx"
      },
      "source": [
        "https://adventofcode.com/2020/day/3"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "SKMKHgXI7Ive"
      },
      "source": [
        "input = r'''....#...............#.#..###.##\n",
        ".#..#....###..............##...\n",
        "....###......#....#.#...#.##..#\n",
        ".......#........#..###...##....\n",
        ".....#..#......#..#..##..#...#.\n",
        "....#..........#....#...#......\n",
        "............###...#............\n",
        "##......#.....#......#.....##..\n",
        "........#.........##..#.#...##.\n",
        "....#.#..#.#...#........#..#...\n",
        ".#.....#.#......#....#..#..#..#\n",
        "#.##..##......#.....##...#..#..\n",
        "#........#..##...###....##.....\n",
        "......#.#..##...#.#.....#......\n",
        "##.......#..#.........#...#....\n",
        ".....##.........#....#.#.###.#.\n",
        "..##...........#.#.#.#.....#.#.\n",
        "....#...............#......#.#.\n",
        "#.#..#....#.....#.....##...#..#\n",
        "#......#..............#.#.##...\n",
        "......###.....#...#........###.\n",
        "####...#.....#...#....#........\n",
        ".......#...#....##...#.........\n",
        ".####..##............#.........\n",
        "#.#...#...#....#...#.#......#..\n",
        "..#..#.....#.......#...#.#...##\n",
        ".#.........#...#......##.#...#.\n",
        ".#.#...#...#.....#.#........#.#\n",
        ".#.....###....###..##.#..##.#..\n",
        ".....##....#......#..#...#...#.\n",
        "#...##....#.......#.....##.##..\n",
        "#...#.....#.#...........#..###.\n",
        "##.#........###...........###.#\n",
        "#...#.#........#.#.....#.......\n",
        "..................#..#.........\n",
        ".....#.#..#.#......#..#.....##.\n",
        ".#.#.......#..##........#..##.#\n",
        ".#.#..#.#...#.......#.#.#..#...\n",
        "...#......#....#....##.#..#....\n",
        "......#.......##....##..#.....#\n",
        "...#.##...##...............#..#\n",
        ".###....#.#.....##..#.......#.#\n",
        "#....#..........#...........#.#\n",
        "...#...............#.#..#....#.\n",
        ".....#..##..........#..###.....\n",
        ".....####.....#.#.......#...#..\n",
        "#.............#...#.......##...\n",
        ".#....##.......#.#......#.#.##.\n",
        ".#..#.......#..##...#...#......\n",
        "#.......#..#..#..#.....#.......\n",
        "##...#.#.#...........#....#....\n",
        ".......#..#.#..............#.#.\n",
        ".....#.......#.......#.#.#.....\n",
        "....##.##.....#......#.......#.\n",
        "#...#..#.#....###....##...#.#.#\n",
        "#..#......#........#.#.#.....#.\n",
        "###..##..#......#.....#.......#\n",
        "..##....#.#.#......#..##...#...\n",
        ".....#..#....#...#.#...#...#...\n",
        ".....#.#..###.#..#...##......#.\n",
        "#.#..#....#..#.....#.#...#.#...\n",
        ".#..#....#.......#..#.......#..\n",
        "#.........#..#..#.........##..#\n",
        "..##.##..#..#...##.............\n",
        ".....###...#..#...##.#..#......\n",
        "#.##.....##..............#.....\n",
        ".......#.##.#.##...#.#.......##\n",
        "...#.#.##...#......#...........\n",
        "##.#........#.....##.....#.....\n",
        ".#.....#.............#......#..\n",
        "....#..##..#..#....#..#.#......\n",
        ".#.....#....##..##..#...##.....\n",
        ".##........#.#.#.#..........#.#\n",
        "...#.#.#..#....#...#..###.##...\n",
        ".#....#....#.#.#.#....#..#.....\n",
        "#.#.......#..#..#...........#.#\n",
        ".....#.....##..#....##.........\n",
        "....#.##..............#........\n",
        ".................#....#.......#\n",
        "...................###...#...#.\n",
        "...#.#..#..##..##....#.....#...\n",
        "#...#..........................\n",
        ".......#..#..#.#..#.....#......\n",
        "..##.#..#......#...#.##..##..#.\n",
        ".##.........#.#...........#....\n",
        "...#...#..##.#......#..#..#....\n",
        ".....#.#....#...#............#.\n",
        ".##..#.....##....#...#.........\n",
        "#......##...#...#............#.\n",
        ".....#.##...#.#....##..........\n",
        ".............#.......#.#.......\n",
        "##....#.#........#....#..##....\n",
        "....#...##.#....##..#.....#.#..\n",
        "...##..#....##......#...#......\n",
        ".####.#..#..#.#...#.#.#....#...\n",
        ".#........#.##....#.#....#.....\n",
        ".........#....##..#..#.........\n",
        "....##...#....##.............#.\n",
        "....#..##.#....#.#..#...##.....\n",
        ".....##...#..#....#......#.#...\n",
        "..........#.......#.##..#.##..#\n",
        ".......#.........#...#.##......\n",
        "....##.#.......#...............\n",
        "....#.......#..##.......##.#.##\n",
        "#.#..#.#....#.#.........###...#\n",
        ".#.##.....##..##...........#.#.\n",
        "...#....##........##...#...#...\n",
        ".#.##....#.#...#.#..#..#...#...\n",
        "#....#.##...#.#..#....#.....#..\n",
        "#..#...#........#...........#..\n",
        "...........#.......#......#..#.\n",
        "....##...#......##.....#......#\n",
        ".#.##.#.#.............#....##..\n",
        ".#...#...##.#.#........#..##.#.\n",
        ".#.#........#.#...#..#........#\n",
        ".###.#.#...#..#..#.#....#..#...\n",
        "..##..##....#.#............#...\n",
        "#..........#........#..#.#.#...\n",
        ".#...#..#..#.#.....#.....#....#\n",
        "#.....#.#.#.....####.......#...\n",
        "...#.#........#.#.###...#.....#\n",
        ".....#.....##....#....#..##...#\n",
        "..#....###.##.#..#..#....#...#.\n",
        ".....#.....#...........#.#.###.\n",
        ".....#......###...#.#...##.....\n",
        "...........###..#...#....#.#..#\n",
        "...#..###.....#....#.#...#.....\n",
        ".......#...#..#..##....#.#.#...\n",
        "...#..####.###........#.....#..\n",
        "..#......###..#..#.##..........\n",
        "....#....##..##..##.......###..\n",
        "...#.......#..#.#....##........\n",
        ".#.#.....#.#.#..........#..#..#\n",
        ".##....##....##...........##.##\n",
        "........#......#.##....##...#.#\n",
        "..#.#.....#..#....#..........#.\n",
        "...........#...............#...\n",
        ".....####.....#.....###.#..#...\n",
        "..........####..##.##.#....#...\n",
        "...#.#.#......#....#..#.#......\n",
        ".#.#......###.....#....#.......\n",
        "..#..#..#.......#..#...#.....#.\n",
        "...#............#......##...###\n",
        "......#.............#..#..###..\n",
        ".#.#......#..##.#..#..#.#...#.#\n",
        ".#.....#.......#..#...........#\n",
        "..#.###.#..#...#.##..#.##......\n",
        "....#.#........#..#.#...#.#.##.\n",
        ".#..##.#..#.#.#...##..#.#......\n",
        ".......#..#..#..#.....#.#.#..#.\n",
        ".##.###..##.....#.##..#........\n",
        "...##..............#.#.##....#.\n",
        "##....#...#...........#........\n",
        "..#........#.#.##..#.#...#..#..\n",
        ".......#.......##.#..#....#.#..\n",
        ".......#....##..#.#.#..#....#..\n",
        "..........#....#..#..#....#....\n",
        "........#.....#.#.....##.......\n",
        "........##.###.........#.#.#...\n",
        "###......####...#.#........#...\n",
        "......#........#......#.....#..\n",
        "##..#.##..##.###..#........##..\n",
        ".#..#...#............##.##..#..\n",
        "...........#..#.#..............\n",
        ".##.#.#....#...............#..#\n",
        ".........##.................#..\n",
        "#............##..##.........##.\n",
        "##........#.........#..##.#...#\n",
        "........#.....#..#.........#.##\n",
        "....#......#.....##.##.........\n",
        ".#.#.....#.#..#..##....#....#..\n",
        ".###...#..##....##.....#.#..##.\n",
        ".#....#.#.......#..#..#...###..\n",
        "..#.#......#.#..#.....###.....#\n",
        "#....#..#...#.....#.......#.##.\n",
        ".#.......##.#.#...#......#.....\n",
        "###....#.#......#....#.#...##..\n",
        "...#....####.......##.....#..#.\n",
        ".#.................#.......##.#\n",
        "...#.###..........#..##......#.\n",
        ".....##.#..............##..#...\n",
        "#.....#..#..........#..#.......\n",
        "..#...#.#.#.......##.#.....#...\n",
        "#........#.#........#.#.....#..\n",
        "#.....#...##....##..##........#\n",
        ".#.##..#...#....#........#..#..\n",
        "#..#.....#....#...##......#....\n",
        "...#...#...#.#.#....##....#.#.#\n",
        "......#...............#.....#..\n",
        ".......#.#..#..##.#.....#......\n",
        "...........##......#...#.......\n",
        "....##..##.....#.#...#.........\n",
        "......##..###.......#....#.##..\n",
        "......#..#.##....#..###..#.....\n",
        ".....#.........#........##.....\n",
        ".....####..................#...\n",
        ".#.#...##................#.....\n",
        ".#..#...#...#.....#.##..#..#...\n",
        ".#................#...###....#.\n",
        "...#....#...........#...#....#.\n",
        ".......#....##...............#.\n",
        ".#.#.##..##.......#....#....###\n",
        "......#.#....#...#..#..........\n",
        "....#.##.#.#...##.#.#......#...\n",
        "##....#...##.....#..#.###.#....\n",
        ".......#......#.........#.#...#\n",
        ".##.#...........##.........#.#.\n",
        "##..##.....#...#..#.#...#....#.\n",
        "#..##......###........#...#....\n",
        ".....#.#.......#...#..##....##.\n",
        ".....#.#..#.....#.......#..##..\n",
        "...#..#..............#.#.......\n",
        ".#.#...###......###............\n",
        ".....#.....#...#.###...........\n",
        ".......#..................#...#\n",
        "#....#...#...#....#....#.#....#\n",
        "....#..#............#.#........\n",
        "#....#..........#.#.#..#..#....\n",
        ".......#....#......#....#......\n",
        ".##.#.#...#...#...##...........\n",
        ".........###.#..#..............\n",
        "...#........##....#....#....###\n",
        "....##..#.......#.#...#.#..#.#.\n",
        ".....##....#..##.........#.....\n",
        "........##..#........#.........\n",
        "...........#..#.##..##...#.....\n",
        ".........#.#..####..#...#.##.##\n",
        "##..#.#.....##.....#.........#.\n",
        "..#...#...#....#.#....#.....#.#\n",
        ".###.#....#.#......#....#......\n",
        ".##.....#....#.......#.#..#.##.\n",
        "#..#..##.....#....#...##.....#.\n",
        "...#.........####.........##..#\n",
        "..#.....#....###.#.#...#..#....\n",
        ".........#....#..#.#.........#.\n",
        ".....###.##..#...#.....#..#..#.\n",
        "....#......#..#.#...#.....#....\n",
        ".......#...#..#....#.......#.#.\n",
        ".#....#............#....####...\n",
        "#..##..##....#.....#...#.....#.\n",
        "...#...##...#.#....#...........\n",
        ".......#####.....#..###.#..#...\n",
        ".....#.....#.#....#.........#..\n",
        ".###.#..#...##.##.#.#..#..#....\n",
        ".#..#.##..#......#..........##.\n",
        "##....#....#.........###..#....\n",
        "..#.............#.......#.#....\n",
        "..#.....#..##...#...###..#.##..\n",
        "##...##...#.#....#..#..........\n",
        "...............#.....#.......#.\n",
        "....#.#......##.#......#...#.#.\n",
        ".........#.............#.#.....\n",
        "...#.#.........................\n",
        "..#..#....##..#...###.##.......\n",
        "...##.#...##..#.#.##.#...#.##..\n",
        ".##....#....#.......#.....##..#\n",
        ".#...............#..#...#......\n",
        "...##.....##.###....#.....#...#\n",
        "...#.....#...####....##....#..#\n",
        "..#.#.###..##.....#........#...\n",
        ".....##.##..#..#.........##....\n",
        "........##.....#..........#.##.\n",
        "..##.#....####..#...........#..\n",
        "##....#..#..#.#.##.....#...##..\n",
        "...#...#......#..#.#....#......\n",
        "##.....##.......##.##....#.....\n",
        ".........#...##...........#....\n",
        ".###.#..#....##...#.....#.....#\n",
        "...#..........#.###..##...#.##.\n",
        "...#..#....#.#.#.......###.....\n",
        "....#..#.#.....##...#.#.#.#....\n",
        ".......##..#...##..##.#....#...\n",
        ".##..........#.#.#.....#.....#.\n",
        "#....#......##...#..##.#..#..#.\n",
        ".#...#.....###..#......#.....#.\n",
        ".#..#.###.......#.##....###....\n",
        "#....#....#....#....#..#.##....\n",
        "..#..#.....#.....#....###.....#\n",
        "##.###..#...##.......#.#.......\n",
        "#...##......##....#.#...#....#.\n",
        "..##.#.#....#...#.....##.......\n",
        ".#....#..#...#...##..##........\n",
        "###......#.##....#.#..##.......\n",
        "...#.....##.#.........#..#.....\n",
        ".......#.#....#.....##......#..\n",
        "#..#..............##.#........#\n",
        "....#.#....#..#.#...##.........\n",
        "..........#..........#.........\n",
        ".#.....#.....#.##....#.##..#..#\n",
        ".......#.......#.#.#.##....#...\n",
        "..#...........#....#.......##..\n",
        "..#.#.#.#...##..#.#.#..#...#.#.\n",
        "..#..#..........#...##....#....\n",
        "....##.#....#.............#....\n",
        ".##...##..........#.#..#...#...\n",
        "#..#..#.##..........##.........\n",
        "............#.......#..#.....##\n",
        "....#......#........#..#.##....\n",
        "#.#......#.#...#.....#.........\n",
        "..#.....#..#..........#.....#..\n",
        ".#..#.#.#.##...#..#.#.........#\n",
        "#...##....#..##..#...#.#.##....\n",
        "#..##.##.#.##.......#.......#..\n",
        "#.#.......#........#.##....#.#.\n",
        "....#..##....##..##......#..##.\n",
        "#.....#....#.#........####.....\n",
        "......#...#...###..#...........\n",
        ".##.#.##...##................##\n",
        "..##.#.....#.#..#......#.#.....\n",
        "......#...#........#.....#.##..\n",
        "#..#.#..#.....#.#..#..##..#.#..\n",
        "...#.........#.#.#.##...#......\n",
        "...#..##....#..#.#....#.###.#..\n",
        "........###................##.#\n",
        "##...........#......##.##.....#\n",
        ".#.#.#....#....#....#..........\n",
        "#.....#........................\n",
        "....#.....#...#......#.........\n",
        "....#.#..#..............#......\n",
        "##.........#..#....#.......#...\n",
        ".###....#..#.#.####.........#..\n",
        "..#.#....#.....###..#..........\n",
        "..........#................#.##'''"
      ],
      "execution_count": 6,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "Ssq2I6LV7QsQ"
      },
      "source": [
        "from itertools import *\n",
        "import numpy as np\n",
        "import networkx as nx\n",
        "import re"
      ],
      "execution_count": 1,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "SSavSblQ8z96"
      },
      "source": [
        "def fun(input):\n",
        "  pos = 0\n",
        "  trees = 0\n",
        "  for line in input.split('\\n'):\n",
        "    if line[pos] == '#':\n",
        "      trees += 1\n",
        "    pos +=3\n",
        "    if pos >= len(line):\n",
        "      pos -= len(line)\n",
        "  return trees\n"
      ],
      "execution_count": 12,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "G863tyJ3-6UX",
        "outputId": "cca13054-f255-4306-ca75-fbde7207b61a"
      },
      "source": [
        "fun(input)"
      ],
      "execution_count": 13,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "211"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 13
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "3Brk8juF-7yE"
      },
      "source": [
        "def fun2(input, x, y):\n",
        "  pos = 0\n",
        "  trees = 0\n",
        "  for lineno, line in enumerate(input.split('\\n')):\n",
        "    if lineno % y == 0:\n",
        "      if line[pos] == '#':\n",
        "        trees += 1\n",
        "      pos += x\n",
        "      if pos >= len(line):\n",
        "        pos -= len(line)\n",
        "  return trees\n"
      ],
      "execution_count": 14,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "AkgedZqC_q1o",
        "outputId": "7c4f727b-0e9f-49be-de75-d088918ff4d5"
      },
      "source": [
        "fun2(input, 1, 1)*fun2(input, 3, 1)*fun2(input, 5, 1)*fun2(input, 7, 1)*fun2(input, 1, 2)"
      ],
      "execution_count": 17,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "3584591857"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 17
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "3A_leI7W_t5_"
      },
      "source": [
        "input2 = r'''Right 1, down 1.\n",
        "Right 3, down 1. (This is the slope you already checked.)\n",
        "Right 5, down 1.\n",
        "Right 7, down 1.\n",
        "Right 1, down 2.'''"
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}