g0201_0300.s0290_word_pattern.Solution.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-kotlin Show documentation

Kotlin-based LeetCode algorithm problem solutions, regularly updated

There is a newer version: 1.28

package g0201_0300.s0290_word_pattern

// #Easy #String #Hash_Table #Data_Structure_II_Day_7_String
// #2022_11_04_Time_177_ms_(89.55%)_Space_35.2_MB_(95.52%)

class Solution {
    fun wordPattern(pattern: String, s: String): Boolean {
        val map: MutableMap = HashMap()
        val words = s.split(" ".toRegex()).dropLastWhile { it.isEmpty() }.toTypedArray()
        if (words.size != pattern.length) {
            return false
        }
        for (i in 0 until pattern.length) {
            if (!map.containsKey(pattern[i])) {
                if (map.containsValue(words[i])) {
                    return false
                }
                map[pattern[i]] = words[i]
            } else {
                if (words[i] != map[pattern[i]]) {
                    return false
                }
            }
        }
        return true
    }
}