如何在Java中实现多维度到唯一值的哈希映射及前缀查询功能？

class Dimension {
    String a;
    String b;
    String c;
    // ... other dimensions

    public Dimension(String a, String b, String c) {
        this.a = a;
        this.b = b;
        this.c = c;
    }

    // equals() and hashCode() methods for HashMap comparison
    @Override
    public boolean equals(Object obj) {
        if (this == obj) return true;
        if (obj == null || getClass() != obj.getClass()) return false;
        Dimension that = (Dimension) obj;
        return Objects.equals(a, that.a) && Objects.equals(b, that.b) && Objects.equals(c, that.c);
    }

    @Override
    public int hashCode() {
        return Objects.hash(a, b, c);
    }
}

class TrieNode {
    String value;
    Map<String, TrieNode> children;
    String uniqueId; // Store unique ID at leaf nodes

    public TrieNode(String value) {
        this.value = value;
        this.children = new HashMap<>();
    }
}

class Trie {
    TrieNode root;

    public Trie() {
        root = new TrieNode("");
    }

    public void insert(Dimension dim, String uniqueId) {
        TrieNode node = root;
        node = insertRecursive(node, dim, uniqueId);
    }

    private TrieNode insertRecursive(TrieNode node, Dimension dim, String uniqueId) {
        if (dim == null) {
            node.uniqueId = uniqueId;
            return node;
        }
        if (dim.a != null) {
            node.children.computeIfAbsent(dim.a, k -> new TrieNode(k));
            node = node.children.get(dim.a);
            if (dim.b != null) {
                node.children.computeIfAbsent(dim.b, k -> new TrieNode(k));
                node = node.children.get(dim.b);
                if (dim.c != null) {
                    node.children.computeIfAbsent(dim.c, k -> new TrieNode(k));
                    node = node.children.get(dim.c);
                }
            }
        }
        node.uniqueId = uniqueId;
        return node;
    }


    public List<String> prefixSearch(String prefix) {
        List<String> result = new ArrayList<>();
        TrieNode node = root;
        for (String part : prefix.split(",")) {
            if (!node.children.containsKey(part)) {
                return result; // Prefix not found
            }
            node = node.children.get(part);
        }
        collectIds(node, result);
        return result;
    }

    private void collectIds(TrieNode node, List<String> result) {
        if (node.uniqueId != null) {
            result.add(node.uniqueId);
        }
        for (TrieNode child : node.children.values()) {
            collectIds(child, result);
        }
    }
}

public class Main {
    public static void main(String[] args) {
        Trie trie = new Trie();
        trie.insert(new Dimension("a", "b", "c"), "u1");
        trie.insert(new Dimension("a", "b", "d"), "u2");
        trie.insert(new Dimension("x", "y", "z"), "v1");

        List<String> results = trie.prefixSearch("a,b");
        System.out.println(results); // Output: [u1, u2]

        results = trie.prefixSearch("a");
        System.out.println(results); // Output: [u1, u2]

        results = trie.prefixSearch("x");
        System.out.println(results); // Output: [v1]
    }
}