mirror of
https://github.com/leanprover/lean4.git
synced 2026-04-08 05:04:07 +00:00
Compare commits
10 Commits
array_repl
...
array_find
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
eafd50d116 | ||
|
|
62e5bc7f72 | ||
|
|
988806604b | ||
|
|
350bad4a4a | ||
|
|
e20be80682 | ||
|
|
b0cf6c5b7a | ||
|
|
14939a4a5f | ||
|
|
4e3f86eadb | ||
|
|
513d53d15c | ||
|
|
8098270af3 |
275
src/Init/Data/Array/Find.lean
Normal file
275
src/Init/Data/Array/Find.lean
Normal file
@@ -0,0 +1,275 @@
|
|||||||
|
/-
|
||||||
|
Copyright (c) 2024 Lean FRO, LLC. All rights reserved.
|
||||||
|
Released under Apache 2.0 license as described in the file LICENSE.
|
||||||
|
Authors: Kim Morrison
|
||||||
|
-/
|
||||||
|
prelude
|
||||||
|
import Init.Data.List.Find
|
||||||
|
import Init.Data.Array.Lemmas
|
||||||
|
import Init.Data.Array.Attach
|
||||||
|
|
||||||
|
/-!
|
||||||
|
# Lemmas about `Array.findSome?`, `Array.find?`.
|
||||||
|
-/
|
||||||
|
|
||||||
|
namespace Array
|
||||||
|
|
||||||
|
open Nat
|
||||||
|
|
||||||
|
/-! ### findSome? -/
|
||||||
|
|
||||||
|
@[simp] theorem findSomeRev?_push_of_isSome (l : Array α) (h : (f a).isSome) : (l.push a).findSomeRev? f = f a := by
|
||||||
|
cases l; simp_all
|
||||||
|
|
||||||
|
@[simp] theorem findSomeRev?_push_of_isNone (l : Array α) (h : (f a).isNone) : (l.push a).findSomeRev? f = l.findSomeRev? f := by
|
||||||
|
cases l; simp_all
|
||||||
|
|
||||||
|
theorem exists_of_findSome?_eq_some {f : α → Option β} {l : Array α} (w : l.findSome? f = some b) :
|
||||||
|
∃ a, a ∈ l ∧ f a = b := by
|
||||||
|
cases l; simp_all [List.exists_of_findSome?_eq_some]
|
||||||
|
|
||||||
|
@[simp] theorem findSome?_eq_none_iff : findSome? p l = none ↔ ∀ x ∈ l, p x = none := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
@[simp] theorem findSome?_isSome_iff {f : α → Option β} {l : Array α} :
|
||||||
|
(l.findSome? f).isSome ↔ ∃ x, x ∈ l ∧ (f x).isSome := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
theorem findSome?_eq_some_iff {f : α → Option β} {l : Array α} {b : β} :
|
||||||
|
l.findSome? f = some b ↔ ∃ (l₁ : Array α) (a : α) (l₂ : Array α), l = l₁.push a ++ l₂ ∧ f a = some b ∧ ∀ x ∈ l₁, f x = none := by
|
||||||
|
cases l
|
||||||
|
simp only [List.findSome?_toArray, List.findSome?_eq_some_iff]
|
||||||
|
constructor
|
||||||
|
· rintro ⟨l₁, a, l₂, rfl, h₁, h₂⟩
|
||||||
|
exact ⟨l₁.toArray, a, l₂.toArray, by simp_all⟩
|
||||||
|
· rintro ⟨l₁, a, l₂, h₀, h₁, h₂⟩
|
||||||
|
exact ⟨l₁.toList, a, l₂.toList, by simpa using congrArg toList h₀, h₁, by simpa⟩
|
||||||
|
|
||||||
|
@[simp] theorem findSome?_guard (l : Array α) : findSome? (Option.guard fun x => p x) l = find? p l := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
@[simp] theorem getElem?_zero_filterMap (f : α → Option β) (l : Array α) : (l.filterMap f)[0]? = l.findSome? f := by
|
||||||
|
cases l; simp [← List.head?_eq_getElem?]
|
||||||
|
|
||||||
|
@[simp] theorem getElem_zero_filterMap (f : α → Option β) (l : Array α) (h) :
|
||||||
|
(l.filterMap f)[0] = (l.findSome? f).get (by cases l; simpa [List.length_filterMap_eq_countP] using h) := by
|
||||||
|
cases l; simp [← List.head_eq_getElem, ← getElem?_zero_filterMap]
|
||||||
|
|
||||||
|
@[simp] theorem back?_filterMap (f : α → Option β) (l : Array α) : (l.filterMap f).back? = l.findSomeRev? f := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
@[simp] theorem back!_filterMap [Inhabited β] (f : α → Option β) (l : Array α) :
|
||||||
|
(l.filterMap f).back! = (l.findSomeRev? f).getD default := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
@[simp] theorem map_findSome? (f : α → Option β) (g : β → γ) (l : Array α) :
|
||||||
|
(l.findSome? f).map g = l.findSome? (Option.map g ∘ f) := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
theorem findSome?_map (f : β → γ) (l : Array β) : findSome? p (l.map f) = l.findSome? (p ∘ f) := by
|
||||||
|
cases l; simp [List.findSome?_map]
|
||||||
|
|
||||||
|
theorem findSome?_append {l₁ l₂ : Array α} : (l₁ ++ l₂).findSome? f = (l₁.findSome? f).or (l₂.findSome? f) := by
|
||||||
|
cases l₁; cases l₂; simp [List.findSome?_append]
|
||||||
|
|
||||||
|
theorem getElem?_zero_flatten (L : Array (Array α)) :
|
||||||
|
(flatten L)[0]? = L.findSome? fun l => l[0]? := by
|
||||||
|
cases L using array_array_induction
|
||||||
|
simp [← List.head?_eq_getElem?, List.head?_flatten, List.findSome?_map, Function.comp_def]
|
||||||
|
|
||||||
|
theorem getElem_zero_flatten.proof {L : Array (Array α)} (h : 0 < L.flatten.size) :
|
||||||
|
(L.findSome? fun l => l[0]?).isSome := by
|
||||||
|
cases L using array_array_induction
|
||||||
|
simp only [List.findSome?_toArray, List.findSome?_map, Function.comp_def, List.getElem?_toArray,
|
||||||
|
List.findSome?_isSome_iff, List.isSome_getElem?]
|
||||||
|
simp only [flatten_toArray_map_toArray, size_toArray, List.length_flatten,
|
||||||
|
Nat.sum_pos_iff_exists_pos, List.mem_map] at h
|
||||||
|
obtain ⟨_, ⟨xs, m, rfl⟩, h⟩ := h
|
||||||
|
exact ⟨xs, m, by simpa using h⟩
|
||||||
|
|
||||||
|
theorem getElem_zero_flatten {L : Array (Array α)} (h) :
|
||||||
|
(flatten L)[0] = (L.findSome? fun l => l[0]?).get (getElem_zero_flatten.proof h) := by
|
||||||
|
have t := getElem?_zero_flatten L
|
||||||
|
simp [getElem?_eq_getElem, h] at t
|
||||||
|
simp [← t]
|
||||||
|
|
||||||
|
theorem back?_flatten {L : Array (Array α)} :
|
||||||
|
(flatten L).back? = (L.findSomeRev? fun l => l.back?) := by
|
||||||
|
cases L using array_array_induction
|
||||||
|
simp [List.getLast?_flatten, ← List.map_reverse, List.findSome?_map, Function.comp_def]
|
||||||
|
|
||||||
|
theorem findSome?_mkArray : findSome? f (mkArray n a) = if n = 0 then none else f a := by
|
||||||
|
simp [mkArray_eq_toArray_replicate, List.findSome?_replicate]
|
||||||
|
|
||||||
|
@[simp] theorem findSome?_mkArray_of_pos (h : 0 < n) : findSome? f (mkArray n a) = f a := by
|
||||||
|
simp [findSome?_mkArray, Nat.ne_of_gt h]
|
||||||
|
|
||||||
|
-- Argument is unused, but used to decide whether `simp` should unfold.
|
||||||
|
@[simp] theorem findSome?_mkArray_of_isSome (_ : (f a).isSome) :
|
||||||
|
findSome? f (mkArray n a) = if n = 0 then none else f a := by
|
||||||
|
simp [findSome?_mkArray]
|
||||||
|
|
||||||
|
@[simp] theorem findSome?_mkArray_of_isNone (h : (f a).isNone) :
|
||||||
|
findSome? f (mkArray n a) = none := by
|
||||||
|
rw [Option.isNone_iff_eq_none] at h
|
||||||
|
simp [findSome?_mkArray, h]
|
||||||
|
|
||||||
|
/-! ### find? -/
|
||||||
|
|
||||||
|
@[simp] theorem find?_singleton (a : α) (p : α → Bool) :
|
||||||
|
#[a].find? p = if p a then some a else none := by
|
||||||
|
simp [singleton_eq_toArray_singleton]
|
||||||
|
|
||||||
|
@[simp] theorem findRev?_push_of_pos (l : Array α) (h : p a) :
|
||||||
|
findRev? p (l.push a) = some a := by
|
||||||
|
cases l; simp [h]
|
||||||
|
|
||||||
|
@[simp] theorem findRev?_cons_of_neg (l : Array α) (h : ¬p a) :
|
||||||
|
findRev? p (l.push a) = findRev? p l := by
|
||||||
|
cases l; simp [h]
|
||||||
|
|
||||||
|
@[simp] theorem find?_eq_none : find? p l = none ↔ ∀ x ∈ l, ¬ p x := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
theorem find?_eq_some_iff_append {xs : Array α} :
|
||||||
|
xs.find? p = some b ↔ p b ∧ ∃ (as bs : Array α), xs = as.push b ++ bs ∧ ∀ a ∈ as, !p a := by
|
||||||
|
rcases xs with ⟨xs⟩
|
||||||
|
simp only [List.find?_toArray, List.find?_eq_some_iff_append, Bool.not_eq_eq_eq_not,
|
||||||
|
Bool.not_true, exists_and_right, and_congr_right_iff]
|
||||||
|
intro w
|
||||||
|
constructor
|
||||||
|
· rintro ⟨as, ⟨⟨x, rfl⟩, h⟩⟩
|
||||||
|
exact ⟨as.toArray, ⟨x.toArray, by simp⟩ , by simpa using h⟩
|
||||||
|
· rintro ⟨as, ⟨⟨x, h'⟩, h⟩⟩
|
||||||
|
exact ⟨as.toList, ⟨x.toList, by simpa using congrArg Array.toList h'⟩,
|
||||||
|
by simpa using h⟩
|
||||||
|
|
||||||
|
@[simp]
|
||||||
|
theorem find?_push_eq_some {xs : Array α} :
|
||||||
|
(xs.push a).find? p = some b ↔ xs.find? p = some b ∨ (xs.find? p = none ∧ (p a ∧ a = b)) := by
|
||||||
|
cases xs; simp
|
||||||
|
|
||||||
|
@[simp] theorem find?_isSome {xs : Array α} {p : α → Bool} : (xs.find? p).isSome ↔ ∃ x, x ∈ xs ∧ p x := by
|
||||||
|
cases xs; simp
|
||||||
|
|
||||||
|
theorem find?_some {xs : Array α} (h : find? p xs = some a) : p a := by
|
||||||
|
cases xs
|
||||||
|
simp at h
|
||||||
|
exact List.find?_some h
|
||||||
|
|
||||||
|
theorem mem_of_find?_eq_some {xs : Array α} (h : find? p xs = some a) : a ∈ xs := by
|
||||||
|
cases xs
|
||||||
|
simp at h
|
||||||
|
simpa using List.mem_of_find?_eq_some h
|
||||||
|
|
||||||
|
theorem get_find?_mem {xs : Array α} (h) : (xs.find? p).get h ∈ xs := by
|
||||||
|
cases xs
|
||||||
|
simp [List.get_find?_mem]
|
||||||
|
|
||||||
|
@[simp] theorem find?_filter {xs : Array α} (p q : α → Bool) :
|
||||||
|
(xs.filter p).find? q = xs.find? (fun a => p a ∧ q a) := by
|
||||||
|
cases xs; simp
|
||||||
|
|
||||||
|
@[simp] theorem getElem?_zero_filter (p : α → Bool) (l : Array α) :
|
||||||
|
(l.filter p)[0]? = l.find? p := by
|
||||||
|
cases l; simp [← List.head?_eq_getElem?]
|
||||||
|
|
||||||
|
@[simp] theorem getElem_zero_filter (p : α → Bool) (l : Array α) (h) :
|
||||||
|
(l.filter p)[0] =
|
||||||
|
(l.find? p).get (by cases l; simpa [← List.countP_eq_length_filter] using h) := by
|
||||||
|
cases l
|
||||||
|
simp [List.getElem_zero_eq_head]
|
||||||
|
|
||||||
|
@[simp] theorem back?_filter (p : α → Bool) (l : Array α) : (l.filter p).back? = l.findRev? p := by
|
||||||
|
cases l; simp
|
||||||
|
|
||||||
|
@[simp] theorem back!_filter [Inhabited α] (p : α → Bool) (l : Array α) :
|
||||||
|
(l.filter p).back! = (l.findRev? p).get! := by
|
||||||
|
cases l; simp [Option.get!_eq_getD]
|
||||||
|
|
||||||
|
@[simp] theorem find?_filterMap (xs : Array α) (f : α → Option β) (p : β → Bool) :
|
||||||
|
(xs.filterMap f).find? p = (xs.find? (fun a => (f a).any p)).bind f := by
|
||||||
|
cases xs; simp
|
||||||
|
|
||||||
|
@[simp] theorem find?_map (f : β → α) (xs : Array β) :
|
||||||
|
find? p (xs.map f) = (xs.find? (p ∘ f)).map f := by
|
||||||
|
cases xs; simp
|
||||||
|
|
||||||
|
@[simp] theorem find?_append {l₁ l₂ : Array α} :
|
||||||
|
(l₁ ++ l₂).find? p = (l₁.find? p).or (l₂.find? p) := by
|
||||||
|
cases l₁
|
||||||
|
cases l₂
|
||||||
|
simp
|
||||||
|
|
||||||
|
@[simp] theorem find?_flatten (xs : Array (Array α)) (p : α → Bool) :
|
||||||
|
xs.flatten.find? p = xs.findSome? (·.find? p) := by
|
||||||
|
cases xs using array_array_induction
|
||||||
|
simp [List.findSome?_map, Function.comp_def]
|
||||||
|
|
||||||
|
theorem find?_flatten_eq_none {xs : Array (Array α)} {p : α → Bool} :
|
||||||
|
xs.flatten.find? p = none ↔ ∀ ys ∈ xs, ∀ x ∈ ys, !p x := by
|
||||||
|
simp
|
||||||
|
|
||||||
|
/--
|
||||||
|
If `find? p` returns `some a` from `xs.flatten`, then `p a` holds, and
|
||||||
|
some array in `xs` contains `a`, and no earlier element of that array satisfies `p`.
|
||||||
|
Moreover, no earlier array in `xs` has an element satisfying `p`.
|
||||||
|
-/
|
||||||
|
theorem find?_flatten_eq_some {xs : Array (Array α)} {p : α → Bool} {a : α} :
|
||||||
|
xs.flatten.find? p = some a ↔
|
||||||
|
p a ∧ ∃ (as : Array (Array α)) (ys zs : Array α) (bs : Array (Array α)),
|
||||||
|
xs = as.push (ys.push a ++ zs) ++ bs ∧
|
||||||
|
(∀ a ∈ as, ∀ x ∈ a, !p x) ∧ (∀ x ∈ ys, !p x) := by
|
||||||
|
cases xs using array_array_induction
|
||||||
|
simp only [flatten_toArray_map_toArray, List.find?_toArray, List.find?_flatten_eq_some]
|
||||||
|
simp only [Bool.not_eq_eq_eq_not, Bool.not_true, exists_and_right, and_congr_right_iff]
|
||||||
|
intro w
|
||||||
|
constructor
|
||||||
|
· rintro ⟨as, ys, ⟨⟨zs, bs, rfl⟩, h₁, h₂⟩⟩
|
||||||
|
exact ⟨as.toArray.map List.toArray, ys.toArray,
|
||||||
|
⟨zs.toArray, bs.toArray.map List.toArray, by simp⟩, by simpa using h₁, by simpa using h₂⟩
|
||||||
|
· rintro ⟨as, ys, ⟨⟨zs, bs, h⟩, h₁, h₂⟩⟩
|
||||||
|
replace h := congrArg (·.map Array.toList) (congrArg Array.toList h)
|
||||||
|
simp [Function.comp_def] at h
|
||||||
|
exact ⟨as.toList.map Array.toList, ys.toList,
|
||||||
|
⟨zs.toList, bs.toList.map Array.toList, by simpa using h⟩,
|
||||||
|
by simpa using h₁, by simpa using h₂⟩
|
||||||
|
|
||||||
|
@[simp] theorem find?_flatMap (xs : Array α) (f : α → Array β) (p : β → Bool) :
|
||||||
|
(xs.flatMap f).find? p = xs.findSome? (fun x => (f x).find? p) := by
|
||||||
|
cases xs
|
||||||
|
simp [List.find?_flatMap, Array.flatMap_toArray]
|
||||||
|
|
||||||
|
theorem find?_flatMap_eq_none {xs : Array α} {f : α → Array β} {p : β → Bool} :
|
||||||
|
(xs.flatMap f).find? p = none ↔ ∀ x ∈ xs, ∀ y ∈ f x, !p y := by
|
||||||
|
simp
|
||||||
|
|
||||||
|
theorem find?_mkArray :
|
||||||
|
find? p (mkArray n a) = if n = 0 then none else if p a then some a else none := by
|
||||||
|
simp [mkArray_eq_toArray_replicate, List.find?_replicate]
|
||||||
|
|
||||||
|
@[simp] theorem find?_mkArray_of_length_pos (h : 0 < n) :
|
||||||
|
find? p (mkArray n a) = if p a then some a else none := by
|
||||||
|
simp [find?_mkArray, Nat.ne_of_gt h]
|
||||||
|
|
||||||
|
@[simp] theorem find?_mkArray_of_pos (h : p a) :
|
||||||
|
find? p (mkArray n a) = if n = 0 then none else some a := by
|
||||||
|
simp [find?_mkArray, h]
|
||||||
|
|
||||||
|
@[simp] theorem find?_mkArray_of_neg (h : ¬ p a) : find? p (mkArray n a) = none := by
|
||||||
|
simp [find?_mkArray, h]
|
||||||
|
|
||||||
|
-- This isn't a `@[simp]` lemma since there is already a lemma for `l.find? p = none` for any `l`.
|
||||||
|
theorem find?_mkArray_eq_none {n : Nat} {a : α} {p : α → Bool} :
|
||||||
|
(mkArray n a).find? p = none ↔ n = 0 ∨ !p a := by
|
||||||
|
simp [mkArray_eq_toArray_replicate, List.find?_replicate_eq_none, Classical.or_iff_not_imp_left]
|
||||||
|
|
||||||
|
@[simp] theorem find?_mkArray_eq_some {n : Nat} {a b : α} {p : α → Bool} :
|
||||||
|
(mkArray n a).find? p = some b ↔ n ≠ 0 ∧ p a ∧ a = b := by
|
||||||
|
simp [mkArray_eq_toArray_replicate]
|
||||||
|
|
||||||
|
@[simp] theorem get_find?_mkArray (n : Nat) (a : α) (p : α → Bool) (h) :
|
||||||
|
((mkArray n a).find? p).get h = a := by
|
||||||
|
simp [mkArray_eq_toArray_replicate]
|
||||||
|
|
||||||
|
end Array
|
||||||
@@ -601,7 +601,7 @@ theorem getElem?_mkArray (n : Nat) (v : α) (i : Nat) :
|
|||||||
|
|
||||||
/-- # mem -/
|
/-- # mem -/
|
||||||
|
|
||||||
theorem mem_toList {a : α} {l : Array α} : a ∈ l.toList ↔ a ∈ l := mem_def.symm
|
@[simp] theorem mem_toList {a : α} {l : Array α} : a ∈ l.toList ↔ a ∈ l := mem_def.symm
|
||||||
|
|
||||||
theorem not_mem_nil (a : α) : ¬ a ∈ #[] := nofun
|
theorem not_mem_nil (a : α) : ¬ a ∈ #[] := nofun
|
||||||
|
|
||||||
@@ -620,19 +620,19 @@ theorem getElem?_of_mem {a : α} {as : Array α} :
|
|||||||
|
|
||||||
@[simp] theorem mem_dite_empty_left {x : α} [Decidable p] {l : ¬ p → Array α} :
|
@[simp] theorem mem_dite_empty_left {x : α} [Decidable p] {l : ¬ p → Array α} :
|
||||||
(x ∈ if h : p then #[] else l h) ↔ ∃ h : ¬ p, x ∈ l h := by
|
(x ∈ if h : p then #[] else l h) ↔ ∃ h : ¬ p, x ∈ l h := by
|
||||||
split <;> simp_all [mem_def]
|
split <;> simp_all
|
||||||
|
|
||||||
@[simp] theorem mem_dite_empty_right {x : α} [Decidable p] {l : p → Array α} :
|
@[simp] theorem mem_dite_empty_right {x : α} [Decidable p] {l : p → Array α} :
|
||||||
(x ∈ if h : p then l h else #[]) ↔ ∃ h : p, x ∈ l h := by
|
(x ∈ if h : p then l h else #[]) ↔ ∃ h : p, x ∈ l h := by
|
||||||
split <;> simp_all [mem_def]
|
split <;> simp_all
|
||||||
|
|
||||||
@[simp] theorem mem_ite_empty_left {x : α} [Decidable p] {l : Array α} :
|
@[simp] theorem mem_ite_empty_left {x : α} [Decidable p] {l : Array α} :
|
||||||
(x ∈ if p then #[] else l) ↔ ¬ p ∧ x ∈ l := by
|
(x ∈ if p then #[] else l) ↔ ¬ p ∧ x ∈ l := by
|
||||||
split <;> simp_all [mem_def]
|
split <;> simp_all
|
||||||
|
|
||||||
@[simp] theorem mem_ite_empty_right {x : α} [Decidable p] {l : Array α} :
|
@[simp] theorem mem_ite_empty_right {x : α} [Decidable p] {l : Array α} :
|
||||||
(x ∈ if p then l else #[]) ↔ p ∧ x ∈ l := by
|
(x ∈ if p then l else #[]) ↔ p ∧ x ∈ l := by
|
||||||
split <;> simp_all [mem_def]
|
split <;> simp_all
|
||||||
|
|
||||||
/-- # get lemmas -/
|
/-- # get lemmas -/
|
||||||
|
|
||||||
@@ -1218,6 +1218,14 @@ theorem push_eq_append_singleton (as : Array α) (x) : as.push x = as ++ #[x] :=
|
|||||||
@[simp] theorem size_append (as bs : Array α) : (as ++ bs).size = as.size + bs.size := by
|
@[simp] theorem size_append (as bs : Array α) : (as ++ bs).size = as.size + bs.size := by
|
||||||
simp only [size, toList_append, List.length_append]
|
simp only [size, toList_append, List.length_append]
|
||||||
|
|
||||||
|
@[simp] theorem empty_append (as : Array α) : #[] ++ as = as := by
|
||||||
|
cases as
|
||||||
|
simp
|
||||||
|
|
||||||
|
@[simp] theorem append_empty (as : Array α) : as ++ #[] = as := by
|
||||||
|
cases as
|
||||||
|
simp
|
||||||
|
|
||||||
theorem getElem_append {as bs : Array α} (h : i < (as ++ bs).size) :
|
theorem getElem_append {as bs : Array α} (h : i < (as ++ bs).size) :
|
||||||
(as ++ bs)[i] = if h' : i < as.size then as[i] else bs[i - as.size]'(by simp at h; omega) := by
|
(as ++ bs)[i] = if h' : i < as.size then as[i] else bs[i - as.size]'(by simp at h; omega) := by
|
||||||
cases as; cases bs
|
cases as; cases bs
|
||||||
@@ -1876,6 +1884,50 @@ namespace Array
|
|||||||
induction as
|
induction as
|
||||||
simp
|
simp
|
||||||
|
|
||||||
|
/-! ### map -/
|
||||||
|
|
||||||
|
@[simp] theorem map_map {f : α → β} {g : β → γ} {as : Array α} :
|
||||||
|
(as.map f).map g = as.map (g ∘ f) := by
|
||||||
|
cases as; simp
|
||||||
|
|
||||||
|
@[simp] theorem map_id_fun : map (id : α → α) = id := by
|
||||||
|
funext l
|
||||||
|
induction l <;> simp_all
|
||||||
|
|
||||||
|
/-- `map_id_fun'` differs from `map_id_fun` by representing the identity function as a lambda, rather than `id`. -/
|
||||||
|
@[simp] theorem map_id_fun' : map (fun (a : α) => a) = id := map_id_fun
|
||||||
|
|
||||||
|
-- This is not a `@[simp]` lemma because `map_id_fun` will apply.
|
||||||
|
theorem map_id (as : Array α) : map (id : α → α) as = as := by
|
||||||
|
cases as <;> simp_all
|
||||||
|
|
||||||
|
/-- `map_id'` differs from `map_id` by representing the identity function as a lambda, rather than `id`. -/
|
||||||
|
-- This is not a `@[simp]` lemma because `map_id_fun'` will apply.
|
||||||
|
theorem map_id' (as : Array α) : map (fun (a : α) => a) as = as := map_id as
|
||||||
|
|
||||||
|
/-- Variant of `map_id`, with a side condition that the function is pointwise the identity. -/
|
||||||
|
theorem map_id'' {f : α → α} (h : ∀ x, f x = x) (as : Array α) : map f as = as := by
|
||||||
|
simp [show f = id from funext h]
|
||||||
|
|
||||||
|
theorem array_array_induction (P : Array (Array α) → Prop) (h : ∀ (xss : List (List α)), P (xss.map List.toArray).toArray)
|
||||||
|
(ass : Array (Array α)) : P ass := by
|
||||||
|
specialize h (ass.toList.map toList)
|
||||||
|
simpa [← toList_map, Function.comp_def, map_id] using h
|
||||||
|
|
||||||
|
/-! ### flatten -/
|
||||||
|
|
||||||
|
@[simp] theorem flatten_empty : flatten (#[] : Array (Array α)) = #[] := rfl
|
||||||
|
|
||||||
|
@[simp] theorem flatten_toArray_map_toArray (xss : List (List α)) :
|
||||||
|
(xss.map List.toArray).toArray.flatten = xss.flatten.toArray := by
|
||||||
|
simp [flatten]
|
||||||
|
suffices ∀ as, List.foldl (fun r a => r ++ a) as (List.map List.toArray xss) = as ++ xss.flatten.toArray by
|
||||||
|
simpa using this #[]
|
||||||
|
intro as
|
||||||
|
induction xss generalizing as with
|
||||||
|
| nil => simp
|
||||||
|
| cons xs xss ih => simp [ih]
|
||||||
|
|
||||||
/-! ### findSomeRevM?, findRevM?, findSomeRev?, findRev? -/
|
/-! ### findSomeRevM?, findRevM?, findSomeRev?, findRev? -/
|
||||||
|
|
||||||
@[simp] theorem findSomeRevM?_eq_findSomeM?_reverse
|
@[simp] theorem findSomeRevM?_eq_findSomeM?_reverse
|
||||||
@@ -1940,6 +1992,27 @@ namespace Array
|
|||||||
cases as
|
cases as
|
||||||
simp
|
simp
|
||||||
|
|
||||||
|
@[simp] theorem flatMap_empty {β} (f : α → Array β) : (#[] : Array α).flatMap f = #[] := rfl
|
||||||
|
|
||||||
|
@[simp] theorem flatMap_toArray_cons {β} (f : α → Array β) (a : α) (as : List α) :
|
||||||
|
(a :: as).toArray.flatMap f = f a ++ as.toArray.flatMap f := by
|
||||||
|
simp [flatMap]
|
||||||
|
suffices ∀ cs, List.foldl (fun bs a => bs ++ f a) (f a ++ cs) as =
|
||||||
|
f a ++ List.foldl (fun bs a => bs ++ f a) cs as by
|
||||||
|
erw [empty_append] -- Why doesn't this work via `simp`?
|
||||||
|
simpa using this #[]
|
||||||
|
intro cs
|
||||||
|
induction as generalizing cs <;> simp_all
|
||||||
|
|
||||||
|
@[simp] theorem flatMap_toArray {β} (f : α → Array β) (as : List α) :
|
||||||
|
as.toArray.flatMap f = (as.flatMap (fun a => (f a).toList)).toArray := by
|
||||||
|
induction as with
|
||||||
|
| nil => simp
|
||||||
|
| cons a as ih =>
|
||||||
|
apply ext'
|
||||||
|
simp [ih]
|
||||||
|
|
||||||
|
|
||||||
end Array
|
end Array
|
||||||
|
|
||||||
/-! ### Deprecations -/
|
/-! ### Deprecations -/
|
||||||
|
|||||||
@@ -372,6 +372,17 @@ theorem getElem?_concat_length (l : List α) (a : α) : (l ++ [a])[l.length]? =
|
|||||||
@[deprecated getElem?_concat_length (since := "2024-06-12")]
|
@[deprecated getElem?_concat_length (since := "2024-06-12")]
|
||||||
theorem get?_concat_length (l : List α) (a : α) : (l ++ [a]).get? l.length = some a := by simp
|
theorem get?_concat_length (l : List α) (a : α) : (l ++ [a]).get? l.length = some a := by simp
|
||||||
|
|
||||||
|
@[simp] theorem isSome_getElem? {l : List α} {n : Nat} : l[n]?.isSome ↔ n < l.length := by
|
||||||
|
by_cases h : n < l.length
|
||||||
|
· simp_all
|
||||||
|
· simp [h]
|
||||||
|
simp_all
|
||||||
|
|
||||||
|
@[simp] theorem isNone_getElem? {l : List α} {n : Nat} : l[n]?.isNone ↔ l.length ≤ n := by
|
||||||
|
by_cases h : n < l.length
|
||||||
|
· simp_all
|
||||||
|
· simp [h]
|
||||||
|
|
||||||
/-! ### mem -/
|
/-! ### mem -/
|
||||||
|
|
||||||
@[simp] theorem not_mem_nil (a : α) : ¬ a ∈ [] := nofun
|
@[simp] theorem not_mem_nil (a : α) : ¬ a ∈ [] := nofun
|
||||||
@@ -1025,6 +1036,10 @@ theorem getLast_eq_getElem : ∀ (l : List α) (h : l ≠ []),
|
|||||||
| _ :: _ :: _, _ => by
|
| _ :: _ :: _, _ => by
|
||||||
simp [getLast, get, Nat.succ_sub_succ, getLast_eq_getElem]
|
simp [getLast, get, Nat.succ_sub_succ, getLast_eq_getElem]
|
||||||
|
|
||||||
|
theorem getElem_length_sub_one_eq_getLast (l : List α) (h) :
|
||||||
|
l[l.length - 1] = getLast l (by cases l; simp at h; simp) := by
|
||||||
|
rw [← getLast_eq_getElem]
|
||||||
|
|
||||||
@[deprecated getLast_eq_getElem (since := "2024-07-15")]
|
@[deprecated getLast_eq_getElem (since := "2024-07-15")]
|
||||||
theorem getLast_eq_get (l : List α) (h : l ≠ []) :
|
theorem getLast_eq_get (l : List α) (h : l ≠ []) :
|
||||||
getLast l h = l.get ⟨l.length - 1, by
|
getLast l h = l.get ⟨l.length - 1, by
|
||||||
@@ -1149,6 +1164,11 @@ theorem head_eq_getElem (l : List α) (h : l ≠ []) : head l h = l[0]'(length_p
|
|||||||
| nil => simp at h
|
| nil => simp at h
|
||||||
| cons _ _ => simp
|
| cons _ _ => simp
|
||||||
|
|
||||||
|
theorem getElem_zero_eq_head (l : List α) (h) : l[0] = head l (by simpa [length_pos] using h) := by
|
||||||
|
cases l with
|
||||||
|
| nil => simp at h
|
||||||
|
| cons _ _ => simp
|
||||||
|
|
||||||
theorem head_eq_iff_head?_eq_some {xs : List α} (h) : xs.head h = a ↔ xs.head? = some a := by
|
theorem head_eq_iff_head?_eq_some {xs : List α} (h) : xs.head h = a ↔ xs.head? = some a := by
|
||||||
cases xs with
|
cases xs with
|
||||||
| nil => simp at h
|
| nil => simp at h
|
||||||
|
|||||||
@@ -1029,3 +1029,12 @@ instance decidableExistsLT [h : DecidablePred p] : DecidablePred fun n => ∃ m
|
|||||||
instance decidableExistsLE [DecidablePred p] : DecidablePred fun n => ∃ m : Nat, m ≤ n ∧ p m :=
|
instance decidableExistsLE [DecidablePred p] : DecidablePred fun n => ∃ m : Nat, m ≤ n ∧ p m :=
|
||||||
fun n => decidable_of_iff (∃ m, m < n + 1 ∧ p m)
|
fun n => decidable_of_iff (∃ m, m < n + 1 ∧ p m)
|
||||||
(exists_congr fun _ => and_congr_left' Nat.lt_succ_iff)
|
(exists_congr fun _ => and_congr_left' Nat.lt_succ_iff)
|
||||||
|
|
||||||
|
/-! ### Results about `List.sum` specialized to `Nat` -/
|
||||||
|
|
||||||
|
protected theorem sum_pos_iff_exists_pos {l : List Nat} : 0 < l.sum ↔ ∃ x ∈ l, 0 < x := by
|
||||||
|
induction l with
|
||||||
|
| nil => simp
|
||||||
|
| cons x xs ih =>
|
||||||
|
simp [← ih]
|
||||||
|
omega
|
||||||
|
|||||||
@@ -55,7 +55,9 @@ theorem get_eq_getD {fallback : α} : (o : Option α) → {h : o.isSome} → o.g
|
|||||||
theorem some_get! [Inhabited α] : (o : Option α) → o.isSome → some (o.get!) = o
|
theorem some_get! [Inhabited α] : (o : Option α) → o.isSome → some (o.get!) = o
|
||||||
| some _, _ => rfl
|
| some _, _ => rfl
|
||||||
|
|
||||||
theorem get!_eq_getD_default [Inhabited α] (o : Option α) : o.get! = o.getD default := rfl
|
theorem get!_eq_getD [Inhabited α] (o : Option α) : o.get! = o.getD default := rfl
|
||||||
|
|
||||||
|
@[deprecated get!_eq_getD (since := "2024-11-18")] abbrev get!_eq_getD_default := @get!_eq_getD
|
||||||
|
|
||||||
theorem mem_unique {o : Option α} {a b : α} (ha : a ∈ o) (hb : b ∈ o) : a = b :=
|
theorem mem_unique {o : Option α} {a b : α} (ha : a ∈ o) (hb : b ∈ o) : a = b :=
|
||||||
some.inj <| ha ▸ hb
|
some.inj <| ha ▸ hb
|
||||||
|
|||||||
Reference in New Issue
Block a user