3043 find the length of the longest common prefix

[kitano-kazuki]

https://leetcode.com/problems/find-the-length-of-the-longest-common-prefix/description

[nodchip]

自分も解いてみました。 Trie 木を 2 つ作り、同時に dfs し、最大の高さを求めました。業務のコードでは、メモリを解放しないとまずいです。

class Solution {
public:
    struct Node {
        Node* children[10] = {};
    };

    Node* ToTrie(const vector<int>& arr) {
        Node* root = new Node();
        for (int integer : arr) {
            std::string s = std::to_string(integer);
            Node* node = root;
            for (char ch : s) {
                int digit = ch - '0';
                if (!node->children[digit]) {
                    node->children[digit] = new Node();
                }
                node = node->children[digit];
            }
        }
        return root;
    }

    int GetHeight(Node* node1, Node* node2) {
        if (!node1 || !node2) {
            return -1;
        }

        int max_height = 0;
        for (int digit = 0; digit < 10; ++digit) {
            int height = GetHeight(node1->children[digit], node2->children[digit]) + 1;
            max_height = max(max_height, height);
        }
        return max_height;
    }

    int longestCommonPrefix(vector<int>& arr1, vector<int>& arr2) {
        Node* root1 = ToTrie(arr1);
        Node* root2 = ToTrie(arr2);
        return GetHeight(root1, root2);
    }
};

[nodchip]

Hash を使うと書いてしまうと、ハッシュ関数を用いて prefix のハッシュ値を求めて何かを行うのかと誤解させてしまうかもしれません。集合を使うと書いたほうが、読み手にとって正確に理解できると思います。