[Swift]LeetCode291. 单词模式 II $ Word Pattern II

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/10692315.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Given a pattern and a string str, find if strfollows the same pattern.

Here follow means a full match, such that there is a bijection between a letter in pattern and a non-empty substring in str.

Example 1:

Input: pattern = "abab", str = "redblueredblue"
Output: true

Example 2:

Input: pattern = "aaaa", str = "asdasdasdasd"
Output: true

Example 3:

Input: pattern = "aabb", str = "xyzabcxzyabc"
Output: false

Notes:
　　You may assume both pattern and str contains only lowercase letters.

---

给定一个模式和一个字符串str，找出str是否遵循相同的模式。

这里follow意味着完全匹配，这样在模式中的字母和str中的非空子字符串之间就有一个双射。

例1：

　　输入：pattern=“abab”，str=“redblueredblue”

　　输出：true

例2：

　　输入：pattern=“aaaa”，str=“asdasdasd”

　　输出：true

例3：

　　输入：pattern=“aabb”，str=“xyzabcxzyabc”

　　输出：false

注意：

　　您可以假定pattern和str都只包含小写字母。

---

Solution:

class Solution {
    var m:[Character:String] = [Character:String]()
    var s:Set<String> = Set<String>()
    func wordPatternMatch(_ pattern:String,_ str:String) -> Bool {
        if pattern.isEmpty {return str.isEmpty}
        var arrChar:[Character] = Array( pattern)
        if m[arrChar[0]] != nil
        {
            var t:String = m[arrChar[0]]!
            if t.count > str.count || str.subString(0, t.count) != t
            {
                return false
            }
            if wordPatternMatch(pattern.subString(1), str.subString(t.count))
            {
                return true
            }
        }
        else
        {
            for i in 1...str.count
            {
                if s.contains(str.subString(0, i)) {continue}
                m[arrChar[0]] = str.subString(0, i)
                s.insert(str.subString(0, i))
                if wordPatternMatch(pattern.subString(1), str.subString(i))
                {
                    return true
                }
                m[arrChar[0]] = nil
                s.remove(str.subString(0, i))
            }
        }
        return false        
    }
}

extension String {
    // 截取字符串：从index到结束处
    // - Parameter index: 开始索引
    // - Returns: 子字符串
    func subString(_ index: Int) -> String {
        let theIndex = self.index(self.endIndex, offsetBy: index - self.count)
        return String(self[theIndex..<endIndex])
    }
    
    // 截取字符串：指定索引和字符数
    // - begin: 开始截取处索引
    // - count: 截取的字符数量
    func subString(_ begin:Int,_ count:Int) -> String {
        let start = self.index(self.startIndex, offsetBy: max(0, begin))
        let end = self.index(self.startIndex, offsetBy:  min(self.count, begin + count))
        return String(self[start..<end]) 
    }
}

---

点击：Playground测试

print(Solution().wordPatternMatch("abab","redblueredblue"))
//Print true
print(Solution().wordPatternMatch("aaaa","asdasdasdasd"))
//Print true
print(Solution().wordPatternMatch("aabb","xyzabcxzyabc"))
//Print false