Documentation

Mathlib.Combinatorics.Enumerative.Composition

Compositions #

A composition of a natural number n is a decomposition n = i₀ + ... + i_{k-1} of n into a sum of positive integers. Combinatorially, it corresponds to a decomposition of {0, ..., n-1} into non-empty blocks of consecutive integers, where the iⱼ are the lengths of the blocks. This notion is closely related to that of a partition of n, but in a composition of n the order of the iⱼs matters.

We implement two different structures covering these two viewpoints on compositions. The first one, made of a list of positive integers summing to n, is the main one and is called Composition n. The second one is useful for combinatorial arguments (for instance to show that the number of compositions of n is 2^(n-1)). It is given by a subset of {0, ..., n} containing 0 and n, where the elements of the subset (other than n) correspond to the leftmost points of each block. The main API is built on Composition n, and we provide an equivalence between the two types.

Main functions #

Let c : Composition n be a composition of n. Then

Compositions can also be used to split lists. Let l be a list of length n and c a composition of n.

We turn to the second viewpoint on compositions, that we realize as a finset of Fin (n+1). c : CompositionAsSet n is a structure made of a finset of Fin (n+1) called c.boundaries and proofs that it contains 0 and n. (Taking a finset of Fin n containing 0 would not make sense in the edge case n = 0, while the previous description works in all cases). The elements of this set (other than n) correspond to leftmost points of blocks. Thus, there is an equiv between Composition n and CompositionAsSet n. We only construct basic API on CompositionAsSet (notably c.length and c.blocks) to be able to construct this equiv, called compositionEquiv n. Since there is a straightforward equiv between CompositionAsSet n and finsets of {1, ..., n-1} (obtained by removing 0 and n from a CompositionAsSet and called compositionAsSetEquiv n), we deduce that CompositionAsSet n and Composition n are both fintypes of cardinality 2^(n - 1) (see compositionAsSet_card and composition_card).

Implementation details #

The main motivation for this structure and its API is in the construction of the composition of formal multilinear series, and the proof that the composition of analytic functions is analytic.

The representation of a composition as a list is very handy as lists are very flexible and already have a well-developed API.

Tags #

Composition, partition

References #

https://en.wikipedia.org/wiki/Composition_(combinatorics)

structure Composition (n : ) :

A composition of n is a list of positive integers summing to n.

Instances For
    theorem Composition.ext {n : } {x y : Composition n} (blocks : x.blocks = y.blocks) :
    x = y
    def instDecidableEqComposition.decEq {n✝ : } (x✝ x✝¹ : Composition n✝) :
    Decidable (x✝ = x✝¹)
    Equations
      Instances For
        structure CompositionAsSet (n : ) :

        Combinatorial viewpoint on a composition of n, by seeing it as non-empty blocks of consecutive integers in {0, ..., n-1}. We register every block by its left end-point, yielding a finset containing 0. As this does not make sense for n = 0, we add n to this finset, and get a finset of {0, ..., n} containing 0 and n. This is the data in the structure CompositionAsSet n.

        Instances For
          theorem CompositionAsSet.ext {n : } {x y : CompositionAsSet n} (boundaries : x.boundaries = y.boundaries) :
          x = y
          def instDecidableEqCompositionAsSet.decEq {n✝ : } (x✝ x✝¹ : CompositionAsSet n✝) :
          Decidable (x✝ = x✝¹)
          Equations
            Instances For

              Compositions #

              A composition of an integer n is a decomposition n = i₀ + ... + i_{k-1} of n into a sum of positive integers.

              @[reducible, inline]
              abbrev Composition.length {n : } (c : Composition n) :

              The length of a composition, i.e., the number of blocks in the composition.

              Equations
                Instances For

                  The blocks of a composition, seen as a function on Fin c.length. When composing analytic functions using compositions, this is the main player.

                  Equations
                    Instances For
                      @[simp]
                      theorem Composition.sum_blocksFun {n : } (c : Composition n) :
                      i : Fin c.length, c.blocksFun i = n
                      theorem Composition.one_le_blocks {n : } (c : Composition n) {i : } (h : i c.blocks) :
                      1 i
                      theorem Composition.blocks_le {n : } (c : Composition n) {i : } (h : i c.blocks) :
                      i n
                      @[simp]
                      theorem Composition.one_le_blocks' {n : } (c : Composition n) {i : } (h : i < c.length) :
                      @[simp]
                      theorem Composition.blocks_pos' {n : } (c : Composition n) (i : ) (h : i < c.length) :
                      0 < c.blocks[i]
                      @[simp]
                      theorem Composition.blocksFun_le {n : } (c : Composition n) (i : Fin c.length) :
                      @[simp]
                      theorem Composition.length_pos_iff {n : } (c : Composition n) :
                      0 < c.length 0 < n
                      theorem Composition.length_pos_of_pos {n : } (c : Composition n) :
                      0 < n0 < c.length

                      Alias of the reverse direction of Composition.length_pos_iff.

                      def Composition.sizeUpTo {n : } (c : Composition n) (i : ) :

                      The sum of the sizes of the blocks in a composition up to i.

                      Equations
                        Instances For
                          theorem Composition.sizeUpTo_succ {n : } (c : Composition n) {i : } (h : i < c.length) :
                          c.sizeUpTo (i + 1) = c.sizeUpTo i + c.blocks[i]
                          theorem Composition.sizeUpTo_succ' {n : } (c : Composition n) (i : Fin c.length) :
                          c.sizeUpTo (i + 1) = c.sizeUpTo i + c.blocksFun i
                          theorem Composition.sizeUpTo_strict_mono {n : } (c : Composition n) {i : } (h : i < c.length) :
                          c.sizeUpTo i < c.sizeUpTo (i + 1)
                          def Composition.boundary {n : } (c : Composition n) :
                          Fin (c.length + 1) ↪o Fin (n + 1)

                          The i-th boundary of a composition, i.e., the leftmost point of the i-th block. We include a virtual point at the right of the last block, to make for a nice equiv with CompositionAsSet n.

                          Equations
                            Instances For
                              def Composition.boundaries {n : } (c : Composition n) :
                              Finset (Fin (n + 1))

                              The boundaries of a composition, i.e., the leftmost point of all the blocks. We include a virtual point at the right of the last block, to make for a nice equiv with CompositionAsSet n.

                              Equations
                                Instances For

                                  To c : Composition n, one can associate a CompositionAsSet n by registering the leftmost point of each block, and adding a virtual point at the right of the last block.

                                  Equations
                                    Instances For

                                      The canonical increasing bijection between Fin (c.length + 1) and c.boundaries is exactly c.boundary.

                                      def Composition.embedding {n : } (c : Composition n) (i : Fin c.length) :

                                      Embedding the i-th block of a composition (identified with Fin (c.blocksFun i)) into Fin n at the relevant position.

                                      Equations
                                        Instances For
                                          @[simp]
                                          theorem Composition.coe_embedding {n : } (c : Composition n) (i : Fin c.length) (j : Fin (c.blocksFun i)) :
                                          ((c.embedding i) j) = c.sizeUpTo i + j
                                          theorem Composition.index_exists {n : } (c : Composition n) {j : } (h : j < n) :
                                          ∃ (i : ), j < c.sizeUpTo (i + 1) i < c.length

                                          index_exists asserts there is some i with j < c.sizeUpTo (i+1). In the next definition index we use Nat.find to produce the minimal such index.

                                          def Composition.index {n : } (c : Composition n) (j : Fin n) :

                                          c.index j is the index of the block in the composition c containing j.

                                          Equations
                                            Instances For
                                              theorem Composition.sizeUpTo_index_le {n : } (c : Composition n) (j : Fin n) :
                                              c.sizeUpTo (c.index j) j
                                              def Composition.invEmbedding {n : } (c : Composition n) (j : Fin n) :
                                              Fin (c.blocksFun (c.index j))

                                              Mapping an element j of Fin n to the element in the block containing it, identified with Fin (c.blocksFun (c.index j)) through the canonical increasing bijection.

                                              Equations
                                                Instances For
                                                  @[simp]
                                                  theorem Composition.coe_invEmbedding {n : } (c : Composition n) (j : Fin n) :
                                                  (c.invEmbedding j) = j - c.sizeUpTo (c.index j)
                                                  @[simp]
                                                  theorem Composition.embedding_comp_inv {n : } (c : Composition n) (j : Fin n) :
                                                  (c.embedding (c.index j)) (c.invEmbedding j) = j
                                                  theorem Composition.mem_range_embedding_iff {n : } (c : Composition n) {j : Fin n} {i : Fin c.length} :
                                                  j Set.range (c.embedding i) c.sizeUpTo i j j < c.sizeUpTo (↑i).succ
                                                  theorem Composition.disjoint_range {n : } (c : Composition n) {i₁ i₂ : Fin c.length} (h : i₁ i₂) :
                                                  Disjoint (Set.range (c.embedding i₁)) (Set.range (c.embedding i₂))

                                                  The embeddings of different blocks of a composition are disjoint.

                                                  @[simp]
                                                  theorem Composition.index_embedding {n : } (c : Composition n) (i : Fin c.length) (j : Fin (c.blocksFun i)) :
                                                  c.index ((c.embedding i) j) = i
                                                  theorem Composition.invEmbedding_comp {n : } (c : Composition n) (i : Fin c.length) (j : Fin (c.blocksFun i)) :
                                                  (c.invEmbedding ((c.embedding i) j)) = j
                                                  def Composition.blocksFinEquiv {n : } (c : Composition n) :
                                                  (i : Fin c.length) × Fin (c.blocksFun i) Fin n

                                                  Equivalence between the disjoint union of the blocks (each of them seen as Fin (c.blocksFun i)) with Fin n.

                                                  Equations
                                                    Instances For
                                                      theorem Composition.blocksFun_congr {n₁ n₂ : } (c₁ : Composition n₁) (c₂ : Composition n₂) (i₁ : Fin c₁.length) (i₂ : Fin c₂.length) (hn : n₁ = n₂) (hc : c₁.blocks = c₂.blocks) (hi : i₁ = i₂) :
                                                      c₁.blocksFun i₁ = c₂.blocksFun i₂
                                                      theorem Composition.sigma_eq_iff_blocks_eq {c c' : (n : ) × Composition n} :
                                                      c = c' c.snd.blocks = c'.snd.blocks

                                                      Two compositions (possibly of different integers) coincide if and only if they have the same sequence of blocks.

                                                      The composition Composition.ones #

                                                      The composition made of blocks all of size 1.

                                                      Equations
                                                        Instances For
                                                          @[simp]
                                                          theorem Composition.ones_blocksFun (n : ) (i : Fin (ones n).length) :
                                                          (ones n).blocksFun i = 1
                                                          @[simp]
                                                          theorem Composition.ones_sizeUpTo (n i : ) :
                                                          (ones n).sizeUpTo i = min i n
                                                          @[simp]
                                                          theorem Composition.ones_embedding {n : } (i : Fin (ones n).length) (h : 0 < (ones n).blocksFun i) :
                                                          ((ones n).embedding i) 0, h = i,
                                                          theorem Composition.eq_ones_iff {n : } {c : Composition n} :
                                                          c = ones n ic.blocks, i = 1
                                                          theorem Composition.ne_ones_iff {n : } {c : Composition n} :
                                                          c ones n ic.blocks, 1 < i

                                                          The composition Composition.single #

                                                          def Composition.single (n : ) (h : 0 < n) :

                                                          The composition made of a single block of size n.

                                                          Equations
                                                            Instances For
                                                              @[simp]
                                                              theorem Composition.single_length {n : } (h : 0 < n) :
                                                              (single n h).length = 1
                                                              @[simp]
                                                              theorem Composition.single_blocks {n : } (h : 0 < n) :
                                                              (single n h).blocks = [n]
                                                              @[simp]
                                                              theorem Composition.single_blocksFun {n : } (h : 0 < n) (i : Fin (single n h).length) :
                                                              (single n h).blocksFun i = n
                                                              @[simp]
                                                              theorem Composition.single_embedding {n : } (h : 0 < n) (i : Fin n) :
                                                              ((single n h).embedding 0) i = i
                                                              theorem Composition.ne_single_iff {n : } (hn : 0 < n) {c : Composition n} :
                                                              c single n hn ∀ (i : Fin c.length), c.blocksFun i < n
                                                              def Composition.cast {n m : } (c : Composition m) (hmn : m = n) :

                                                              Change n in (c : Composition n) to a propositionally equal value.

                                                              Equations
                                                                Instances For
                                                                  @[simp]
                                                                  theorem Composition.cast_blocks {n m : } (c : Composition m) (hmn : m = n) :
                                                                  (c.cast hmn).blocks = c.blocks
                                                                  @[simp]
                                                                  theorem Composition.cast_rfl {n : } (c : Composition n) :
                                                                  c.cast = c
                                                                  theorem Composition.cast_heq {n m : } (c : Composition m) (hmn : m = n) :
                                                                  c.cast hmn c
                                                                  theorem Composition.cast_eq_cast {n m : } (c : Composition m) (hmn : m = n) :
                                                                  c.cast hmn = cast c
                                                                  def Composition.append {n m : } (c₁ : Composition m) (c₂ : Composition n) :

                                                                  Append two compositions to get a composition of the sum of numbers.

                                                                  Equations
                                                                    Instances For
                                                                      @[simp]
                                                                      theorem Composition.append_blocks {n m : } (c₁ : Composition m) (c₂ : Composition n) :
                                                                      (c₁.append c₂).blocks = c₁.blocks ++ c₂.blocks

                                                                      Reverse the order of blocks in a composition.

                                                                      Equations
                                                                        Instances For
                                                                          @[simp]
                                                                          theorem Composition.reverse_inj {n : } {c₁ c₂ : Composition n} :
                                                                          c₁.reverse = c₂.reverse c₁ = c₂
                                                                          @[simp]
                                                                          theorem Composition.reverse_single {n : } (hn : 0 < n) :
                                                                          (single n hn).reverse = single n hn
                                                                          @[simp]
                                                                          theorem Composition.reverse_eq_single {n : } {hn : 0 < n} {c : Composition n} :
                                                                          c.reverse = single n hn c = single n hn
                                                                          theorem Composition.reverse_append {n m : } (c₁ : Composition m) (c₂ : Composition n) :
                                                                          (c₁.append c₂).reverse = (c₂.reverse.append c₁.reverse).cast
                                                                          @[irreducible]
                                                                          def Composition.recOnSingleAppend {motive : (n : ) → Composition nSort u_1} {n : } (c : Composition n) (zero : motive 0 (ones 0)) (single_append : (k n : ) → (c : Composition n) → motive n cmotive (k + 1 + n) ((single (k + 1) ).append c)) :
                                                                          motive n c

                                                                          Induction (recursion) principle on c : Composition _ that corresponds to the usual induction on the list of blocks of c.

                                                                          Equations
                                                                            Instances For
                                                                              def Composition.recOnAppendSingle {motive : (n : ) → Composition nSort u_1} {n : } (c : Composition n) (zero : motive 0 (ones 0)) (append_single : (k n : ) → (c : Composition n) → motive n cmotive (n + (k + 1)) (c.append (single (k + 1) ))) :
                                                                              motive n c

                                                                              Induction (recursion) principle on c : Composition _ that corresponds to the reverse induction on the list of blocks of c.

                                                                              Equations
                                                                                Instances For

                                                                                  Splitting a list #

                                                                                  Given a list of length n and a composition c of n, one can split l into c.length sublists of respective lengths c.blocksFun 0, ..., c.blocksFun (c.length-1). This is inverse to the join operation.

                                                                                  def List.splitWrtCompositionAux {α : Type u_1} :
                                                                                  List αList List (List α)

                                                                                  Auxiliary for List.splitWrtComposition.

                                                                                  Equations
                                                                                    Instances For
                                                                                      def List.splitWrtComposition {n : } {α : Type u_1} (l : List α) (c : Composition n) :
                                                                                      List (List α)

                                                                                      Given a list of length n and a composition [i₁, ..., iₖ] of n, split l into a list of k lists corresponding to the blocks of the composition, of respective lengths i₁, ..., iₖ. This makes sense mostly when n = l.length, but this is not necessary for the definition.

                                                                                      Equations
                                                                                        Instances For
                                                                                          @[simp]
                                                                                          theorem List.length_splitWrtComposition {n : } {α : Type u_1} (l : List α) (c : Composition n) :

                                                                                          When one splits a list along a composition c, the number of sublists thus created is c.length.

                                                                                          When one splits a list along a composition c, the lengths of the sublists thus created are given by the block sizes in c.

                                                                                          theorem List.getElem_splitWrtCompositionAux {α : Type u_1} (l : List α) (ns : List ) {i : } (hi : i < (l.splitWrtCompositionAux ns).length) :
                                                                                          (l.splitWrtCompositionAux ns)[i] = drop (take i ns).sum (take (take (i + 1) ns).sum l)
                                                                                          theorem List.getElem_splitWrtComposition' {n : } {α : Type u_1} (l : List α) (c : Composition n) {i : } (hi : i < (l.splitWrtComposition c).length) :
                                                                                          (l.splitWrtComposition c)[i] = drop (c.sizeUpTo i) (take (c.sizeUpTo (i + 1)) l)

                                                                                          The i-th sublist in the splitting of a list l along a composition c, is the slice of l between the indices c.sizeUpTo i and c.sizeUpTo (i+1), i.e., the indices in the i-th block of the composition.

                                                                                          theorem List.getElem_splitWrtComposition {n : } {α : Type u_1} (l : List α) (c : Composition n) (i : ) (h : i < (l.splitWrtComposition c).length) :
                                                                                          (l.splitWrtComposition c)[i] = drop (c.sizeUpTo i) (take (c.sizeUpTo (i + 1)) l)
                                                                                          @[simp]

                                                                                          If one splits a list along a composition, and then flattens the sublists, one gets back the original list.

                                                                                          @[simp]

                                                                                          If one joins a list of lists and then splits the flattening along the right composition, one gets back the original list of lists.

                                                                                          Compositions as sets #

                                                                                          Combinatorial viewpoints on compositions, seen as finite subsets of Fin (n+1) containing 0 and n, where the points of the set (other than n) correspond to the leftmost points of each block.

                                                                                          Bijection between compositions of n and subsets of {0, ..., n-2}, defined by considering the restriction of the subset to {1, ..., n-1} and shifting to the left by one.

                                                                                          Equations
                                                                                            Instances For

                                                                                              Number of blocks in a CompositionAsSet.

                                                                                              Equations
                                                                                                Instances For

                                                                                                  Canonical increasing bijection from Fin c.boundaries.card to c.boundaries.

                                                                                                  Equations
                                                                                                    Instances For

                                                                                                      Size of the i-th block in a CompositionAsSet, seen as a function on Fin c.length.

                                                                                                      Equations
                                                                                                        Instances For

                                                                                                          List of the sizes of the blocks in a CompositionAsSet.

                                                                                                          Equations
                                                                                                            Instances For

                                                                                                              Associating a Composition n to a CompositionAsSet n, by registering the sizes of the blocks as a list of positive integers.

                                                                                                              Equations
                                                                                                                Instances For

                                                                                                                  Equivalence between compositions and compositions as sets #

                                                                                                                  In this section, we explain how to go back and forth between a Composition and a CompositionAsSet, by showing that their blocks and length and boundaries correspond to each other, and construct an equivalence between them called compositionEquiv.

                                                                                                                  Equivalence between Composition n and CompositionAsSet n.

                                                                                                                  Equations
                                                                                                                    Instances For