mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
224 lines
5.7 KiB
Python
224 lines
5.7 KiB
Python
|
{
|
||
|
"cells": [
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"This notebook was prepared by [Donne Martin](https://github.com/donnemartin). Source and license info is on [GitHub](https://github.com/donnemartin/interactive-coding-challenges)."
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"# Solution Notebook"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"## Problem: Island Perimeter.\n",
|
||
|
"\n",
|
||
|
"See the [LeetCode](https://leetcode.com/problems/island-perimeter/) problem page.\n",
|
||
|
"\n",
|
||
|
"You are given a map in form of a two-dimensional integer grid where 1 represents land and 0 represents water. Grid cells are connected horizontally/vertically (not diagonally). The grid is completely surrounded by water, and there is exactly one island (i.e., one or more connected land cells). The island doesn't have \"lakes\" (water inside that isn't connected to the water around the island). One cell is a square with side length 1. The grid is rectangular, width and height don't exceed 100. Determine the perimeter of the island.\n",
|
||
|
"\n",
|
||
|
"Example:\n",
|
||
|
"\n",
|
||
|
"<pre>\n",
|
||
|
"[[0,1,0,0],\n",
|
||
|
" [1,1,1,0],\n",
|
||
|
" [0,1,0,0],\n",
|
||
|
" [1,1,0,0]]\n",
|
||
|
"</pre>\n",
|
||
|
"\n",
|
||
|
"Answer: 16\n",
|
||
|
"\n",
|
||
|
"* [Constraints](#Constraints)\n",
|
||
|
"* [Test Cases](#Test-Cases)\n",
|
||
|
"* [Algorithm](#Algorithm)\n",
|
||
|
"* [Code](#Code)\n",
|
||
|
"* [Unit Test](#Unit-Test)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"## Constraints\n",
|
||
|
"\n",
|
||
|
"* Can we assume the inputs are valid?\n",
|
||
|
" * No, check for None\n",
|
||
|
"* Can we assume this fits memory?\n",
|
||
|
" * Yes"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"## Test Cases\n",
|
||
|
"\n",
|
||
|
"<pre>\n",
|
||
|
"* None -> TypeError\n",
|
||
|
"* [[1, 0]] -> 4\n",
|
||
|
"* [[0, 1, 0, 0],\n",
|
||
|
" [1, 1, 1, 0],\n",
|
||
|
" [0, 1, 0, 0],\n",
|
||
|
" [1, 1, 0, 0]] -> 16\n",
|
||
|
"</pre>"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"## Algorithm\n",
|
||
|
"\n",
|
||
|
"For each cell in the grid:\n",
|
||
|
"* Check left, right, up, down\n",
|
||
|
" * For each check, if we are at the edge or the cell we are checking is land, increment sides\n",
|
||
|
"\n",
|
||
|
"Complexity:\n",
|
||
|
"* Time: O(rows * cols)\n",
|
||
|
"* Space: O(1)"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"## Code"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
|
"execution_count": 1,
|
||
|
"metadata": {
|
||
|
"collapsed": false
|
||
|
},
|
||
|
"outputs": [],
|
||
|
"source": [
|
||
|
"class Solution(object):\n",
|
||
|
"\n",
|
||
|
" def island_perimeter(self, grid):\n",
|
||
|
" if grid is None:\n",
|
||
|
" raise TypeError('grid cannot be None')\n",
|
||
|
" sides = 0\n",
|
||
|
" num_rows = len(grid)\n",
|
||
|
" num_cols = len(grid[0])\n",
|
||
|
" for i in range(num_rows):\n",
|
||
|
" for j in range(num_cols):\n",
|
||
|
" if grid[i][j] == 1:\n",
|
||
|
" # Check left\n",
|
||
|
" if j == 0 or grid[i][j - 1] == 0:\n",
|
||
|
" sides += 1\n",
|
||
|
" # Check right\n",
|
||
|
" if j == num_cols - 1 or grid[i][j + 1] == 0:\n",
|
||
|
" sides += 1\n",
|
||
|
" # Check up\n",
|
||
|
" if i == 0 or grid[i - 1][j] == 0:\n",
|
||
|
" sides += 1\n",
|
||
|
" # Check down\n",
|
||
|
" if i == num_rows - 1 or grid[i + 1][j] == 0:\n",
|
||
|
" sides += 1\n",
|
||
|
" return sides"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "markdown",
|
||
|
"metadata": {},
|
||
|
"source": [
|
||
|
"## Unit Test"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
|
"execution_count": 2,
|
||
|
"metadata": {
|
||
|
"collapsed": false
|
||
|
},
|
||
|
"outputs": [
|
||
|
{
|
||
|
"name": "stdout",
|
||
|
"output_type": "stream",
|
||
|
"text": [
|
||
|
"Overwriting test_island_perimeter.py\n"
|
||
|
]
|
||
|
}
|
||
|
],
|
||
|
"source": [
|
||
|
"%%writefile test_island_perimeter.py\n",
|
||
|
"from nose.tools import assert_equal, assert_raises\n",
|
||
|
"\n",
|
||
|
"\n",
|
||
|
"class TestIslandPerimeter(object):\n",
|
||
|
"\n",
|
||
|
" def test_island_perimeter(self):\n",
|
||
|
" solution = Solution()\n",
|
||
|
" assert_raises(TypeError, solution.island_perimeter, None)\n",
|
||
|
" data = [[1, 0]]\n",
|
||
|
" expected = 4\n",
|
||
|
" assert_equal(solution.island_perimeter(data), expected)\n",
|
||
|
" data = [[0, 1, 0, 0],\n",
|
||
|
" [1, 1, 1, 0],\n",
|
||
|
" [0, 1, 0, 0],\n",
|
||
|
" [1, 1, 0, 0]]\n",
|
||
|
" expected = 16\n",
|
||
|
" assert_equal(solution.island_perimeter(data), expected)\n",
|
||
|
" print('Success: test_island_perimeter')\n",
|
||
|
"\n",
|
||
|
"\n",
|
||
|
"def main():\n",
|
||
|
" test = TestIslandPerimeter()\n",
|
||
|
" test.test_island_perimeter()\n",
|
||
|
"\n",
|
||
|
"\n",
|
||
|
"if __name__ == '__main__':\n",
|
||
|
" main()"
|
||
|
]
|
||
|
},
|
||
|
{
|
||
|
"cell_type": "code",
|
||
|
"execution_count": 3,
|
||
|
"metadata": {
|
||
|
"collapsed": false
|
||
|
},
|
||
|
"outputs": [
|
||
|
{
|
||
|
"name": "stdout",
|
||
|
"output_type": "stream",
|
||
|
"text": [
|
||
|
"Success: test_island_perimeter\n"
|
||
|
]
|
||
|
}
|
||
|
],
|
||
|
"source": [
|
||
|
"%run -i test_island_perimeter.py"
|
||
|
]
|
||
|
}
|
||
|
],
|
||
|
"metadata": {
|
||
|
"kernelspec": {
|
||
|
"display_name": "Python 3",
|
||
|
"language": "python",
|
||
|
"name": "python3"
|
||
|
},
|
||
|
"language_info": {
|
||
|
"codemirror_mode": {
|
||
|
"name": "ipython",
|
||
|
"version": 3
|
||
|
},
|
||
|
"file_extension": ".py",
|
||
|
"mimetype": "text/x-python",
|
||
|
"name": "python",
|
||
|
"nbconvert_exporter": "python",
|
||
|
"pygments_lexer": "ipython3",
|
||
|
"version": "3.4.3"
|
||
|
}
|
||
|
},
|
||
|
"nbformat": 4,
|
||
|
"nbformat_minor": 0
|
||
|
}
|