src/HOL/Library/Nat_Bijection.thy
author wenzelm
Sun Mar 13 22:55:50 2011 +0100 (2011-03-13)
changeset 41959 b460124855b8
parent 39302 d7728f65b353
child 51414 587f493447d9
permissions -rw-r--r--
tuned headers;
wenzelm@41959
     1
(*  Title:      HOL/Library/Nat_Bijection.thy
huffman@35700
     2
    Author:     Brian Huffman
huffman@35700
     3
    Author:     Florian Haftmann
huffman@35700
     4
    Author:     Stefan Richter
huffman@35700
     5
    Author:     Tobias Nipkow
huffman@35700
     6
    Author:     Alexander Krauss
huffman@35700
     7
*)
huffman@35700
     8
huffman@35700
     9
header {* Bijections between natural numbers and other types *}
huffman@35700
    10
huffman@35700
    11
theory Nat_Bijection
huffman@35700
    12
imports Main Parity
huffman@35700
    13
begin
huffman@35700
    14
huffman@35700
    15
subsection {* Type @{typ "nat \<times> nat"} *}
huffman@35700
    16
huffman@35700
    17
text "Triangle numbers: 0, 1, 3, 6, 10, 15, ..."
huffman@35700
    18
huffman@35700
    19
definition
huffman@35700
    20
  triangle :: "nat \<Rightarrow> nat"
huffman@35700
    21
where
huffman@35700
    22
  "triangle n = n * Suc n div 2"
huffman@35700
    23
huffman@35700
    24
lemma triangle_0 [simp]: "triangle 0 = 0"
huffman@35700
    25
unfolding triangle_def by simp
huffman@35700
    26
huffman@35700
    27
lemma triangle_Suc [simp]: "triangle (Suc n) = triangle n + Suc n"
huffman@35700
    28
unfolding triangle_def by simp
huffman@35700
    29
huffman@35700
    30
definition
huffman@35700
    31
  prod_encode :: "nat \<times> nat \<Rightarrow> nat"
huffman@35700
    32
where
huffman@35700
    33
  "prod_encode = (\<lambda>(m, n). triangle (m + n) + m)"
huffman@35700
    34
huffman@35700
    35
text {* In this auxiliary function, @{term "triangle k + m"} is an invariant. *}
huffman@35700
    36
huffman@35700
    37
fun
huffman@35700
    38
  prod_decode_aux :: "nat \<Rightarrow> nat \<Rightarrow> nat \<times> nat"
huffman@35700
    39
where
huffman@35700
    40
  "prod_decode_aux k m =
huffman@35700
    41
    (if m \<le> k then (m, k - m) else prod_decode_aux (Suc k) (m - Suc k))"
huffman@35700
    42
huffman@35700
    43
declare prod_decode_aux.simps [simp del]
huffman@35700
    44
huffman@35700
    45
definition
huffman@35700
    46
  prod_decode :: "nat \<Rightarrow> nat \<times> nat"
huffman@35700
    47
where
huffman@35700
    48
  "prod_decode = prod_decode_aux 0"
huffman@35700
    49
huffman@35700
    50
lemma prod_encode_prod_decode_aux:
huffman@35700
    51
  "prod_encode (prod_decode_aux k m) = triangle k + m"
huffman@35700
    52
apply (induct k m rule: prod_decode_aux.induct)
huffman@35700
    53
apply (subst prod_decode_aux.simps)
huffman@35700
    54
apply (simp add: prod_encode_def)
huffman@35700
    55
done
huffman@35700
    56
huffman@35700
    57
lemma prod_decode_inverse [simp]: "prod_encode (prod_decode n) = n"
huffman@35700
    58
unfolding prod_decode_def by (simp add: prod_encode_prod_decode_aux)
huffman@35700
    59
huffman@35700
    60
lemma prod_decode_triangle_add:
huffman@35700
    61
  "prod_decode (triangle k + m) = prod_decode_aux k m"
huffman@35700
    62
apply (induct k arbitrary: m)
huffman@35700
    63
apply (simp add: prod_decode_def)
huffman@35700
    64
apply (simp only: triangle_Suc add_assoc)
huffman@35700
    65
apply (subst prod_decode_aux.simps, simp)
huffman@35700
    66
done
huffman@35700
    67
huffman@35700
    68
lemma prod_encode_inverse [simp]: "prod_decode (prod_encode x) = x"
huffman@35700
    69
unfolding prod_encode_def
huffman@35700
    70
apply (induct x)
huffman@35700
    71
apply (simp add: prod_decode_triangle_add)
huffman@35700
    72
apply (subst prod_decode_aux.simps, simp)
huffman@35700
    73
done
huffman@35700
    74
huffman@35700
    75
lemma inj_prod_encode: "inj_on prod_encode A"
huffman@35700
    76
by (rule inj_on_inverseI, rule prod_encode_inverse)
huffman@35700
    77
huffman@35700
    78
lemma inj_prod_decode: "inj_on prod_decode A"
huffman@35700
    79
by (rule inj_on_inverseI, rule prod_decode_inverse)
huffman@35700
    80
huffman@35700
    81
lemma surj_prod_encode: "surj prod_encode"
huffman@35700
    82
by (rule surjI, rule prod_decode_inverse)
huffman@35700
    83
huffman@35700
    84
lemma surj_prod_decode: "surj prod_decode"
huffman@35700
    85
by (rule surjI, rule prod_encode_inverse)
huffman@35700
    86
huffman@35700
    87
lemma bij_prod_encode: "bij prod_encode"
huffman@35700
    88
by (rule bijI [OF inj_prod_encode surj_prod_encode])
huffman@35700
    89
huffman@35700
    90
lemma bij_prod_decode: "bij prod_decode"
huffman@35700
    91
by (rule bijI [OF inj_prod_decode surj_prod_decode])
huffman@35700
    92
huffman@35700
    93
lemma prod_encode_eq: "prod_encode x = prod_encode y \<longleftrightarrow> x = y"
huffman@35700
    94
by (rule inj_prod_encode [THEN inj_eq])
huffman@35700
    95
huffman@35700
    96
lemma prod_decode_eq: "prod_decode x = prod_decode y \<longleftrightarrow> x = y"
huffman@35700
    97
by (rule inj_prod_decode [THEN inj_eq])
huffman@35700
    98
huffman@35700
    99
text {* Ordering properties *}
huffman@35700
   100
huffman@35700
   101
lemma le_prod_encode_1: "a \<le> prod_encode (a, b)"
huffman@35700
   102
unfolding prod_encode_def by simp
huffman@35700
   103
huffman@35700
   104
lemma le_prod_encode_2: "b \<le> prod_encode (a, b)"
huffman@35700
   105
unfolding prod_encode_def by (induct b, simp_all)
huffman@35700
   106
huffman@35700
   107
huffman@35700
   108
subsection {* Type @{typ "nat + nat"} *}
huffman@35700
   109
huffman@35700
   110
definition
huffman@35700
   111
  sum_encode  :: "nat + nat \<Rightarrow> nat"
huffman@35700
   112
where
huffman@35700
   113
  "sum_encode x = (case x of Inl a \<Rightarrow> 2 * a | Inr b \<Rightarrow> Suc (2 * b))"
huffman@35700
   114
huffman@35700
   115
definition
huffman@35700
   116
  sum_decode  :: "nat \<Rightarrow> nat + nat"
huffman@35700
   117
where
huffman@35700
   118
  "sum_decode n = (if even n then Inl (n div 2) else Inr (n div 2))"
huffman@35700
   119
huffman@35700
   120
lemma sum_encode_inverse [simp]: "sum_decode (sum_encode x) = x"
huffman@35700
   121
unfolding sum_decode_def sum_encode_def
huffman@35700
   122
by (induct x) simp_all
huffman@35700
   123
huffman@35700
   124
lemma sum_decode_inverse [simp]: "sum_encode (sum_decode n) = n"
huffman@35700
   125
unfolding sum_decode_def sum_encode_def numeral_2_eq_2
huffman@35700
   126
by (simp add: even_nat_div_two_times_two odd_nat_div_two_times_two_plus_one
huffman@35700
   127
         del: mult_Suc)
huffman@35700
   128
huffman@35700
   129
lemma inj_sum_encode: "inj_on sum_encode A"
huffman@35700
   130
by (rule inj_on_inverseI, rule sum_encode_inverse)
huffman@35700
   131
huffman@35700
   132
lemma inj_sum_decode: "inj_on sum_decode A"
huffman@35700
   133
by (rule inj_on_inverseI, rule sum_decode_inverse)
huffman@35700
   134
huffman@35700
   135
lemma surj_sum_encode: "surj sum_encode"
huffman@35700
   136
by (rule surjI, rule sum_decode_inverse)
huffman@35700
   137
huffman@35700
   138
lemma surj_sum_decode: "surj sum_decode"
huffman@35700
   139
by (rule surjI, rule sum_encode_inverse)
huffman@35700
   140
huffman@35700
   141
lemma bij_sum_encode: "bij sum_encode"
huffman@35700
   142
by (rule bijI [OF inj_sum_encode surj_sum_encode])
huffman@35700
   143
huffman@35700
   144
lemma bij_sum_decode: "bij sum_decode"
huffman@35700
   145
by (rule bijI [OF inj_sum_decode surj_sum_decode])
huffman@35700
   146
huffman@35700
   147
lemma sum_encode_eq: "sum_encode x = sum_encode y \<longleftrightarrow> x = y"
huffman@35700
   148
by (rule inj_sum_encode [THEN inj_eq])
huffman@35700
   149
huffman@35700
   150
lemma sum_decode_eq: "sum_decode x = sum_decode y \<longleftrightarrow> x = y"
huffman@35700
   151
by (rule inj_sum_decode [THEN inj_eq])
huffman@35700
   152
huffman@35700
   153
huffman@35700
   154
subsection {* Type @{typ "int"} *}
huffman@35700
   155
huffman@35700
   156
definition
huffman@35700
   157
  int_encode :: "int \<Rightarrow> nat"
huffman@35700
   158
where
huffman@35700
   159
  "int_encode i = sum_encode (if 0 \<le> i then Inl (nat i) else Inr (nat (- i - 1)))"
huffman@35700
   160
huffman@35700
   161
definition
huffman@35700
   162
  int_decode :: "nat \<Rightarrow> int"
huffman@35700
   163
where
huffman@35700
   164
  "int_decode n = (case sum_decode n of Inl a \<Rightarrow> int a | Inr b \<Rightarrow> - int b - 1)"
huffman@35700
   165
huffman@35700
   166
lemma int_encode_inverse [simp]: "int_decode (int_encode x) = x"
huffman@35700
   167
unfolding int_decode_def int_encode_def by simp
huffman@35700
   168
huffman@35700
   169
lemma int_decode_inverse [simp]: "int_encode (int_decode n) = n"
huffman@35700
   170
unfolding int_decode_def int_encode_def using sum_decode_inverse [of n]
huffman@35700
   171
by (cases "sum_decode n", simp_all)
huffman@35700
   172
huffman@35700
   173
lemma inj_int_encode: "inj_on int_encode A"
huffman@35700
   174
by (rule inj_on_inverseI, rule int_encode_inverse)
huffman@35700
   175
huffman@35700
   176
lemma inj_int_decode: "inj_on int_decode A"
huffman@35700
   177
by (rule inj_on_inverseI, rule int_decode_inverse)
huffman@35700
   178
huffman@35700
   179
lemma surj_int_encode: "surj int_encode"
huffman@35700
   180
by (rule surjI, rule int_decode_inverse)
huffman@35700
   181
huffman@35700
   182
lemma surj_int_decode: "surj int_decode"
huffman@35700
   183
by (rule surjI, rule int_encode_inverse)
huffman@35700
   184
huffman@35700
   185
lemma bij_int_encode: "bij int_encode"
huffman@35700
   186
by (rule bijI [OF inj_int_encode surj_int_encode])
huffman@35700
   187
huffman@35700
   188
lemma bij_int_decode: "bij int_decode"
huffman@35700
   189
by (rule bijI [OF inj_int_decode surj_int_decode])
huffman@35700
   190
huffman@35700
   191
lemma int_encode_eq: "int_encode x = int_encode y \<longleftrightarrow> x = y"
huffman@35700
   192
by (rule inj_int_encode [THEN inj_eq])
huffman@35700
   193
huffman@35700
   194
lemma int_decode_eq: "int_decode x = int_decode y \<longleftrightarrow> x = y"
huffman@35700
   195
by (rule inj_int_decode [THEN inj_eq])
huffman@35700
   196
huffman@35700
   197
huffman@35700
   198
subsection {* Type @{typ "nat list"} *}
huffman@35700
   199
huffman@35700
   200
fun
huffman@35700
   201
  list_encode :: "nat list \<Rightarrow> nat"
huffman@35700
   202
where
huffman@35700
   203
  "list_encode [] = 0"
huffman@35700
   204
| "list_encode (x # xs) = Suc (prod_encode (x, list_encode xs))"
huffman@35700
   205
huffman@35700
   206
function
huffman@35700
   207
  list_decode :: "nat \<Rightarrow> nat list"
huffman@35700
   208
where
huffman@35700
   209
  "list_decode 0 = []"
huffman@35700
   210
| "list_decode (Suc n) = (case prod_decode n of (x, y) \<Rightarrow> x # list_decode y)"
huffman@35700
   211
by pat_completeness auto
huffman@35700
   212
huffman@35700
   213
termination list_decode
huffman@35700
   214
apply (relation "measure id", simp_all)
huffman@35700
   215
apply (drule arg_cong [where f="prod_encode"])
haftmann@37591
   216
apply (drule sym)
huffman@35700
   217
apply (simp add: le_imp_less_Suc le_prod_encode_2)
huffman@35700
   218
done
huffman@35700
   219
huffman@35700
   220
lemma list_encode_inverse [simp]: "list_decode (list_encode x) = x"
huffman@35700
   221
by (induct x rule: list_encode.induct) simp_all
huffman@35700
   222
huffman@35700
   223
lemma list_decode_inverse [simp]: "list_encode (list_decode n) = n"
huffman@35700
   224
apply (induct n rule: list_decode.induct, simp)
huffman@35700
   225
apply (simp split: prod.split)
huffman@35700
   226
apply (simp add: prod_decode_eq [symmetric])
huffman@35700
   227
done
huffman@35700
   228
huffman@35700
   229
lemma inj_list_encode: "inj_on list_encode A"
huffman@35700
   230
by (rule inj_on_inverseI, rule list_encode_inverse)
huffman@35700
   231
huffman@35700
   232
lemma inj_list_decode: "inj_on list_decode A"
huffman@35700
   233
by (rule inj_on_inverseI, rule list_decode_inverse)
huffman@35700
   234
huffman@35700
   235
lemma surj_list_encode: "surj list_encode"
huffman@35700
   236
by (rule surjI, rule list_decode_inverse)
huffman@35700
   237
huffman@35700
   238
lemma surj_list_decode: "surj list_decode"
huffman@35700
   239
by (rule surjI, rule list_encode_inverse)
huffman@35700
   240
huffman@35700
   241
lemma bij_list_encode: "bij list_encode"
huffman@35700
   242
by (rule bijI [OF inj_list_encode surj_list_encode])
huffman@35700
   243
huffman@35700
   244
lemma bij_list_decode: "bij list_decode"
huffman@35700
   245
by (rule bijI [OF inj_list_decode surj_list_decode])
huffman@35700
   246
huffman@35700
   247
lemma list_encode_eq: "list_encode x = list_encode y \<longleftrightarrow> x = y"
huffman@35700
   248
by (rule inj_list_encode [THEN inj_eq])
huffman@35700
   249
huffman@35700
   250
lemma list_decode_eq: "list_decode x = list_decode y \<longleftrightarrow> x = y"
huffman@35700
   251
by (rule inj_list_decode [THEN inj_eq])
huffman@35700
   252
huffman@35700
   253
huffman@35700
   254
subsection {* Finite sets of naturals *}
huffman@35700
   255
huffman@35700
   256
subsubsection {* Preliminaries *}
huffman@35700
   257
huffman@35700
   258
lemma finite_vimage_Suc_iff: "finite (Suc -` F) \<longleftrightarrow> finite F"
huffman@35700
   259
apply (safe intro!: finite_vimageI inj_Suc)
huffman@35700
   260
apply (rule finite_subset [where B="insert 0 (Suc ` Suc -` F)"])
huffman@35700
   261
apply (rule subsetI, case_tac x, simp, simp)
huffman@35700
   262
apply (rule finite_insert [THEN iffD2])
huffman@35700
   263
apply (erule finite_imageI)
huffman@35700
   264
done
huffman@35700
   265
huffman@35700
   266
lemma vimage_Suc_insert_0: "Suc -` insert 0 A = Suc -` A"
huffman@35700
   267
by auto
huffman@35700
   268
huffman@35700
   269
lemma vimage_Suc_insert_Suc:
huffman@35700
   270
  "Suc -` insert (Suc n) A = insert n (Suc -` A)"
huffman@35700
   271
by auto
huffman@35700
   272
huffman@35700
   273
lemma even_nat_Suc_div_2: "even x \<Longrightarrow> Suc x div 2 = x div 2"
huffman@35700
   274
by (simp only: numeral_2_eq_2 even_nat_plus_one_div_two)
huffman@35700
   275
huffman@35700
   276
lemma div2_even_ext_nat:
huffman@35700
   277
  "\<lbrakk>x div 2 = y div 2; even x = even y\<rbrakk> \<Longrightarrow> (x::nat) = y"
huffman@35700
   278
apply (rule mod_div_equality [of x 2, THEN subst])
huffman@35700
   279
apply (rule mod_div_equality [of y 2, THEN subst])
huffman@35700
   280
apply (case_tac "even x")
huffman@35700
   281
apply (simp add: numeral_2_eq_2 even_nat_equiv_def)
huffman@35700
   282
apply (simp add: numeral_2_eq_2 odd_nat_equiv_def)
huffman@35700
   283
done
huffman@35700
   284
huffman@35700
   285
subsubsection {* From sets to naturals *}
huffman@35700
   286
huffman@35700
   287
definition
huffman@35700
   288
  set_encode :: "nat set \<Rightarrow> nat"
huffman@35700
   289
where
huffman@35700
   290
  "set_encode = setsum (op ^ 2)"
huffman@35700
   291
huffman@35700
   292
lemma set_encode_empty [simp]: "set_encode {} = 0"
huffman@35700
   293
by (simp add: set_encode_def)
huffman@35700
   294
huffman@35700
   295
lemma set_encode_insert [simp]:
huffman@35700
   296
  "\<lbrakk>finite A; n \<notin> A\<rbrakk> \<Longrightarrow> set_encode (insert n A) = 2^n + set_encode A"
huffman@35700
   297
by (simp add: set_encode_def)
huffman@35700
   298
huffman@35700
   299
lemma even_set_encode_iff: "finite A \<Longrightarrow> even (set_encode A) \<longleftrightarrow> 0 \<notin> A"
huffman@35700
   300
unfolding set_encode_def by (induct set: finite, auto)
huffman@35700
   301
huffman@35700
   302
lemma set_encode_vimage_Suc: "set_encode (Suc -` A) = set_encode A div 2"
huffman@35700
   303
apply (cases "finite A")
huffman@35700
   304
apply (erule finite_induct, simp)
huffman@35700
   305
apply (case_tac x)
huffman@35700
   306
apply (simp add: even_nat_Suc_div_2 even_set_encode_iff vimage_Suc_insert_0)
huffman@35700
   307
apply (simp add: finite_vimageI add_commute vimage_Suc_insert_Suc)
huffman@35700
   308
apply (simp add: set_encode_def finite_vimage_Suc_iff)
huffman@35700
   309
done
huffman@35700
   310
huffman@35700
   311
lemmas set_encode_div_2 = set_encode_vimage_Suc [symmetric]
huffman@35700
   312
huffman@35700
   313
subsubsection {* From naturals to sets *}
huffman@35700
   314
huffman@35700
   315
definition
huffman@35700
   316
  set_decode :: "nat \<Rightarrow> nat set"
huffman@35700
   317
where
huffman@35700
   318
  "set_decode x = {n. odd (x div 2 ^ n)}"
huffman@35700
   319
huffman@35700
   320
lemma set_decode_0 [simp]: "0 \<in> set_decode x \<longleftrightarrow> odd x"
huffman@35700
   321
by (simp add: set_decode_def)
huffman@35700
   322
huffman@35700
   323
lemma set_decode_Suc [simp]:
huffman@35700
   324
  "Suc n \<in> set_decode x \<longleftrightarrow> n \<in> set_decode (x div 2)"
huffman@35700
   325
by (simp add: set_decode_def div_mult2_eq)
huffman@35700
   326
huffman@35700
   327
lemma set_decode_zero [simp]: "set_decode 0 = {}"
huffman@35700
   328
by (simp add: set_decode_def)
huffman@35700
   329
huffman@35700
   330
lemma set_decode_div_2: "set_decode (x div 2) = Suc -` set_decode x"
huffman@35700
   331
by auto
huffman@35700
   332
huffman@35700
   333
lemma set_decode_plus_power_2:
huffman@35700
   334
  "n \<notin> set_decode z \<Longrightarrow> set_decode (2 ^ n + z) = insert n (set_decode z)"
huffman@35700
   335
 apply (induct n arbitrary: z, simp_all)
nipkow@39302
   336
  apply (rule set_eqI, induct_tac x, simp, simp add: even_nat_Suc_div_2)
nipkow@39302
   337
 apply (rule set_eqI, induct_tac x, simp, simp add: add_commute)
huffman@35700
   338
done
huffman@35700
   339
huffman@35700
   340
lemma finite_set_decode [simp]: "finite (set_decode n)"
huffman@35700
   341
apply (induct n rule: nat_less_induct)
huffman@35700
   342
apply (case_tac "n = 0", simp)
huffman@35700
   343
apply (drule_tac x="n div 2" in spec, simp)
huffman@35700
   344
apply (simp add: set_decode_div_2)
huffman@35700
   345
apply (simp add: finite_vimage_Suc_iff)
huffman@35700
   346
done
huffman@35700
   347
huffman@35700
   348
subsubsection {* Proof of isomorphism *}
huffman@35700
   349
huffman@35700
   350
lemma set_decode_inverse [simp]: "set_encode (set_decode n) = n"
huffman@35700
   351
apply (induct n rule: nat_less_induct)
huffman@35700
   352
apply (case_tac "n = 0", simp)
huffman@35700
   353
apply (drule_tac x="n div 2" in spec, simp)
huffman@35700
   354
apply (simp add: set_decode_div_2 set_encode_vimage_Suc)
huffman@35700
   355
apply (erule div2_even_ext_nat)
huffman@35700
   356
apply (simp add: even_set_encode_iff)
huffman@35700
   357
done
huffman@35700
   358
huffman@35700
   359
lemma set_encode_inverse [simp]: "finite A \<Longrightarrow> set_decode (set_encode A) = A"
huffman@35700
   360
apply (erule finite_induct, simp_all)
huffman@35700
   361
apply (simp add: set_decode_plus_power_2)
huffman@35700
   362
done
huffman@35700
   363
huffman@35700
   364
lemma inj_on_set_encode: "inj_on set_encode (Collect finite)"
huffman@35700
   365
by (rule inj_on_inverseI [where g="set_decode"], simp)
huffman@35700
   366
huffman@35700
   367
lemma set_encode_eq:
huffman@35700
   368
  "\<lbrakk>finite A; finite B\<rbrakk> \<Longrightarrow> set_encode A = set_encode B \<longleftrightarrow> A = B"
huffman@35700
   369
by (rule iffI, simp add: inj_onD [OF inj_on_set_encode], simp)
huffman@35700
   370
huffman@35700
   371
end