Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add binary-search-tree exercise #161

Merged
merged 1 commit into from
Oct 24, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions config.json
Original file line number Diff line number Diff line change
Expand Up @@ -555,6 +555,14 @@
"prerequisites": [],
"difficulty": 4
},
{
"slug": "binary-search-tree",
"name": "Binary Search Tree",
"uuid": "c0c96f3d-833e-4d15-b4ae-ccc711a0e165",
"practices": [],
"prerequisites": [],
"difficulty": 4
},
{
"slug": "custom-set",
"name": "Custom Set",
Expand Down
47 changes: 47 additions & 0 deletions exercises/practice/binary-search-tree/.docs/instructions.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
# Instructions

Insert and search for numbers in a binary tree.

When we need to represent sorted data, an array does not make a good data structure.

Say we have the array `[1, 3, 4, 5]`, and we add 2 to it so it becomes `[1, 3, 4, 5, 2]`.
Now we must sort the entire array again!
We can improve on this by realizing that we only need to make space for the new item `[1, nil, 3, 4, 5]`, and then adding the item in the space we added.
But this still requires us to shift many elements down by one.

Binary Search Trees, however, can operate on sorted data much more efficiently.

A binary search tree consists of a series of connected nodes.
Each node contains a piece of data (e.g. the number 3), a variable named `left`, and a variable named `right`.
The `left` and `right` variables point at `nil`, or other nodes.
Since these other nodes in turn have other nodes beneath them, we say that the left and right variables are pointing at subtrees.
All data in the left subtree is less than or equal to the current node's data, and all data in the right subtree is greater than the current node's data.

For example, if we had a node containing the data 4, and we added the data 2, our tree would look like this:

4
/
2

If we then added 6, it would look like this:

4
/ \
2 6

If we then added 3, it would look like this

4
/ \
2 6
\
3

And if we then added 1, 5, and 7, it would look like this

4
/ \
/ \
2 6
/ \ / \
1 3 5 7
24 changes: 24 additions & 0 deletions exercises/practice/binary-search-tree/.meta/Example.roc
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
module [fromList, toList]

BinaryTree : [Nil, Node { value : U64, left : BinaryTree, right : BinaryTree }]

fromList : List U64 -> BinaryTree
fromList = \data ->
data |> List.walk Nil insert

insert : BinaryTree, U64 -> BinaryTree
insert = \tree, value ->
when tree is
Nil -> Node { value, left: Nil, right: Nil }
Node node ->
if value <= node.value then
Node { node & left: (node.left |> insert value) }
else
Node { node & right: (node.right |> insert value) }

toList : BinaryTree -> List U64
toList = \tree ->
when tree is
Nil -> []
Node node ->
node.left |> toList |> List.append node.value |> List.concat (node.right |> toList)
18 changes: 18 additions & 0 deletions exercises/practice/binary-search-tree/.meta/config.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
{
"authors": [
"ageron"
],
"files": {
"solution": [
"BinarySearchTree.roc"
],
"test": [
"binary-search-tree-test.roc"
],
"example": [
".meta/Example.roc"
]
},
"blurb": "Insert and search for numbers in a binary tree.",
"source": "Josh Cheek"
}
50 changes: 50 additions & 0 deletions exercises/practice/binary-search-tree/.meta/template.j2
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
{%- import "generator_macros.j2" as macros with context -%}
{{ macros.canonical_ref() }}
{{ macros.header() }}

import {{ exercise | to_pascal }} exposing [fromList, toList]

{% macro to_int_list(list) -%}
[{% for item in list %}{{ item }},{% endfor %}]
{%- endmacro %}

{% macro to_tree(tree) -%}
{% if tree is none %}Nil{% else %}Node {
value: {{ tree["data"] }},
left: {{ to_tree(tree["left"]) }},
right: {{ to_tree(tree["right"]) }},
}{% endif %}
{%- endmacro %}

{% for supercase in cases %}
##
## {{ supercase["description"] }}
##
{%- if supercase["cases"] %}
{%- set subcases = supercase["cases"] %}
{%- else %}
{%- set subcases = [supercase] %}
{%- endif %}

{% for subcase in subcases -%}
{%- if subcase["description"] != supercase["description"] %}
# {{ subcase["description"] }}
{%- endif %}
expect
data = {{ to_int_list(subcase["input"]["treeData"]) }}
{%- if subcase["property"] == "data" %}
result = data |> fromList
expected = {{ to_tree(subcase["expected"]) }}
result == expected
{%- else %}
tree = data |> fromList
result = tree |> toList
expected = {{ to_int_list(subcase["expected"]) }}
result == expected
{%- endif %}

{% endfor %}

{% endfor %}


40 changes: 40 additions & 0 deletions exercises/practice/binary-search-tree/.meta/tests.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# This is an auto-generated file.
#
# Regenerating this file via `configlet sync` will:
# - Recreate every `description` key/value pair
# - Recreate every `reimplements` key/value pair, where they exist in problem-specifications
# - Remove any `include = true` key/value pair (an omitted `include` key implies inclusion)
# - Preserve any other key/value pair
#
# As user-added comments (using the # character) will be removed when this file
# is regenerated, comments can be added via a `comment` key.

[e9c93a78-c536-4750-a336-94583d23fafa]
description = "data is retained"

[7a95c9e8-69f6-476a-b0c4-4170cb3f7c91]
description = "insert data at proper node -> smaller number at left node"

[22b89499-9805-4703-a159-1a6e434c1585]
description = "insert data at proper node -> same number at left node"

[2e85fdde-77b1-41ed-b6ac-26ce6b663e34]
description = "insert data at proper node -> greater number at right node"

[dd898658-40ab-41d0-965e-7f145bf66e0b]
description = "can create complex tree"

[9e0c06ef-aeca-4202-b8e4-97f1ed057d56]
description = "can sort data -> can sort single number"

[425e6d07-fceb-4681-a4f4-e46920e380bb]
description = "can sort data -> can sort if second number is smaller than first"

[bd7532cc-6988-4259-bac8-1d50140079ab]
description = "can sort data -> can sort if second number is same as first"

[b6d1b3a5-9d79-44fd-9013-c83ca92ddd36]
description = "can sort data -> can sort if second number is greater than first"

[d00ec9bd-1288-4171-b968-d44d0808c1c8]
description = "can sort data -> can sort complex tree"
11 changes: 11 additions & 0 deletions exercises/practice/binary-search-tree/BinarySearchTree.roc
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
module [fromList, toList]

BinaryTree : [Nil, Node { value : U64, left : BinaryTree, right : BinaryTree }]

fromList : List U64 -> BinaryTree
fromList = \data ->
crash "Please implement the 'fromList' function"

toList : BinaryTree -> List U64
toList = \tree ->
crash "Please implement the 'toList' function"
157 changes: 157 additions & 0 deletions exercises/practice/binary-search-tree/binary-search-tree-test.roc
Original file line number Diff line number Diff line change
@@ -0,0 +1,157 @@
# These tests are auto-generated with test data from:
# https://github.com/exercism/problem-specifications/tree/main/exercises/binary-search-tree/canonical-data.json
# File last updated on 2024-10-21
app [main] {
pf: platform "https://github.com/roc-lang/basic-cli/releases/download/0.15.0/SlwdbJ-3GR7uBWQo6zlmYWNYOxnvo8r6YABXD-45UOw.tar.br",
}

main =
Task.ok {}

import BinarySearchTree exposing [fromList, toList]

##
## data is retained
##

expect
data = [4]
result = data |> fromList
expected = Node {
value: 4,
left: Nil,
right: Nil,
}
result == expected

##
## insert data at proper node
##

# smaller number at left node
expect
data = [4, 2]
result = data |> fromList
expected = Node {
value: 4,
left: Node {
value: 2,
left: Nil,
right: Nil,
},
right: Nil,
}
result == expected

# same number at left node
expect
data = [4, 4]
result = data |> fromList
expected = Node {
value: 4,
left: Node {
value: 4,
left: Nil,
right: Nil,
},
right: Nil,
}
result == expected

# greater number at right node
expect
data = [4, 5]
result = data |> fromList
expected = Node {
value: 4,
left: Nil,
right: Node {
value: 5,
left: Nil,
right: Nil,
},
}
result == expected

##
## can create complex tree
##

expect
data = [4, 2, 6, 1, 3, 5, 7]
result = data |> fromList
expected = Node {
value: 4,
left: Node {
value: 2,
left: Node {
value: 1,
left: Nil,
right: Nil,
},
right: Node {
value: 3,
left: Nil,
right: Nil,
},
},
right: Node {
value: 6,
left: Node {
value: 5,
left: Nil,
right: Nil,
},
right: Node {
value: 7,
left: Nil,
right: Nil,
},
},
}
result == expected

##
## can sort data
##

# can sort single number
expect
data = [2]
tree = data |> fromList
result = tree |> toList
expected = [2]
result == expected

# can sort if second number is smaller than first
expect
data = [2, 1]
tree = data |> fromList
result = tree |> toList
expected = [1, 2]
result == expected

# can sort if second number is same as first
expect
data = [2, 2]
tree = data |> fromList
result = tree |> toList
expected = [2, 2]
result == expected

# can sort if second number is greater than first
expect
data = [2, 3]
tree = data |> fromList
result = tree |> toList
expected = [2, 3]
result == expected

# can sort complex tree
expect
data = [2, 1, 3, 6, 7, 5]
tree = data |> fromList
result = tree |> toList
expected = [1, 2, 3, 5, 6, 7]
result == expected