leetcode: 模式匹配

題目

你有兩個字符串，即pattern和value。 pattern字符串由字母"a"和"b"組成，用於描述字符串中的模式。例如，字符串"catcatgocatgo"匹配模式"aabab"（其中"cat"是"a"，“go"是"b”），該字符串也匹配像"a"、"ab"和"b"這樣的模式。但需注意"a"和"b"不能同時表示相同的字符串。編寫一個方法判斷value字符串是否匹配pattern字符串。

示例 1：

輸入： pattern = "abba", value = "dogcatcatdog"
輸出： true

示例 2：

輸入： pattern = "abba", value = "dogcatcatfish"
輸出： false

示例 3：

輸入： pattern = "aaaa", value = "dogcatcatdog"
輸出： false

示例 4：

輸入： pattern = "abba", value = "dogdogdogdog"
輸出： true
解釋： "a"="dogdog",b=""，反之也符合規則

提示：

• 0 <= len(pattern) <= 1000
• 0 <= len(value) <= 1000
• 你可以假設pattern只包含字母"a"和"b"，value僅包含小寫字母。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/pattern-matching-lcci
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

以下是我的go實現

func patternMatching(pattern string, value string) bool {
	// ac,bc 表示pattern中a和b出現的次數；al,bl 表示a和b模式對應的字符長度；c,l表示只有一種模式時候的次數和長度
	var ac, al, bc, bl, c, l int
	var as, bs, cs string
	// vl,pl表示value和pattern的長度
	var vl, pl = len(value), len(pattern)
	// 模式爲空，則value爲空爲true，否則false
	if pl == 0 {
		if vl == 0 {
			return true
		}
		return false
	}
	// 遍歷統計pattern中ac,bc
	for i := 0; i < pl; i++ {
		if string(pattern[i]) == "a" {
			ac++
		} else {
			bc++
		}
	}
	// 如果value爲空，ac和bc都不爲空，則返回false
	if vl == 0 {
		if ac != 0 && bc != 0 {
			return false
		}
	}
	// 只有a或b單一模式，value長度除與pattern長度則爲每個模式對應的字符長度，則可以推導出模式對應字符，重複c次則爲預計的字符串，與實際value比較則可以判斷是否匹配
	if ac == 0 || bc == 0 {
		if ac == 0 {
			c = bc
		} else {
			c = ac
		}
		if vl%c != 0 {
			return false
		}
		l = vl / c
		cs = value[:l]
		if strings.Repeat(cs, c) == value {
			return true
		}
		return false
	}
	// 遍歷可能的al，從0到vl
	for al = 0; al <= vl; al++ {
		// 計算對應的bl
		if (vl-ac*al)%bc == 0 {
			bl = (vl - ac*al) / bc
			if bl<0{
				continue
			}
		} else {
			continue
		}
		// 通過a和b的index，加上al和bl，可以推導出a和b對應的字符串as,bs
		var ai, bi int
		ai = strings.Index(pattern, "a")
		bi = strings.Index(pattern, "b")
		if ai == 0 {
			as = value[:al]
			bs = value[bi*al : bi*al+bl]
		} else {
			bs = value[:bl]
			as = value[ai*bl : ai*bl+al]
		}
		// 根據as和bs，帶入到pattern，拼裝出對應的字符串tmp
		var tmp = ""
		for i := 0; i < pl; i++ {
			if string(pattern[i]) == "a" {
				tmp += as
			} else {
				tmp += bs
			}
		}
		// tmp和value是否一致判斷
		if tmp == value {
			return true
		}
	}
	return false
}

執行結果：通過
執行用時：0 ms
內存消耗：3.3 MB