Leetcode290. 单词规律
题目:
给定一种规律 pattern 和一个字符串 str ,判断 str 是否遵循相同的规律。
这里的 遵循 指完全匹配,例如, pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应规律。
示例1:
输入: pattern = "abba", str = "dog cat cat dog"
输出: true
题解:
维护一个map,其中pattern每个字符为key,str按空格切分后的值作为value,查看每个key和value是否匹配。
scala代码:
def wordPattern(pattern: String, str: String): Boolean = {
var flag = true
val arr = str.split(" ")
val map = new util.HashMap[Char, String]()
if (pattern.length != arr.length) {
flag = false
} else {
for (i <- 0 until pattern.length; if (flag)) {
val key = pattern.charAt(i)
if (!map.containsKey(key)) {
map.put(key, arr(i))
} else {
if (map.get(key) != arr(i)) {
flag = false
}
}
}
}
flag
}