LeetCode in Kotlin

2452. Words Within Two Edits of Dictionary

Medium

You are given two string arrays, queries and dictionary. All words in each array comprise of lowercase English letters and have the same length.

In one edit you can take a word from queries, and change any letter in it to any other letter. Find all words from queries that, after a maximum of two edits, equal some word from dictionary.

Return a list of all words from queries, that match with some word from dictionary after a maximum of two edits. Return the words in the same order they appear in queries.

Example 1:

Input: queries = [“word”,”note”,”ants”,”wood”], dictionary = [“wood”,”joke”,”moat”]

Output: [“word”,”note”,”wood”]

Explanation:

• Changing the ‘r’ in “word” to ‘o’ allows it to equal the dictionary word “wood”.

• Changing the ‘n’ to ‘j’ and the ‘t’ to ‘k’ in “note” changes it to “joke”.

• It would take more than 2 edits for “ants” to equal a dictionary word.

• “wood” can remain unchanged (0 edits) and match the corresponding dictionary word.

Thus, we return [“word”,”note”,”wood”].

Example 2:

Input: queries = [“yes”], dictionary = [“not”]

Output: []

Explanation:

Applying any two edits to “yes” cannot make it equal to “not”. Thus, we return an empty array.

Constraints:

• 1 <= queries.length, dictionary.length <= 100
• n == queries[i].length == dictionary[j].length
• 1 <= n <= 100
• All queries[i] and dictionary[j] are composed of lowercase English letters.

Solution

class Solution {
    private var root: Node? = null

    internal class Node {
        var childs = HashMap<Char, Node?>()
    }

    private fun insert(s: String) {
        var curr = root
        for (ch in s.toCharArray()) {
            if (curr!!.childs[ch] == null) {
                curr.childs[ch] = Node()
            }
            curr = curr.childs[ch]
        }
    }

    private fun search(word: String, curr: Node?, i: Int, edits: Int): Boolean {
        // if reached the end with less than or equal 2 edits then return truem
        if (i == word.length) {
            return edits <= 2
        }
        // more than 2 mismatch don't go further
        if (edits > 2) {
            return false
        }
        // there might be a case start is matching but others are diff and that's a edge case to
        // handle
        var ans = false
        for (ch in curr!!.childs.keys) {
            ans = ans or search(
                word,
                curr.childs[ch],
                i + 1,
                if (ch == word[i]) edits else edits + 1,
            )
        }
        return ans
    }

    fun twoEditWords(queries: Array<String>, dictionary: Array<String>): List<String> {
        root = Node()
        for (s in dictionary) {
            insert(s)
        }
        val ans: MutableList<String> = ArrayList()
        for (s in queries) {
            val found = search(s, root, 0, 0)
            if (found) {
                ans.add(s)
            }
        }
        return ans
    }
}

This site is open source. Improve this page.