Theory Trie.Tries

(*  Author: Tobias Nipkow  *)

section "Tries (List Version)"

theory Tries
imports Trie
begin

text‹This is a specialization of tries where values are lists.›

type_synonym ('k,'v)tries = "('k,'v list)trie"

definition lookup_tries :: "('k,'v)tries  'k list  'v list" where
"lookup_tries t ks = (case lookup_trie t ks of None  [] | Some vs  vs)"

definition update_with_tries ::
  "'k list  ('v list  'v list)  ('k, 'v) tries  ('k, 'v) tries" where
"update_with_tries ks f =
  update_with_trie ks (λvo. case vo of None  f [] | Some vs  f vs)"

definition insert_tries :: "'k list  'v  ('k,'v)tries  ('k,'v)tries" where
"insert_tries ks v =
  update_with_trie ks (λvo. case vo of None  [v] | Some vs  v#vs)"

definition inserts_tries :: "('v  'k list)  'v list  ('k,'v)tries  ('k,'v)tries" where
"inserts_tries key = fold (%v. insert_tries (key v) v)"

definition tries_of_list :: "('v  'k list)  'v list  ('k,'v)tries" where
"tries_of_list key vs = inserts_tries key vs (Trie None [])"

definition set_tries :: "('k,'v)tries  'v set" where
"set_tries t = Union {gs. a. gs = set(lookup_tries t a)}"

definition all_tries :: "('v  bool)  ('k,'v)tries  bool" where
"all_tries P = all_trie (list_all P)"


subsection @{const lookup_tries}

lemma lookup_Nil[simp]:
  "lookup_tries (Trie vo ps) [] = (case vo of None  [] | Some vs  vs)"
by (simp add: lookup_tries_def)

lemma lookup_Cons[simp]: "lookup_tries (Trie vo ps) (a#as) =
  (case map_of ps a of None  [] | Some at  lookup_tries at as)"
by (simp add: lookup_tries_def split: option.split)

lemma lookup_empty[simp]: "lookup_tries (Trie None []) as = []"
by(case_tac as, simp_all add: lookup_tries_def)

theorem lookup_update:
 "lookup_tries (update_trie as vs t) bs =
 (if as=bs then vs else lookup_tries t bs)"
by(auto simp add: lookup_tries_def lookup_update)

theorem lookup_update_with:
 "lookup_tries (update_with_tries as f t) bs =
 (if as=bs then f(lookup_tries t as) else lookup_tries t bs)"
by(auto simp add: lookup_tries_def update_with_tries_def lookup_update_with_trie
  split: option.split)


subsection @{const insert_tries}, @{const inserts_tries}, @{const tries_of_list}

lemma invar_insert_tries: "invar_trie t  invar_trie(insert_tries as v t)"
by(simp add: insert_tries_def invar_update_with_trie split: option.split)

lemma invar_inserts_tries:
  "invar_trie t  invar_trie (inserts_tries key xs t)"
by(induct xs arbitrary: t)(auto simp: invar_insert_tries inserts_tries_def)

lemma invar_of_list: "invar_trie (tries_of_list key xs)"
by(simp add: tries_of_list_def invar_inserts_tries)

lemma set_lookup_insert_tries: "set (lookup_tries (insert_tries ks a t) ks') =
  (if ks' = ks then Set.insert a (set(lookup_tries t ks')) else set(lookup_tries t ks'))"
by(simp add: lookup_tries_def insert_tries_def lookup_update_with_trie set_eq_iff split: option.split)

lemma in_set_lookup_inserts_tries:
  "(v  set(lookup_tries (inserts_tries key vs t) (key v))) =
   (v  set vs  set(lookup_tries t (key v)))"
by(induct vs arbitrary: t)
  (auto simp add: inserts_tries_def set_lookup_insert_tries)

lemma in_set_lookup_of_list:
  "v  set(lookup_tries (tries_of_list key vs) (key v)) = (v  set vs)"
by(simp add: tries_of_list_def in_set_lookup_inserts_tries)

lemma in_set_lookup_inserts_triesD:
  "v  set(lookup_tries (inserts_tries key vs t) xs) 
  v  set vs  set(lookup_tries t xs)"
apply(induct vs arbitrary: t)
 apply (simp add: inserts_tries_def)
apply (simp add: inserts_tries_def)
apply (fastforce simp add: set_lookup_insert_tries split: if_splits)
done
 
lemma in_set_lookup_of_listD:
  "v  set(lookup_tries (tries_of_list f vs) xs)  v  set vs"
by(auto simp: tries_of_list_def dest: in_set_lookup_inserts_triesD)


subsection @{const set_tries}

lemma set_tries_eq_ran: "set_tries t = Union(set ` ran(lookup_trie t))"
apply(auto simp add: set_eq_iff set_tries_def lookup_tries_def ran_def)
 apply metis
by (metis option.inject)

lemma set_tries_empty[simp]: "set_tries (Trie None []) = {}"
by(simp add: set_tries_def)

lemma set_tries_insert[simp]:
  "set_tries (insert_tries a x t) = Set.insert x (set_tries t)"
apply(auto simp: set_tries_def lookup_update set_lookup_insert_tries)
by (metis insert_iff)

lemma set_insert_tries:
  "set_tries (inserts_tries key xs t) =
   set xs Un set_tries t"
by(induct xs arbitrary: t) (auto simp: inserts_tries_def)

lemma set_tries_of_list[simp]:
  "set_tries(tries_of_list key xs) = set xs"
by(simp add: tries_of_list_def set_insert_tries)

lemma in_set_lookup_set_triesD:
  "xset (lookup_tries t a)  x  set_tries t"
by(auto simp: set_tries_def)

end