(* Title: HOL/Library/Multiset_Permutations.thy Author: Manuel Eberl (TU München) Defines the set of permutations of a given multiset (or set), i.e. the set of all lists whose entries correspond to the multiset (resp. set). *) section \Permutations of a Multiset\ theory Multiset_Permutations imports Complex_Main Multiset Permutations begin (* TODO Move *) lemma mset_tl: "xs \ [] \ mset (tl xs) = mset xs - {#hd xs#}" by (cases xs) simp_all lemma mset_set_image_inj: assumes "inj_on f A" shows "mset_set (f ` A) = image_mset f (mset_set A)" proof (cases "finite A") case True from this and assms show ?thesis by (induction A) auto qed (insert assms, simp add: finite_image_iff) lemma multiset_remove_induct [case_names empty remove]: assumes "P {#}" "\A. A \ {#} \ (\x. x \# A \ P (A - {#x#})) \ P A" shows "P A" proof (induction A rule: full_multiset_induct) case (less A) hence IH: "P B" if "B \# A" for B using that by blast show ?case proof (cases "A = {#}") case True thus ?thesis by (simp add: assms) next case False hence "P (A - {#x#})" if "x \# A" for x using that by (intro IH) (simp add: mset_subset_diff_self) from False and this show "P A" by (rule assms) qed qed lemma map_list_bind: "map g (List.bind xs f) = List.bind xs (map g \ f)" by (simp add: List.bind_def map_concat) lemma mset_eq_mset_set_imp_distinct: "finite A \ mset_set A = mset xs \ distinct xs" proof (induction xs arbitrary: A) case (Cons x xs A) from Cons.prems(2) have "x \# mset_set A" by simp with Cons.prems(1) have [simp]: "x \ A" by simp from Cons.prems have "x \# mset_set (A - {x})" by simp also from Cons.prems have "mset_set (A - {x}) = mset_set A - {#x#}" by (subst mset_set_Diff) simp_all also have "mset_set A = mset (x#xs)" by (simp add: Cons.prems) also have "\ - {#x#} = mset xs" by simp finally have [simp]: "x \ set xs" by (simp add: in_multiset_in_set) from Cons.prems show ?case by (auto intro!: Cons.IH[of "A - {x}"] simp: mset_set_Diff) qed simp_all (* END TODO *) subsection \Permutations of a multiset\ definition permutations_of_multiset :: "'a multiset \ 'a list set" where "permutations_of_multiset A = {xs. mset xs = A}" lemma permutations_of_multisetI: "mset xs = A \ xs \ permutations_of_multiset A" by (simp add: permutations_of_multiset_def) lemma permutations_of_multisetD: "xs \ permutations_of_multiset A \ mset xs = A" by (simp add: permutations_of_multiset_def) lemma permutations_of_multiset_Cons_iff: "x # xs \ permutations_of_multiset A \ x \# A \ xs \ permutations_of_multiset (A - {#x#})" by (auto simp: permutations_of_multiset_def) lemma permutations_of_multiset_empty [simp]: "permutations_of_multiset {#} = {[]}" unfolding permutations_of_multiset_def by simp lemma permutations_of_multiset_nonempty: assumes nonempty: "A \ {#}" shows "permutations_of_multiset A = (\x\set_mset A. ((#) x) ` permutations_of_multiset (A - {#x#}))" (is "_ = ?rhs") proof safe fix xs assume "xs \ permutations_of_multiset A" hence mset_xs: "mset xs = A" by (simp add: permutations_of_multiset_def) hence "xs \ []" by (auto simp: nonempty) then obtain x xs' where xs: "xs = x # xs'" by (cases xs) simp_all with mset_xs have "x \ set_mset A" "xs' \ permutations_of_multiset (A - {#x#})" by (auto simp: permutations_of_multiset_def) with xs show "xs \ ?rhs" by auto qed (auto simp: permutations_of_multiset_def) lemma permutations_of_multiset_singleton [simp]: "permutations_of_multiset {#x#} = {[x]}" by (simp add: permutations_of_multiset_nonempty) lemma permutations_of_multiset_doubleton: "permutations_of_multiset {#x,y#} = {[x,y], [y,x]}" by (simp add: permutations_of_multiset_nonempty insert_commute) lemma rev_permutations_of_multiset [simp]: "rev ` permutations_of_multiset A = permutations_of_multiset A" proof have "rev ` rev ` permutations_of_multiset A \ rev ` permutations_of_multiset A" unfolding permutations_of_multiset_def by auto also have "rev ` rev ` permutations_of_multiset A = permutations_of_multiset A" by (simp add: image_image) finally show "permutations_of_multiset A \ rev ` permutations_of_multiset A" . next show "rev ` permutations_of_multiset A \ permutations_of_multiset A" unfolding permutations_of_multiset_def by auto qed lemma length_finite_permutations_of_multiset: "xs \ permutations_of_multiset A \ length xs = size A" by (auto simp: permutations_of_multiset_def) lemma permutations_of_multiset_lists: "permutations_of_multiset A \ lists (set_mset A)" by (auto simp: permutations_of_multiset_def) lemma finite_permutations_of_multiset [simp]: "finite (permutations_of_multiset A)" proof (rule finite_subset) show "permutations_of_multiset A \ {xs. set xs \ set_mset A \ length xs = size A}" by (auto simp: permutations_of_multiset_def) show "finite {xs. set xs \ set_mset A \ length xs = size A}" by (rule finite_lists_length_eq) simp_all qed lemma permutations_of_multiset_not_empty [simp]: "permutations_of_multiset A \ {}" proof - from ex_mset[of A] guess xs .. thus ?thesis by (auto simp: permutations_of_multiset_def) qed lemma permutations_of_multiset_image: "permutations_of_multiset (image_mset f A) = map f ` permutations_of_multiset A" proof safe fix xs assume A: "xs \ permutations_of_multiset (image_mset f A)" from ex_mset[of A] obtain ys where ys: "mset ys = A" .. with A have "mset xs = mset (map f ys)" by (simp add: permutations_of_multiset_def) from mset_eq_permutation[OF this] guess \ . note \ = this with ys have "xs = map f (permute_list \ ys)" by (simp add: permute_list_map) moreover from \ ys have "permute_list \ ys \ permutations_of_multiset A" by (simp add: permutations_of_multiset_def) ultimately show "xs \ map f ` permutations_of_multiset A" by blast qed (auto simp: permutations_of_multiset_def) subsection \Cardinality of permutations\ text \ In this section, we prove some basic facts about the number of permutations of a multiset. \ context begin private lemma multiset_prod_fact_insert: "(\y\set_mset (A+{#x#}). fact (count (A+{#x#}) y)) = (count A x + 1) * (\y\set_mset A. fact (count A y))" proof - have "(\y\set_mset (A+{#x#}). fact (count (A+{#x#}) y)) = (\y\set_mset (A+{#x#}). (if y = x then count A x + 1 else 1) * fact (count A y))" by (intro prod.cong) simp_all also have "\ = (count A x + 1) * (\y\set_mset (A+{#x#}). fact (count A y))" by (simp add: prod.distrib prod.delta) also have "(\y\set_mset (A+{#x#}). fact (count A y)) = (\y\set_mset A. fact (count A y))" by (intro prod.mono_neutral_right) (auto simp: not_in_iff) finally show ?thesis . qed private lemma multiset_prod_fact_remove: "x \# A \ (\y\set_mset A. fact (count A y)) = count A x * (\y\set_mset (A-{#x#}). fact (count (A-{#x#}) y))" using multiset_prod_fact_insert[of "A - {#x#}" x] by simp lemma card_permutations_of_multiset_aux: "card (permutations_of_multiset A) * (\x\set_mset A. fact (count A x)) = fact (size A)" proof (induction A rule: multiset_remove_induct) case (remove A) have "card (permutations_of_multiset A) = card (\x\set_mset A. (#) x ` permutations_of_multiset (A - {#x#}))" by (simp add: permutations_of_multiset_nonempty remove.hyps) also have "\ = (\x\set_mset A. card (permutations_of_multiset (A - {#x#})))" by (subst card_UN_disjoint) (auto simp: card_image) also have "\ * (\x\set_mset A. fact (count A x)) = (\x\set_mset A. card (permutations_of_multiset (A - {#x#})) * (\y\set_mset A. fact (count A y)))" by (subst sum_distrib_right) simp_all also have "\ = (\x\set_mset A. count A x * fact (size A - 1))" proof (intro sum.cong refl) fix x assume x: "x \# A" have "card (permutations_of_multiset (A - {#x#})) * (\y\set_mset A. fact (count A y)) = count A x * (card (permutations_of_multiset (A - {#x#})) * (\y\set_mset (A - {#x#}). fact (count (A - {#x#}) y)))" (is "?lhs = _") by (subst multiset_prod_fact_remove[OF x]) simp_all also note remove.IH[OF x] also from x have "size (A - {#x#}) = size A - 1" by (simp add: size_Diff_submset) finally show "?lhs = count A x * fact (size A - 1)" . qed also have "(\x\set_mset A. count A x * fact (size A - 1)) = size A * fact (size A - 1)" by (simp add: sum_distrib_right size_multiset_overloaded_eq) also from remove.hyps have "\ = fact (size A)" by (cases "size A") auto finally show ?case . qed simp_all theorem card_permutations_of_multiset: "card (permutations_of_multiset A) = fact (size A) div (\x\set_mset A. fact (count A x))" "(\x\set_mset A. fact (count A x) :: nat) dvd fact (size A)" by (simp_all flip: card_permutations_of_multiset_aux[of A]) lemma card_permutations_of_multiset_insert_aux: "card (permutations_of_multiset (A + {#x#})) * (count A x + 1) = (size A + 1) * card (permutations_of_multiset A)" proof - note card_permutations_of_multiset_aux[of "A + {#x#}"] also have "fact (size (A + {#x#})) = (size A + 1) * fact (size A)" by simp also note multiset_prod_fact_insert[of A x] also note card_permutations_of_multiset_aux[of A, symmetric] finally have "card (permutations_of_multiset (A + {#x#})) * (count A x + 1) * (\y\set_mset A. fact (count A y)) = (size A + 1) * card (permutations_of_multiset A) * (\x\set_mset A. fact (count A x))" by (simp only: mult_ac) thus ?thesis by (subst (asm) mult_right_cancel) simp_all qed lemma card_permutations_of_multiset_remove_aux: assumes "x \# A" shows "card (permutations_of_multiset A) * count A x = size A * card (permutations_of_multiset (A - {#x#}))" proof - from assms have A: "A - {#x#} + {#x#} = A" by simp from assms have B: "size A = size (A - {#x#}) + 1" by (subst A [symmetric], subst size_union) simp show ?thesis using card_permutations_of_multiset_insert_aux[of "A - {#x#}" x, unfolded A] assms by (simp add: B) qed lemma real_card_permutations_of_multiset_remove: assumes "x \# A" shows "real (card (permutations_of_multiset (A - {#x#}))) = real (card (permutations_of_multiset A) * count A x) / real (size A)" using assms by (subst card_permutations_of_multiset_remove_aux[OF assms]) auto lemma real_card_permutations_of_multiset_remove': assumes "x \# A" shows "real (card (permutations_of_multiset A)) = real (size A * card (permutations_of_multiset (A - {#x#}))) / real (count A x)" using assms by (subst card_permutations_of_multiset_remove_aux[OF assms, symmetric]) simp end subsection \Permutations of a set\ definition permutations_of_set :: "'a set \ 'a list set" where "permutations_of_set A = {xs. set xs = A \ distinct xs}" lemma permutations_of_set_altdef: "finite A \ permutations_of_set A = permutations_of_multiset (mset_set A)" by (auto simp add: permutations_of_set_def permutations_of_multiset_def mset_set_set in_multiset_in_set [symmetric] mset_eq_mset_set_imp_distinct) lemma permutations_of_setI [intro]: assumes "set xs = A" "distinct xs" shows "xs \ permutations_of_set A" using assms unfolding permutations_of_set_def by simp lemma permutations_of_setD: assumes "xs \ permutations_of_set A" shows "set xs = A" "distinct xs" using assms unfolding permutations_of_set_def by simp_all lemma permutations_of_set_lists: "permutations_of_set A \ lists A" unfolding permutations_of_set_def by auto lemma permutations_of_set_empty [simp]: "permutations_of_set {} = {[]}" by (auto simp: permutations_of_set_def) lemma UN_set_permutations_of_set [simp]: "finite A \ (\xs\permutations_of_set A. set xs) = A" using finite_distinct_list by (auto simp: permutations_of_set_def) lemma permutations_of_set_infinite: "\finite A \ permutations_of_set A = {}" by (auto simp: permutations_of_set_def) lemma permutations_of_set_nonempty: "A \ {} \ permutations_of_set A = (\x\A. (\xs. x # xs) ` permutations_of_set (A - {x}))" by (cases "finite A") (simp_all add: permutations_of_multiset_nonempty mset_set_empty_iff mset_set_Diff permutations_of_set_altdef permutations_of_set_infinite) lemma permutations_of_set_singleton [simp]: "permutations_of_set {x} = {[x]}" by (subst permutations_of_set_nonempty) auto lemma permutations_of_set_doubleton: "x \ y \ permutations_of_set {x,y} = {[x,y], [y,x]}" by (subst permutations_of_set_nonempty) (simp_all add: insert_Diff_if insert_commute) lemma rev_permutations_of_set [simp]: "rev ` permutations_of_set A = permutations_of_set A" by (cases "finite A") (simp_all add: permutations_of_set_altdef permutations_of_set_infinite) lemma length_finite_permutations_of_set: "xs \ permutations_of_set A \ length xs = card A" by (auto simp: permutations_of_set_def distinct_card) lemma finite_permutations_of_set [simp]: "finite (permutations_of_set A)" by (cases "finite A") (simp_all add: permutations_of_set_infinite permutations_of_set_altdef) lemma permutations_of_set_empty_iff [simp]: "permutations_of_set A = {} \ \finite A" unfolding permutations_of_set_def using finite_distinct_list[of A] by auto lemma card_permutations_of_set [simp]: "finite A \ card (permutations_of_set A) = fact (card A)" by (simp add: permutations_of_set_altdef card_permutations_of_multiset del: One_nat_def) lemma permutations_of_set_image_inj: assumes inj: "inj_on f A" shows "permutations_of_set (f ` A) = map f ` permutations_of_set A" by (cases "finite A") (simp_all add: permutations_of_set_infinite permutations_of_set_altdef permutations_of_multiset_image mset_set_image_inj inj finite_image_iff) lemma permutations_of_set_image_permutes: "\ permutes A \ map \ ` permutations_of_set A = permutations_of_set A" by (subst permutations_of_set_image_inj [symmetric]) (simp_all add: permutes_inj_on permutes_image) subsection \Code generation\ text \ First, we give code an implementation for permutations of lists. \ declare length_remove1 [termination_simp] fun permutations_of_list_impl where "permutations_of_list_impl xs = (if xs = [] then [[]] else List.bind (remdups xs) (\x. map ((#) x) (permutations_of_list_impl (remove1 x xs))))" fun permutations_of_list_impl_aux where "permutations_of_list_impl_aux acc xs = (if xs = [] then [acc] else List.bind (remdups xs) (\x. permutations_of_list_impl_aux (x#acc) (remove1 x xs)))" declare permutations_of_list_impl_aux.simps [simp del] declare permutations_of_list_impl.simps [simp del] lemma permutations_of_list_impl_Nil [simp]: "permutations_of_list_impl [] = [[]]" by (simp add: permutations_of_list_impl.simps) lemma permutations_of_list_impl_nonempty: "xs \ [] \ permutations_of_list_impl xs = List.bind (remdups xs) (\x. map ((#) x) (permutations_of_list_impl (remove1 x xs)))" by (subst permutations_of_list_impl.simps) simp_all lemma set_permutations_of_list_impl: "set (permutations_of_list_impl xs) = permutations_of_multiset (mset xs)" by (induction xs rule: permutations_of_list_impl.induct) (subst permutations_of_list_impl.simps, simp_all add: permutations_of_multiset_nonempty set_list_bind) lemma distinct_permutations_of_list_impl: "distinct (permutations_of_list_impl xs)" by (induction xs rule: permutations_of_list_impl.induct, subst permutations_of_list_impl.simps) (auto intro!: distinct_list_bind simp: distinct_map o_def disjoint_family_on_def) lemma permutations_of_list_impl_aux_correct': "permutations_of_list_impl_aux acc xs = map (\xs. rev xs @ acc) (permutations_of_list_impl xs)" by (induction acc xs rule: permutations_of_list_impl_aux.induct, subst permutations_of_list_impl_aux.simps, subst permutations_of_list_impl.simps) (auto simp: map_list_bind intro!: list_bind_cong) lemma permutations_of_list_impl_aux_correct: "permutations_of_list_impl_aux [] xs = map rev (permutations_of_list_impl xs)" by (simp add: permutations_of_list_impl_aux_correct') lemma distinct_permutations_of_list_impl_aux: "distinct (permutations_of_list_impl_aux acc xs)" by (simp add: permutations_of_list_impl_aux_correct' distinct_map distinct_permutations_of_list_impl inj_on_def) lemma set_permutations_of_list_impl_aux: "set (permutations_of_list_impl_aux [] xs) = permutations_of_multiset (mset xs)" by (simp add: permutations_of_list_impl_aux_correct set_permutations_of_list_impl) declare set_permutations_of_list_impl_aux [symmetric, code] value [code] "permutations_of_multiset {#1,2,3,4::int#}" text \ Now we turn to permutations of sets. We define an auxiliary version with an accumulator to avoid having to map over the results. \ function permutations_of_set_aux where "permutations_of_set_aux acc A = (if \finite A then {} else if A = {} then {acc} else (\x\A. permutations_of_set_aux (x#acc) (A - {x})))" by auto termination by (relation "Wellfounded.measure (card \ snd)") (simp_all add: card_gt_0_iff) lemma permutations_of_set_aux_altdef: "permutations_of_set_aux acc A = (\xs. rev xs @ acc) ` permutations_of_set A" proof (cases "finite A") assume "finite A" thus ?thesis proof (induction A arbitrary: acc rule: finite_psubset_induct) case (psubset A acc) show ?case proof (cases "A = {}") case False note [simp del] = permutations_of_set_aux.simps from psubset.hyps False have "permutations_of_set_aux acc A = (\y\A. permutations_of_set_aux (y#acc) (A - {y}))" by (subst permutations_of_set_aux.simps) simp_all also have "\ = (\y\A. (\xs. rev xs @ acc) ` (\xs. y # xs) ` permutations_of_set (A - {y}))" by (intro SUP_cong refl, subst psubset) (auto simp: image_image) also from False have "\ = (\xs. rev xs @ acc) ` permutations_of_set A" by (subst (2) permutations_of_set_nonempty) (simp_all add: image_UN) finally show ?thesis . qed simp_all qed qed (simp_all add: permutations_of_set_infinite) declare permutations_of_set_aux.simps [simp del] lemma permutations_of_set_aux_correct: "permutations_of_set_aux [] A = permutations_of_set A" by (simp add: permutations_of_set_aux_altdef) text \ In another refinement step, we define a version on lists. \ declare length_remove1 [termination_simp] fun permutations_of_set_aux_list where "permutations_of_set_aux_list acc xs = (if xs = [] then [acc] else List.bind xs (\x. permutations_of_set_aux_list (x#acc) (List.remove1 x xs)))" definition permutations_of_set_list where "permutations_of_set_list xs = permutations_of_set_aux_list [] xs" declare permutations_of_set_aux_list.simps [simp del] lemma permutations_of_set_aux_list_refine: assumes "distinct xs" shows "set (permutations_of_set_aux_list acc xs) = permutations_of_set_aux acc (set xs)" using assms by (induction acc xs rule: permutations_of_set_aux_list.induct) (subst permutations_of_set_aux_list.simps, subst permutations_of_set_aux.simps, simp_all add: set_list_bind cong: SUP_cong) text \ The permutation lists contain no duplicates if the inputs contain no duplicates. Therefore, these functions can easily be used when working with a representation of sets by distinct lists. The same approach should generalise to any kind of set implementation that supports a monadic bind operation, and since the results are disjoint, merging should be cheap. \ lemma distinct_permutations_of_set_aux_list: "distinct xs \ distinct (permutations_of_set_aux_list acc xs)" by (induction acc xs rule: permutations_of_set_aux_list.induct) (subst permutations_of_set_aux_list.simps, auto intro!: distinct_list_bind simp: disjoint_family_on_def permutations_of_set_aux_list_refine permutations_of_set_aux_altdef) lemma distinct_permutations_of_set_list: "distinct xs \ distinct (permutations_of_set_list xs)" by (simp add: permutations_of_set_list_def distinct_permutations_of_set_aux_list) lemma permutations_of_list: "permutations_of_set (set xs) = set (permutations_of_set_list (remdups xs))" by (simp add: permutations_of_set_aux_correct [symmetric] permutations_of_set_aux_list_refine permutations_of_set_list_def) lemma permutations_of_list_code [code]: "permutations_of_set (set xs) = set (permutations_of_set_list (remdups xs))" "permutations_of_set (List.coset xs) = Code.abort (STR ''Permutation of set complement not supported'') (\_. permutations_of_set (List.coset xs))" by (simp_all add: permutations_of_list) value [code] "permutations_of_set (set ''abcd'')" end