Bước tới nội dung

Mã hóa Huffman

Bách khoa toàn thư mở Wikipedia

Trong khoa học máy tínhlý thuyết thông tin, mã hóa Huffman là một thuật toán mã hóa dùng để nén dữ liệu. Nó dựa trên bảng tần suất xuất hiện các ký tự cần mã hóa để xây dựng một bộ mã nhị phân cho các ký tự đó sao cho dung lượng (số bit) sau khi mã hóa là nhỏ nhất.

Tác giả

[sửa | sửa mã nguồn]

Thuật toán được đề xuất bởi David A. Huffman khi ông còn là sinh viên Ph.D. tại MIT, và công bố năm 1952 trong bài báo "A Method for the Construction of Minimum-Redundancy Codes". Sau này Huffman đã trở thành một giảng viên ở MIT và sau đó ở khoa Khoa học máy tính của Đại học California, Santa Cruz, Trường Kỹ nghệ Baskin (Baskin School of Engineering).

Dẫn nhập

[sửa | sửa mã nguồn]

Mã tiền tố

[sửa | sửa mã nguồn]

Để mã hóa các ký hiệu (ký tự, chữ số,...) ta thay chúng bằng các xâu nhị phân, được gọi là từ mã của ký hiệu đó. Chẳng hạn bộ mã ASCII, mã hóa cho 256 ký hiệu là biểu diễn nhị phân của các số từ 0 đến 255, mỗi từ mã gồm 8 bit. Trong ASCII từ mã của ký tự "a" là 1100001, của ký tự "A" là 1000001. Trong cách mã hóa này các từ mã của tất cả 256 ký hiệu có độ dài bằng nhau (mỗi từ mã 8 bit). Nó được gọi là mã hóa với độ dài không đổi.

Khi mã hóa một tài liệu có thể không sử dụng đến tất cả 256 ký hiệu. Hơn nữa trong tài liệu chữ cái "a" chỉ có thể xuất hiện 1000000 lần còn chữ cái "A" có thể chỉ xuất hiện 2, 3 lần. Như vậy ta có thể không cần dùng đủ 8 bit để mã hóa cho một ký hiệu, hơn nữa độ dài (số bít) dành cho mỗi ký hiệu có thể khác nhau, ký hiệu nào xuất hiện nhiều lần thì nên dùng số bit ít, ký hiệu nào xuất hiện ít thì có thể mã hóa bằng từ mã dài hơn. Như vậy ta có việc mã hóa với độ dài thay đổi. Tuy nhiên, nếu mã hóa với độ dài thay đổi, khi giải mã ta làm thế nào phân biệt được xâu bít nào là mã hóa của ký hiệu nào. Một trong các giải pháp là dùng các dấu phẩy (",") hoặc một ký hiệu quy ước nào đó để tách từ mã của các ký tự đứng cạnh nhau. Nhưng như thế số các dấu phẩy sẽ chiếm một không gian đáng kể trong bảng mã. Một cách giải quyết khác dẫn đến khái niệm mã tiền tố

  • Mã tiền tố là bộ các từ mã của một tập hợp các ký hiệu sao cho từ mã của mỗi ký hiệu không là tiền tố (phần đầu) của từ mã một ký hiệu khác trong bộ mã ấy.

Đương nhiên mã hóa với độ dài không đổi là mã tiền tố.

Ví dụ: Giả sử mã hóa từ "ARRAY", tập các ký hiệu cần mã hóa gồm 3 chữ cái "A","R","Y".
  • Nếu mã hóa bằng các từ mã có độ dài bằng nhau ta dùng ít nhất 2 bit cho một chữ cái chẳng hạn "A"=00, "R"=01, "Y"=10. Khi đó mã hóa của cả từ là 0001010010. Để giải mã ta đọc hai bit một và đối chiếu với bảng mã.
  • Nếu mã hóa "A"=0, "R"=01, "Y"=11 thì bộ từ mã này không là mã tiền tố vì từ mã của "A" là tiền tố của từ mã của "R". Để mã hóa cả từ ARRAY phải đặt dấu ngăn cách vào giữa các từ mã 0,01,01,0,11
  • Nếu mã hóa "A"=0, "R"=10, "Y"=11 thì bộ mã này là mã tiền tố. Với bộ mã tiền tố này khi mã hóa xâu "ARRAY" ta có 01010011.

Biểu diễn mã tiền tố trên cây nhị phân

[sửa | sửa mã nguồn]
  • Nếu có một cây nhị phân lá ta có thể tạo một bộ mã tiền tố cho ký hiệu bằng cách đặt mỗi ký hiệu vào một lá. Từ mã của mỗi ký hiệu được tạo ra khi đi từ gốc tới lá chứa ký hiệu đó, nếu đi qua cạnh trái thì ta thêm số 0, đi qua cạnh phải thì thêm số 1.
  • Ví dụ: Cây 3 lá sau đây biểu diễn bộ mã của A,R,Y trong ví dụ trên
   *
  0/ \1
  A  *
   0/ \1
   R  Y
  • Từ mã của "A" là 0, của "R" là 10, của "Y" là 11.

Mã tiền tố tối ưu

[sửa | sửa mã nguồn]

Từ ví dụ trên thấy mã hóa của xâu "ARRAY" bằng mã độ dài cố định mất 10 bit, bằng mã tiền tố đã đưa ra mất 8 bit, tiết kiệm được 20%. Bài toán đặt ra là bộ mã tiền tố đã tối ưu chưa.

Bài toán

[sửa | sửa mã nguồn]
  • Cho tập A các ký hiệu và trọng số (tần suất) của chúng.
  • Tìm một bộ mã tiền tố với tổng độ dài mã hóa là nhỏ nhất.
Cây Huffman sinh ra từ xâu ký tự "this is an example of a huffman tree". Tổng số bit để mã hóa là 135, không kể các ký tự trống.

Dữ liệu

[sửa | sửa mã nguồn]
Input

Bảng chữ cái .
Tập các trọng số (tần suất xuất hiện) tương ứng , ví dụ: .

Output

Bộ mã , là tập hợp các từ mã (nhị phân), trong đó là từ mã của .

Yêu cầu

Đặt là trọng số của bộ mã . Điều kiện là: với mọi bộ mã .

  • Trong ví dụ sau, với các tần số như trên mã 1 sẽ tốn không gian hơn mã 2.
Input Ký tự a b c d e  
tần suất 0.10 0.15 0.30 0.16 0.29 1,00
Mã 1 Từ mã 000 001 010 011 110  
Độ dài từ mã (bits) 3 3 3 3 3 3,00
Mã 2 Từ mã 000 001 10 01 11
Độ dài từ mã (bits) 3 3 2 2 2 2,25

Giải thuật

[sửa | sửa mã nguồn]

Giải thuật tham lam

[sửa | sửa mã nguồn]

Trong giải thuật tham lam giải bài toán xây dựng cây mã tiền tố tối ưu của Huffman, ở mỗi bước ta chọn hai chữ cái có tần số thấp nhất để mã hóa bằng từ mã dài nhất. Giả sử có tập A gồm ký hiệu và hàm trọng số tương ứng .

  • Khởi tạo: Tạo một rừng gồm n cây, mỗi cây chỉ có một nút gốc, mỗi nút gốc tương ứng với một ký tự và có trọng số là tần số/tần suất của ký tự đó .
  • Lặp:
    • Mỗi bước sau thực hiện cho đến khi rừng chỉ còn một cây:
    • Chọn hai cây có trọng số ở gốc nhỏ nhất hợp thành một cây bằng cách thêm một gốc mới nối với hai gốc đã chọn. Trọng số của gốc mới bằng tổng trọng số của hai gốc tạo thành nó.

Như vậy ở mỗi bước số cây bớt đi một. Khi rừng chỉ còn một cây thì cây đó biểu diễn mã tiền tố tối ưu với các ký tự đặt ở các lá tương ứng.

Cho bảng tần suất của 5 chữ cái A, B, C, D, E như sau tương ứng là 0.10; 0.15; 0.30; 0.16; 0.29

A B C D E
0.10 0.15 0.30 0.16 0.29

Quá trình xây dựng cây Huffman diễn ra như sau:

Bước 1 và bước 2
Bước 1 và bước 2
Bước 3 và bước 4
Bước 5
Bước 5

Như vậy bộ mã tối ưu tương ứng là:

A B C D E
010 011 11 00 10

Hàng đợi có ưu tiên

[sửa | sửa mã nguồn]

Trong mỗi bước của thuật toán xây dựng cây Huffman, ta luôn phải chọn ra hai gốc có trọng số nhỏ nhất. Để làm việc này ta sắp xếp các gốc vào một hàng đợi ưu tiên theo tiêu chuẩn trọng số nhỏ nhất. Một trong các cấu trúc dữ liệu thuận lợi cho tiêu chuẩn này là cấu trúc đống (với phần tử có trọng số nhỏ nhất nằm trên đỉnh của đống).

Trong đoạn mã giả dưới đây ta dựa trên một mảng các ký hiệu có tần suất tương ứng là

Tạo hàng đợi bằng đống (heap)

[sửa | sửa mã nguồn]

Ta tạo một đống trên cơ sở sắp xếp lại các chỉ số của A và W. Ta lưu trữ đống dưới dạng mảng, ký hiệu nó là Heap[1..n]. Trước hết đưa chỉ số của các chữ cái theo thứ tự ban đầu vào mảng Heap[1..n] với Heap[i]=i. với mọi i=1..n.

Procedure DownHeap(List W,Int k,Int Count)
{
Int i:=k, v:=W(Heap(k)), j 

While 2*i<=Count {
j:=2*i
if j 1<= Count and W(Heap(j 1))>W(Heap(j)) then j:=j 1
if W(Heap(j))< v then Heap(i):=Heap(j)
else break 
i:=j
Heap(j):=Heap(k)
}
}
Procedure MakeHeap(List W,Int n)
{
Int k 
For k:=int(n/2) downto 1 {
DownHeap (W,k,n)
}
}

Xây dựng cây Huffman

[sửa | sửa mã nguồn]

Ta sẽ lưu trữ cấu trúc của cây mã Huffman vào một mảng. Cây Huffman gồm n lá mỗi lá chứa chỉ số của chữ cái tương ứng. Mỗi lần ghép 2 cây thành một ta phải thêm một đỉnh, như vậy cây biểu diễn mã Huffman gồm 2.n-1 đỉnh. Ta ký hiệu cây này là Huff[1..2n-1]. Vì mỗi gốc mới bổ sung đều có trọng số nên ta mở rộng mảng W[1..n] các trọng số thành mảng W' [1..2n-1]. Gọi m là số đỉnh của cây sẽ xây dựng. lúc đầu ta có n lá, đỉnh bổ sung lần đầu sẽ là n 1, lần thứ 2 là n 2,... Khi lấy ra hai ký tự có tần số nhỏ nhất chẳng hạn ký tự thứ i làm con trái và ký tự thứ j làm con phải của đỉnh mới bổ sung có chỉ số m ta đặt Huff[i]=-m, Huff[j]=m.

Procedure MakeTreeHuffman(List W,Int n)
{
List Heap,Huff 
Int i,j,count:=n,m:=n
MakeHeap(W,n) 
 While Count >1 {
i:=Heap(1)
Heap(1):= Heap(count)
Count:=Count-1
DownHeap(W,1,Count)
j:=Heap(1)
m:=m 1 
Huff(i):=-m
Huff(j):=m
W(m):=W(i) W(j)
Heap(1):=m
DownHeap(W,1,Count) 
} 
Return Huff
}

Xây dựng bộ mã

[sửa | sửa mã nguồn]

Sau khi cấu trúc của cây Huffman được lưu vào mảng Huff ta dễ dàng xây dựng mảng Code[1..n] cho bộ mã nhị phân tiền tố tối ưu của các ký tự A[1..n].

Procedure CodingHuffman(List Huff, n){
Int k:=1,j
 While k<=n {
Code(k):=""
j:=Huff(k)
While Abs(j)<=2*n-1 {
 If j>0 then Code(k)="1" Code(k) 
 else Code(k)="0" Code(k)
j:=Huff(abs(j)) 
} 
k:=k 1  
} 
Return Code
}

Nén file bằng mã Huffman

[sửa | sửa mã nguồn]

Trong các bước trên, giả sử đã xây dựng được bộ mã Huffman của 256 ký hiệu có mã ASCII từ 0 đến 255 chứa trong mảng Code[1..256]. Việc nén file có thể phân tích sơ bộ như sau:

  1. Đọc từng byte của file cần nén cho đến khi hết tệp,
  2. Chuyển theo bộ mã thành xâu nhị phân,
  3. Ghép với xâu nhị phân còn dư từ bước trước,
  4. Nếu có đủ 8 bit trong xâu thì cắt ra tám bít đầu tiên ghi vào tệp nén,
  5. Nếu đã hết tệp cần nén thì dừng...

Tham khảo

[sửa | sửa mã nguồn]

Liên kết ngoài

[sửa | sửa mã nguồn]