力扣 290. 单词规律

题目

给定一种规律 pattern 和一个字符串 str ，判断 str 是否遵循相同的规律。

这里的 遵循 指完全匹配，例如， pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应规律。

示例

输入: pattern = “abba”, str = “dog cat cat dog”
输出: true

输入:pattern = “abba”, str = “dog cat cat fish”
输出: false

输入: pattern = “aaaa”, str = “dog cat cat dog”
输出: false

输入: pattern = “abba”, str = “dog dog dog dog”
输出: false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/word-pattern
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：map+set

Java实现

class Solution {
    public boolean wordPattern(String pattern, String s) {
        String[] var = s.split(" ");
        if (pattern.length() != var.length) return false;

        Map<Character, String> map = new HashMap<>();

        for (int i = 0; i < pattern.length(); i++){
            //a/b 对应多个单词的情况
            if (!map.containsKey(pattern.charAt(i))) map.put(pattern.charAt(i), var[i]);
            else {
                if (!map.get(pattern.charAt(i)).equals(var[i])) return false;
            }
        }
		
		//a/b/c对应一个单词的情况
        Set<String> set = new HashSet<>();
        for (Character c : map.keySet()){
            if (!set.contains(map.get(c))) set.add(map.get(c));
            else return false;
        }
        return true;
    }
}