|
| 1 | +import java.util.ArrayList; |
| 2 | +import java.util.HashMap; |
| 3 | +import java.util.PriorityQueue; |
| 4 | + |
| 5 | +public class HuffmanEncoder { |
| 6 | + private HashMap<Character, String> encoder = new HashMap<>(); |
| 7 | + private HashMap<String, Character> decoder = new HashMap<>(); |
| 8 | + |
| 9 | + private class Node implements Comparable<Node> { |
| 10 | + char data; |
| 11 | + int freq; |
| 12 | + Node left; |
| 13 | + Node right; |
| 14 | + |
| 15 | + @Override |
| 16 | + public int compareTo(Node o) { |
| 17 | + return this.freq - o.freq; |
| 18 | + } |
| 19 | + } |
| 20 | + |
| 21 | + public HuffmanEncoder(String feeder) { |
| 22 | + // Step 1 -> Feeder to fmap |
| 23 | + HashMap<Character, Integer> fmap = new HashMap<>(); |
| 24 | + |
| 25 | + for (int i = 0; i < feeder.length(); i++) { |
| 26 | + char ch = feeder.charAt(i); |
| 27 | + fmap.put(ch, fmap.containsKey(ch) ? fmap.get(ch) + 1 : 1); |
| 28 | + } |
| 29 | + |
| 30 | + // Step 2 -> fmap to Priority Queue |
| 31 | + ArrayList<Character> keys = new ArrayList<>(fmap.keySet()); |
| 32 | + PriorityQueue<Node> pq = new PriorityQueue<>(); |
| 33 | + for (Character key : keys) { |
| 34 | + Node node = new Node(); |
| 35 | + node.freq = fmap.get(key); |
| 36 | + node.data = key; |
| 37 | + pq.add(node); |
| 38 | + } |
| 39 | + |
| 40 | + // Step 3 -> while PQ.size > 1, remove 2 from PQ, merge them and add in |
| 41 | + // PQ |
| 42 | + while (pq.size() > 1) { |
| 43 | + Node one = pq.remove(); |
| 44 | + Node two = pq.remove(); |
| 45 | + Node merged = new Node(); |
| 46 | + merged.freq = one.freq + two.freq; |
| 47 | + merged.left = one; |
| 48 | + merged.right = two; |
| 49 | + pq.add(merged); |
| 50 | + } |
| 51 | + |
| 52 | + Node lastNodeInPQ = pq.remove(); |
| 53 | + traverse(lastNodeInPQ, ""); |
| 54 | + System.out.println(encoder); |
| 55 | + } |
| 56 | + |
| 57 | + private void traverse(Node node, String psf) { |
| 58 | + if (node.left == null && node.right == null) { |
| 59 | + encoder.put(node.data, psf); |
| 60 | + decoder.put(psf, node.data); |
| 61 | + return; |
| 62 | + } |
| 63 | + traverse(node.left, psf + "0"); |
| 64 | + traverse(node.right, psf + "1"); |
| 65 | + } |
| 66 | + |
| 67 | + public String encode(String str) { |
| 68 | + String coded = ""; |
| 69 | + for (int i = 0; i < str.length(); i++) { |
| 70 | + coded += encoder.get(str.charAt(i)); |
| 71 | + } |
| 72 | + |
| 73 | + return coded; |
| 74 | + } |
| 75 | + |
| 76 | + public String decode(String str) { |
| 77 | + String ch = ""; |
| 78 | + String decoded = ""; |
| 79 | + |
| 80 | + for (int i = 0; i < str.length(); i++) { |
| 81 | + ch += str.substring(i, i + 1); |
| 82 | + if (decoder.containsKey(ch)) { |
| 83 | + decoded += decoder.get(ch); |
| 84 | + ch = ""; |
| 85 | + } |
| 86 | + } |
| 87 | + |
| 88 | + return decoded; |
| 89 | + } |
| 90 | + |
| 91 | +} |
0 commit comments