lean4-htt/tests/lean/run/tree_height.lean
2016-07-31 12:50:11 -07:00

36 lines
1.1 KiB
Text

import logic data.nat
open nat algebra
inductive tree (A : Type) :=
| leaf : A → tree A
| node : tree A → tree A → tree A
namespace tree
definition height {A : Type} (t : tree A) : nat :=
tree.rec_on t
(λ a, zero)
(λ t₁ t₂ h₁ h₂, succ (max h₁ h₂))
definition height_lt {A : Type} : tree A → tree A → Prop :=
inv_image lt (@height A)
definition height_lt.wf (A : Type) : well_founded (@height_lt A) :=
inv_image.wf height lt.wf
theorem height_lt.node_left {A : Type} (t₁ t₂ : tree A) : height_lt t₁ (node t₁ t₂) :=
lt_succ_of_le (le_max_left (height t₁) (height t₂))
theorem height_lt.node_right {A : Type} (t₁ t₂ : tree A) : height_lt t₂ (node t₁ t₂) :=
lt_succ_of_le (le_max_right (height t₁) (height t₂))
theorem height_lt.trans {A : Type} : transitive (@height_lt A) :=
inv_image.trans lt height @nat.lt_trans
example : height_lt (leaf (2:nat)) (node (leaf 1) (leaf 2)) :=
height_lt.node_right _ _
example : height_lt (leaf (2:nat)) (node (node (leaf 1) (leaf 2)) (leaf 3)) :=
height_lt.trans (height_lt.node_right _ _) (height_lt.node_left _ _)
end tree